InPeerReview/RemoteSensingChangeDetection-RSCD.HA2F
2
1a
2 8��g� � @ sL d dl ZG dd� de�ZG dd� de�Zdd� Zdd � Zd3d � Zd d
� ZdS )� Nc @ sR e Zd ZdZdd� Zdd� Zddd�Zd d4� Zd d � Zd
d� Z dd� Z5dd� Z dS )� AverageMeter�1Computes and stores the average and current valuec C s" d| _ d | _d | _d | _d | _d S �NF)� initialized�val�avg�sum�count��self� r �CC:\Users\82607\Desktop\code\ChangeViT-master\.\model\metric_tool.py�__init__ s6 zAverageMeter.__init__c C s& || _ || _|| | _|| _d| _d S )NT)r r r r r �r r �weightr r r
�7initialize s8 9zAverageMeter.initialize� c C s$ | j s| �||� n | �||� d S �N)r r �addr r r r
�update s zAverageMeter.updatec C s8 || _ | j|| 7 _| j|7 _| j| j | _d S r )r r r r r r r r
r s zAverageMeter.addc C s | j S r )r r10 r r r
�value" s zAverageMeter.valuec C s | j S r )r r11 r r r
�average% s zAverageMeter.averagec C s t | j�}|S r ��cm2scorer �r Z scores_dictr r r
�12get_scores( s 13zAverageMeter.get_scoresc C s14 d| _ d S r )r r15 r r r
�clear, s zAverageMeter.clearN)r ) �__name__�16__module__� __qualname__�__doc__r r r r r r r r r r r r
r s 17r c s2 e Zd ZdZ� fdd�Zd dd�Zdd� Z� ZS )18�ConfuseMatrixMeterr c s t t| ��� || _d S r )�superr! r �n_class)r r# �� __class__r r
r 4 s zConfuseMatrixMeter.__init__r c C s( t | j||d�}| �||� t|�}|S )uJ 获得当前混淆矩阵，并计算当前F1得分，并更新混淆矩阵)� num_classes� label_gts� label_preds)�get_confuse_matrixr# r �cm2F1)r �pr�gtr r Z
current_scorer r r
� update_cm8 s z ConfuseMatrixMeter.update_cmc C s t | j�}|S r r r r r r
r ? s 19z ConfuseMatrixMeter.get_scores)r )r r r r r r- r �
__classcell__r r r$ r
r! 1 s 20r! c C s t | �tdd� | D �� }|S )Nc s s | ]}|d d V qdS )g�����ư>�����Nr )�.0�xr r r
� <genexpr>E � z harmonic_mean.<locals>.<genexpr>)�lenr )�xs�
harmonic_meanr r r
r6 D s r6 c C s~ | }|d }|d }|d }|d }||| t �t j�j }||| t �t j�j }d| | || t �t j�j }|S )N�r r �r r �r r �r r � ��np�finfo�float32�eps) �confusion_matrix�hist�tp�fn�fp�tn�recall� precision�f1r r r
r* I s "r* c C s | }|d }|d }|d }|d }|| || | | t �t j�j }||| t �t j�j }||| t �t j�j }d| | || t �t j�j } ||| | t �t j�j }21|| || || || || | | d } || d| } | |22| |||| d�}
|
S )Nr7 r8 r9 r: r; r )�Kappa�IoU�F1�OArG rH ZPrer< )rA rB rC rD rE rF �oarG rH rI Ziou�pre�kappaZ23score_dictr r r
r X s &" 4r c sH � fdd�}t �� � f�}t||�D ] \}}|||�� |�� �7 }q$|S )u! 计算一组预测的混淆矩阵c sD | dk| � k @ }t j� | | �t� || � d d��� � �}|S )a# 24 Collect values for Confusion Matrix25 For reference, please see: https://en.wikipedia.org/wiki/Confusion_matrix26 :param label_gt: <np.array> ground-truth27 :param label_pred: <np.array> prediction28 :return: <np.ndarray> values for confusion matrix29 r r; )� minlength)r= �bincount�astype�int�reshape)Zlabel_gtZ30label_pred�maskrB �r& r r
� __fast_hists s ��z'get_confuse_matrix.<locals>.__fast_hist)r= �zeros�zip�flatten)r& r' r( rX rA �lt�lpr rW r
r) p s31
r) ) �numpyr= �objectr r! r6 r* r r) r r r r
�<module> s ,