pytorch图像分割图像输入与loss选择

做的是医学图像分割,图像如图所示,左边为原图,右边为标签,目前所采用的是对dataset写子类的方法输入图像,图像除了尺寸变化和转变tensor外未做其他改变。loss函数用的是BCELoss和MSELoss,效果都不好,看网上教程用NLLLoss2d,但是一用就报错,应该是数据输入格式不对,用这个loss应该图像格式应该怎样呢?希望能有大神详细解答一下
已邀请:

brisyramshere - 90后,图像处理

赞同来自:

同样关注这个问题,关于医学图像分割的loss怎么选择。
我用的u-net,输入输出是相同尺寸的矩阵,其输入输出格式都是FloatTensor。
不知道楼主用的是什么网络进行的图像分割?

要回复问题请先登录注册