pytorch-nested-unet
pytorch-nested-unet copied to clipboard
训练自己的数据集疑问
是每个类别一个文件夹,每个类别一张图,该类别区域是255,其他区域是0
比如我有字母ABC三类,那么我需要3个文件夹,比如一张原图上面有AB两个字母,那么0文件夹的标签图是上只有A二值图,1文件夹标签图是上只有B二值图
是这样子吗
是同一个图片上不同区域的mask,用来做多类别分割的。比如一个图片上又有人又有车,人和车分别放在不同的文件夹里面。具体你可以看看代码
@wuzuowuyou 大佬搞明白这个问题了吗,我感觉二楼把我说糊涂了。。。
@wuzuowuyou 请问楼上这个问题解决了吗?同款疑问。我是按照你这样去分类的,但是预测结果不对,iou为0
如你有5个类,那么mask文件夹里面需要有0,1,2,3,4五个文件夹,然后第一类的黑白两色mask文件存在0文件夹里面,第二类的黑白两色mask文件存在1文件夹里面,如此类推。另外要注意的是,算图片里没有的类(如第一类),你也需要生成一张全黑的mask放在0文件夹里面,其他文件夹也是如此。