56 lines (56 with data), 7.2 kB

B


`╔aŃ
@s
dd
lm

mZ
ddlm
Z

dd
lZddlTGddäde
jâZGddäde
jâZGd	d
äd
e
jâZ	Gddäde
jâZ
Gd
däde
jâZGddäde
jâZGddäde
jâZ
ddäZGddäde
jâZGddäde
jâZe	e
ee
eeeeedť	ZddäZddäZedkÉ
r
d
S)ÚN)
┌nnÚ
)
┌
*cs$eZ
dZçf
d
däZddäZçZS)┌BCELossc
stt
|âáí
táí|_dS)
N)┌superr┌__init__r┌BCEWithLogitsLoss┌bce_loss)┌self┌kwargs)
┌	__class__ę˙4/home/wangqiuli/Code/Newcode1222_monk/loss/losses.pyr
s

zBCELoss.__init__cCs|á|
|íS)
N)
r	)r
┌output┌targetr
r
r┌forwards
zBCELoss.forward)┌__name__┌
__module__┌__qualname__rr┌
__classcell__r
r
)
rrr	s
rcs$eZ
dZçf
d
däZddäZçZS)┌MSELossc
stt
|âáí
tá
í|_dS)
N)rrrr┌mse_loss)r
r)
rr
rrs

zMSELoss.__init__cCs|á|
|íS)
N)
r)r
rrr
r
rrs
zMSELoss.forward)rrrrrrr
r
)
rrrs
rcs$eZ
dZçf
d
däZddäZçZS)┌CELossc
stt
|âáí
táí|_dS)
N)rrrr┌CrossEntropyLoss┌ce_loss)r
r)
rr
rrs

zCELoss.__init__cCs|á|
|íS)
N)
r)r
rrr
r
rrs
zCELoss.forward)rrrrrrr
r
)
rrrs
rcs0eZ
dZçf
d
däZd	ddä
Zd
ddä
ZçZS)┌CELoss2dc
stt
|âáí
dS)
N)rrr)r
r)
rr
rr"s
zCELoss2d.__init__NTcCsÉ|
áí\}}}}|
á
d
díá
ddíáí}
|
|á|||d
íád
d
d
|ídk}
|
ád|í}
|dk}	||	}tj|
||ddŹ}
|rî|
|	jáí}
|
S)NrÚÚr
Ú    F)┌weight┌size_average)	┌size┌	transpose┌
contiguous┌view┌repeat┌
F┌
cross_entropy┌data┌sum)r
┌inputrrr ┌
n┌
c┌
h┌
w┌mask┌lossr
r
r┌cross_entropy2d%s

$





zCELoss2d.cross_entropy2dcCs|á|
|||íS)
N)
r1)r
rrrr r
r
rr3s
zCELoss2d.forward)NT)NT)rrrrr1rrr
r
)
rrr!s

rcs$eZ
dZçf
d
däZddäZçZS)┌BinaryDiceLossc

stt
|âáí
dS)
N)rr2r)
r
)
rr
rr7s
zBinaryDiceLoss.__init__c
Csn|áíd
}d}|
á
|dí}|á
|dí}||}d|ádí
||ádí
|ádí
|}d|áí|}	|	S)Nr
rrr)r!r$r))
r
r*┌targets┌
NZsmoothZ
input_flatZtargets_flat┌intersectionZ
N_dice_effr0r
r
rr:s

*
zBinaryDiceLoss.forward)rrrrrrr
r
)
rrr26s
r2cs$eZ
dZçf
d
däZddäZçZS)┌DiceLossc

stt
|âáí
dS)
N)rr6r)
r
)
rr
rrKs
zDiceLoss.__init__cCsĂ|
d
k}
|tá
|í
k}t|
â
}
t|â
}|
|}tj|ddŹ}tj|ddŹ}|
|
}tj|ddŹ}tj|ddŹ}||}tj|ddŹ}tj|ddŹ}d|||d}ddtá|í
|ádí
}|S)NgÓ?r)
┌dimgŹÝÁá¸Ă░>rr
)┌torch┌max┌to_float_and_cudar)r!)r
r*r┌num┌den1┌den2┌dice┌
dice_totalr
r
rrNs 









zDiceLoss.forward)rrrrrrr
r
)
rrr6Js
r6cs(eZ
dZd
Zçf
ddäZddäZçZS)┌ContrastiveLosszt
    Contrastive loss function.
    Based on: http://yann.lecun.com/exdb/publis/pdf/hadsell-chopra-lecun-06.pdf
    c
stt
|âáí
dS)
N)rr@r)r
r)
rr
rrls
zContrastiveLoss.__init__cCs"tá
|
|í}tátá|d
íí
}|S)Nr)r&┌pairwise_distancer8┌mean┌pow)r
Zoutput1Zoutput2Zeuclidean_distanceZloss_contrastiver
r
rros


zContrastiveLoss.forward)rrr┌__doc__rrrr
r
)
rrr@fsr@cCs$tá
|á|
í
ád
í
á|ádí
díí
S)z]
    attention value of a feature map
    :param x: feature
    :return: attention value
    rr
r)r&┌	normalizerCrBr$r!)┌
x┌expr
r
r┌attsrHcs&eZ
dZçf
d
däZdddä
ZçZS)┌IMLossc
stt
|âáí
dS)
N)rrIr)r
r)
rr
rr}s
zIMLoss.__init__ÚcCsX|
jd
|jd
kr0t
j|
|áíddůddŹ}
tjt|
|ât||âád
í
ddŹáíS)NrÚ■   ┌bilinear)
┌moder)
r7)	┌shaper&┌interpolater!r8r)rHrCrB)r
Zfeature1Zfeature2rGr
r
rrüs


zIMLoss.forward)
rJ)rrrrrrr
r
)
rrrI|s
rIcs$eZ
dZçf
d
däZddäZçZS)┌
KDChannelLossc
stt
|âáí
dS)
N)rrPr)r
r)
rr
rrçs
zKDChannelLoss.__init__c
CsÄ
d
}g}g}d}É
xnt|â
DÉ
]`}	d}
t
á|ddů|	ddůddůfdíád|ddgí
}|
|}t
á|dddgít
á|ddů|	ddůddůfí
|
}
tj|
|ddŹ}|á|í

