18 lines (18 with data), 1.9 kB

3

á\ľa░Ń@s\dd
lZddl
TddlmZ
dd
ljjZddlmZ
GddädejâZ	Gdd	äd	ejâZ
d
S)
ÚNÚ
)
┌
*)
┌nn)
┌Variablecs*eZ
dZd
Zdçf
ddä	ZddäZçZS)	┌	FocalLossz.
    Focal loss for binary classification
    Úšđ?cstt
|âjâ
|
|_||_dS)
N)┌superr┌__init__┌gamma┌alpha)┌selfrr)
┌	__class__ę˙(/home/ws/yanghan/Newcode/losses/focal.pyr
s


zFocalLoss.__init__cCsttj
|
d
dŹ}|ddůd
f}|jd
||j|tj|â
d
|j||jd
|tjd
|â
}|jâS)Nr)
┌dim)┌torch┌softmaxrr┌log┌mean)r
┌input┌target┌pt┌
p┌lossrrr┌forwards


"
,
zFocalLoss.forward)rr)┌__name__┌
__module__┌__qualname__┌__doc__r
r┌
__classcell__rr)
rrrs
rcs&eZ
dZd	çf
ddä	ZddäZçZS)
┌FocalLoss2drr
TÚ cs*tt
|âjâ
|
|_||_||_||_dS)
N)r	r!r
rr┌ignore_index┌size_average)r
rrr$r#)
rrrr
s





zFocalLoss2d.__init__cCsbt|
â
}
t
|â
}tj|
|d
|jdŹ}tj|â
}|jd||j|}|j	rV|j
âS|jâSdS)N┌none)┌	reductionr#r)┌to_float_and_cuda┌to_long_and_cuda┌
F┌
cross_entropyr#r┌exprrr$r┌sum)r
┌inputs┌targets┌ce_lossr┌
focal_lossrrrr#s






zFocalLoss2d.forward)rr
Tr")rrrr
rr rr)
rrr!s
r!)r┌utilsrZtorch.nn.functional┌
functionalr)Ztorch.autogradr┌Modulerr!rrrr┌<module>
s