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

关于训练自己的数据集

Open ShiMinghao0208 opened this issue 4 years ago • 4 comments

请问一下大佬,如果训练自己的数据集,是不是dataloader返回的jpg和png就够用了,labels是针对voc数据集的白边的,再用diceloss的时候把参数里的label换成png?

ShiMinghao0208 avatar Nov 05 '21 08:11 ShiMinghao0208

我没看懂诶

bubbliiiing avatar Nov 08 '21 16:11 bubbliiiing

大佬,我看你代码里面说:网上常见的数据集总共对输入图片分两类,背景的像素点值为0,目标的像素点值为255。这样的数据集可以正常运行但是预测是没有效果的! 需要改成,背景的像素点值为0,目标的像素点值为1。那么我的分类是十类的话,像素点的值设置成多少合适? ①我之前用的是0-255,predict出来是全黑的,所以还是在考虑voc格式问题。 ②没办法用labelme,因为我们的分类很密集,手动标记不太可能。 所以想问问这个具体要求是?

Yinghui-Lee avatar Dec 04 '21 10:12 Yinghui-Lee

大佬,我看你代码里面说:网上常见的数据集总共对输入图片分两类,背景的像素点值为0,目标的像素点值为255。这样的数据集可以正常运行但是预测是没有效果的! 需要改成,背景的像素点值为0,目标的像素点值为1。那么我的分类是十类的话,像素点的值设置成多少合适? ①我之前用的是0-255,predict出来是全黑的,所以还是在考虑voc格式问题。 ②没办法用labelme,因为我们的分类很密集,手动标记不太可能。 所以想问问这个具体要求是?

Yinghui-Lee avatar Dec 04 '21 10:12 Yinghui-Lee

你可以先看看数据集解析的视频……然后改成那个格式,你可以用os.listdir遍历标签,把值都设置成1

bubbliiiing avatar Dec 06 '21 15:12 bubbliiiing