Implementation of dice loss - vision - PyTorch Forums
discuss.pytorch.org › t › implementation-of-diceAug 16, 2019 · Hi All, I am trying to implement dice loss for semantic segmentation using FCN_resnet101. For some reason, the dice loss is not changing and the model is not updated. import torch import torchvision import loader from loader import DataLoaderSegmentation import torch.nn as nn import torch.optim as optim import numpy as np from torch.utils.data.sampler import SubsetRandomSampler from torch ...
Multi class dice loss function - PyTorch Forums
discuss.pytorch.org › t › multi-class-dice-lossOct 04, 2020 · Hello everyone, i am trying to use dice loss for my 3D point cloud semantic segmentation model. Although, I have implemented the function by referencing some of the codes, I am not sure whether it is correct as my IoU for my validation set does not increase compare to using cross entropy loss solely. Below is my function for multi class dice loss: def diceLoss(prediction_g, label_g, num_class ...