bubbliiiing/unet-pytorch

关于非VOC数据集(医药数据集)标签格式问题

tsieyy opened this issue · 0 comments

作者在 train_medical.py 的注释中说明标签需要将背景像素值设置为0,类别设置为1、2、3.... 等等,我按照作者所说设置之后训练出来就是全黑(predict画出来的为全红),后来尝试了仓库中原本存在的数据集标签的格式,效果良好。即把要分割的目标的像素点设置成0,背景像素值设置为255,标签图最终效果为:目标值为黑色,背景为白色。
多类型分割没有尝试过。