o õq›g(ã@s<ddlZddlmZmZGdd„deƒZdd„Zdd„ZdS) éN)ÚFunctionÚVariablec@s eZdZdZdd„Zdd„ZdS)Ú DiceCoeffz"Dice coeff for individual examplescCsd| ||¡d}t | d¡| d¡¡|_t |¡t |¡||_d|j ¡||j ¡}|S)Ng-Cëâ6?éÿÿÿÿé)Úsave_for_backwardÚtorchÚdotÚviewÚinterÚsumÚunionÚfloat)ÚselfÚinputÚtargetÚepsÚt©rú8C:\muni\DP\seg1\Segment1\utils\dice_loss_single_class.pyÚforwardszDiceCoeff.forwardcCsV|j\}}d}}|jdr |d||j|j|j|j}|jdr'd}||fS)Nrré)Úsaved_variablesÚneeds_input_gradr r)rÚgrad_outputrrÚ grad_inputZgrad_targetrrrÚbackwards ÿ zDiceCoeff.backwardN)Ú__name__Ú __module__Ú__qualname__Ú__doc__rrrrrrrsrcCsn|jr t d¡ ¡ ¡}nt d¡ ¡}d}tt||ƒƒD]\}}|tƒ |d|d¡}|}q||dS)zDice coeff for batchesrr) Úis_cudarÚFloatTensorÚcudaÚzero_Ú enumerateÚziprr)rrÚsÚjÚiÚcrrrÚ dice_coeff$sr+cCsdt||ƒS)Nr)r+)rrrrrÚdice_coeff_loss4sr,)rÚtorch.autogradrrrr+r,rrrrÚs