t
á|ddů|	ddůddůfdíád|ddgí
}||}t
á|dddgít
á|ddů|	ddůddůfí
|
}tj||ddŹ}|á|í

t
á|t
á||í
í
t
á|t
á||í
í
d}||7}qW||}|S)	Ngg@gŹÝÁá¸Ă░>rr
rr)
r7)	┌ranger8┌	unsqueezer%r)r&┌softmax┌append┌log)r
Z
source_logitsZ
target_logits┌gt┌num_classesZkd_lossZsource_probZtarget_prob┌temperature┌
i┌epsZs_maskZs_logits_mask_outZs_logits_avgZs_soft_probZt_maskZt_logits_mask_outZt_logits_avgZt_soft_probr0r
r
rrŐs(

2

8

2

8

4zKDChannelLoss.forward)rrrrrrr
r
)
rrrPćs
rP)	r'r1┌	BCE_logitr@r2r6rI┌	KDChannelrcCstá
íS)
N)┌	__factory┌keysr
r
r
r┌	get_names╝s
r_c
Ks(|tá
íkrtd
á|í
â
é
t|f|
Ä
S)NzUnknown loss: {})r]r^┌KeyError┌format)┌namerr
r
r┌	init_loss└s


rc┌__main__)Ztorch.nn.functionalr┌
functionalr&r8┌utils┌Modulerrrrr2r6r@rHrIrPr]r_rcrr
r
r
r┌<module>s4



*