pytorch-nested-unet icon indicating copy to clipboard operation
pytorch-nested-unet copied to clipboard

训练自己的数据集疑问

Open wuzuowuyou opened this issue 4 years ago • 4 comments

是每个类别一个文件夹,每个类别一张图,该类别区域是255,其他区域是0

比如我有字母ABC三类,那么我需要3个文件夹,比如一张原图上面有AB两个字母,那么0文件夹的标签图是上只有A二值图,1文件夹标签图是上只有B二值图

是这样子吗

wuzuowuyou avatar May 14 '20 04:05 wuzuowuyou

是同一个图片上不同区域的mask,用来做多类别分割的。比如一个图片上又有人又有车,人和车分别放在不同的文件夹里面。具体你可以看看代码

molly00ecla avatar Sep 15 '20 06:09 molly00ecla

@wuzuowuyou 大佬搞明白这个问题了吗,我感觉二楼把我说糊涂了。。。

CangHaiQingYue avatar Sep 30 '20 07:09 CangHaiQingYue

@wuzuowuyou 请问楼上这个问题解决了吗?同款疑问。我是按照你这样去分类的,但是预测结果不对,iou为0

snowyseaons avatar Aug 18 '21 01:08 snowyseaons

如你有5个类,那么mask文件夹里面需要有0,1,2,3,4五个文件夹,然后第一类的黑白两色mask文件存在0文件夹里面,第二类的黑白两色mask文件存在1文件夹里面,如此类推。另外要注意的是,算图片里没有的类(如第一类),你也需要生成一张全黑的mask放在0文件夹里面,其他文件夹也是如此。

x55901 avatar Sep 14 '21 08:09 x55901