Downloads: 1
20 lines (20 with data), 2.6 kB
B
Y®eN ã @ sh d dl Z d dlmZ d dlm mZ d dlmZ G dd deZdd Z dd Z
G d d
d
ej Z dS ) é N)ÚFunctionc @ s e Zd ZdZdd Zdd ZdS )Ú DiceCoeffz"Dice coeff for individual examplesc C sl | ||¡ d}t | ¡ d¡| ¡ d¡¡| _t |¡t |¡ | | _d| j ¡ | | j ¡ }|S )Ng-C ëâ6?éÿÿÿÿé ) Úsave_for_backwardÚtorchÚdotÚ
contiguousÚviewÚinterÚsumÚunionÚfloat)ÚselfÚinputÚtargetÚepsÚt© r ú@/Users/hanxuegu/Documents/Documents/bone/bone_paper_codes/dsc.pyÚforward s " zDiceCoeff.forwardc C sV | j \}}d }}| jd r@|d || j | j | j| j }| jd rNd }||fS )Nr r é )Úsaved_variablesÚneeds_input_gradr
r )r Ú grad_outputr r Ú
grad_inputZ grad_targetr r r Úbackward s
$
zDiceCoeff.backwardN)Ú__name__Ú
__module__Ú __qualname__Ú__doc__r r r r r r r s r c C sj | j rt d¡ ¡ ¡ }nt d¡ ¡ }x4tt| |D ]"\}}|t |d |d ¡ }q8W ||d S )zDice coeff for batchesr r ) Úis_cudar Ú FloatTensorÚcudaÚzero_Ú enumerateÚzipr r )r r ÚsÚiÚcr r r Ú
dice_coeff" s r* c C s~ d}| d¡}| d¡}t t ||¡¡}d| t ||¡ ¡ t ||¡ ¡ | }d| t ||¡ ¡ | }d| | S )Ng:0âyE>r g @g ð?r )r
r r Úmul)r r ÚsmoothÚy_true_fÚy_pred_fÚ intersectionÚdiceÚsenr r r Ú DICESEN_loss. s
, r2 c s& e Zd Z fddZdddZ ZS )ÚDiceSensitivityLossc s || _ tt| ¡ d S )N)Ú n_classesÚsuperr3 Ú__init__)r r4 )Ú __class__r r r6 8 s z DiceSensitivityLoss.__init__ç ð?c C s | j dkrt |¡}ntj|dd}| d¡}| d¡}|| ¡ }d| | | ¡ | ¡ | }d| t ||¡ ¡ | }d| | S )Nr )Údimr g @g ð?r )r4 r ÚsigmoidÚFÚsoftmaxr
r r+ ) r ÚinputsÚtargetsr, r- r. r/ r0 r1 r r r r < s
zDiceSensitivityLoss.forward)r8 )r r r r6 r Ú
__classcell__r r )r7 r r3 7 s r3 )
r Útorch.nnÚnnÚtorch.nn.functionalÚ
functionalr; Útorch.autogradr r r* r2 ÚModuler3 r r r r Ú<module> s