Open Gift-OYS opened 1 year ago
We have updated the new code, are there any problems left?
Thank you for this nice work. I am running this code and when I first run the full_supervised_segmentation/run.sh, it cannot find utils.cldice. I'm wondering if there is any missing code or if I need to install anything additional.
It seems that in
/semi_supervised_segmentation/losses.py
line 195, it should bece_loss = self.ce(net_output, target.long()) if self.weight_ce != 0 else 0
instead ofce_loss = self.ce(net_output, target[:, 0].long()) if self.weight_ce != 0 else 0
? Because the shape of net_output id(64, 2, 64, 64)
, and the shape of target.long() is(64, 64, 64)