Closed helenxxz closed 1 year ago
hey, everyone! How do you implement the edge pixel weighting in the code? I only see this: self.criterion = torch.nn.BCELoss() loss = self.criterion(SR_flat,GT_flat) Hope to get your answers, thank you!
self.criterion = torch.nn.BCELoss() loss = self.criterion(SR_flat,GT_flat)
hey, everyone! How do you implement the edge pixel weighting in the code? I only see this:
self.criterion = torch.nn.BCELoss() loss = self.criterion(SR_flat,GT_flat)
Hope to get your answers, thank you!