Datasets
Models
Downloads: 1
54 lines (54 with data), 7.1 kB
3
F3Ža Ń
@ s d dl jjZd dlmZ d dlZddlT G ddä dejâZG ddä dejâZG d d
ä d
ejâZ G d d ä d ejâZ
G d
dä dejâZ G ddä dejâZ G ddä dejâZ
ddä ZG ddä dejâZG ddä dejâZe e
ee
e e eeedť Zddä Zd d ä Zed kÉrdS ) Ú N)┌nnÚ )┌*c s$ e Zd Zç fddäZddä Zç ZS )┌BCELossc s t t| âjâ tjâ | _d S )N)┌superr ┌__init__r ┌BCEWithLogitsLoss┌bce_loss)┌self┌kwargs)┌ __class__ę ˙'/home/ws/yanghan/Newcode/loss/losses.pyr
s zBCELoss.__init__c C s | j ||âS )N)r )r
┌output┌targetr
r
r ┌forward s zBCELoss.forward)┌__name__┌
__module__┌ __qualname__r r ┌
__classcell__r
r
)r r r s r c s$ e Zd Zç fddäZddä Zç ZS )┌MSELossc s t t| âjâ tjâ | _d S )N)r r r r ┌mse_loss)r
r )r r
r r s zMSELoss.__init__c C s | j ||âS )N)r )r
r r r
r
r r s zMSELoss.forward)r r r r r r r
r
)r r r s r c s$ e Zd Zç fddäZddä Zç ZS )┌CELossc s t t| âjâ tjâ | _d S )N)r r r r ┌CrossEntropyLoss┌ce_loss)r
r )r r
r r s zCELoss.__init__c C s | j ||âS )N)r )r
r r r
r
r r s zCELoss.forward)r r r r r r r
r
)r r r s r c s0 e Zd Zç fddäZd ddäZd
ddäZç ZS ) ┌CELoss2dc s t t| âjâ d S )N)r r r )r
r )r r
r r " s zCELoss2d.__init__NTc C sÉ |j â \}}}}|jddâjddâjâ }||j|||dâjddd|âdk }|jd|â}|dk} || }tj|||ddŹ}
|rî|
| jjâ }
|
S )Nr Ú Ú r F)┌weight┌ size_averageÚ ) ┌size┌ transpose┌
contiguous┌view┌repeat┌F┌
cross_entropy┌data┌sum) r
┌inputr r r ┌n┌c┌h┌w┌mask┌lossr
r
r ┌cross_entropy2d% s $ zCELoss2d.cross_entropy2dc C s | j ||||âS )N)r1 )r
r r r r r
r
r r 3 s zCELoss2d.forward)NT)NT)r r r r r1 r r r
r
)r r r ! s
r c s$ e Zd Zç fddäZddä Zç ZS )┌BinaryDiceLossc s t t| âjâ d S )N)r r2 r )r
)r r
r r 7 s zBinaryDiceLoss.__init__c
C sn |j â d }d}|j|dâ}|j|dâ}|| }d|jdâ | |jdâ|jdâ | }d|jâ | } | S )Nr r r r r )r! r$ r) )
r
r* ┌targets┌NZsmoothZ
input_flatZ targets_flat┌ intersectionZ
N_dice_effr0 r
r
r r : s *zBinaryDiceLoss.forward)r r r r r r r
r
)r r r2 6 s r2 c s$ e Zd Zç fddäZddä Zç ZS )┌DiceLossc s t t| âjâ d S )N)r r6 r )r
)r r
r r K s zDiceLoss.__init__c C sĂ |dk}|t j|âk}t|â}t|â}|| }t j|ddŹ}t j|ddŹ}|| }t j|ddŹ}t j|ddŹ}|| }t j|ddŹ}t j|ddŹ}d||| d }ddt j|â |jdâ }|S )Ng Ó?r )┌dimgŹÝÁá¸Ă░>r r )┌torch┌max┌to_float_and_cudar) r! )r
r* r ┌num┌den1┌den2┌dice┌
dice_totalr
r
r r N s zDiceLoss.forward)r r r r r r r
r
)r r r6 J s r6 c s( e Zd ZdZç fddäZddä Zç ZS )┌ContrastiveLosszt
Contrastive loss function.
Based on: http://yann.lecun.com/exdb/publis/pdf/hadsell-chopra-lecun-06.pdf
c s t t| âjâ d S )N)r r@ r )r
r )r r
r r l s zContrastiveLoss.__init__c C s" t j||â}tjtj|dââ}|S )Nr )r& ┌pairwise_distancer8 ┌mean┌pow)r
Zoutput1Zoutput2Zeuclidean_distanceZloss_contrastiver
r
r r o s zContrastiveLoss.forward)r r r ┌__doc__r r r r
r
)r r r@ f s r@ c C s$ t j| j|âjdâj| jdâdââS )z]
attention value of a feature map
:param x: feature
:return: attention value
r r r )r& ┌ normalizerC rB r$ r! )┌x┌expr
r
r ┌att s rH c s& e Zd Zç fddäZdddäZç ZS )┌IMLossc s t t| âjâ d S )N)r rI r )r
r )r r
r r } s zIMLoss.__init__Ú c C sX |j d |j d kr0tj||jâ dd ů ddŹ}tjt||ât||â jdâddŹjâ S )Nr ┌bilinear)┌moder )r7 Ú■ ) ┌shaper& ┌ interpolater! r8 r) rH rC rB )r
Zfeature1Zfeature2rG r
r
r r ü s zIMLoss.forward)rJ )r r r r r r r
r
)r r rI | s rI c s$ e Zd Zç fddäZddä Zç ZS )┌
KDChannelLossc s t t| âjâ d S )N)r rP r )r
r )r r
r r ç s zKDChannelLoss.__init__c C sÄ d}g }g }d}Éxnt |âD É]`} d}
tj|d d ů| d d ůd d ůf dâjd|ddgâ} || } tj| dddgâtj|d d ů| d d ůd d ůf â|
}
tj|
| ddŹ}|j|â tj|d d ů| d d ůd d ůf dâjd|ddgâ}|| }tj|dddgâtj|d d ů| d d ůd d ůf â|
}tj|| ddŹ}|j|â tj|tj|| â âtj|tj|| â â d }||7 }q W || }|S ) Ng g @gŹÝÁá¸Ă░>r r r r )r7 ) ┌ranger8 ┌ unsqueezer% r) r& ┌softmax┌append┌log)r
Z
source_logitsZ
target_logits┌gt┌ num_classesZkd_lossZ source_probZ target_prob┌ temperature┌i┌epsZs_maskZs_logits_mask_outZ s_logits_avgZ s_soft_probZt_maskZt_logits_mask_outZ t_logits_avgZ t_soft_probr0 r
r
r r Ő s( 28
28
4 zKDChannelLoss.forward)r r r r r r r
r
)r r rP ć s rP ) r' r1 ┌ BCE_logitr@ r2 r6 rI Z KDChannelr c C s t jâ S )N)┌ __factory┌keysr
r
r
r ┌ get_names╝ s r^ c K s( | t jâ krtdj| ââét | f |ÄS )NzUnknown loss: {})r\ r] ┌KeyError┌format)┌namer r
r
r ┌ init_loss└ s rb ┌__main__)Ztorch.nn.functionalr ┌
functionalr& r8 ┌utils┌Moduler r r r r2 r6 r@ rH rI rP r\ r^ rb r r
r
r
r ┌<module> s4
*