- 博客(1)
- 资源 (1)
- 收藏
- 关注
原创 pytorch: DiceLoss MulticlassDiceLoss
pytorch的自定义多类dice_loss 和单类dice_loss:import torchimport torch.nn as nn class DiceLoss(nn.Module): def __init__(self): super(DiceLoss, self).__init__() def forward(self, input, target): N = ...
2019-04-25 10:50:09 1336 3
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人