search

bubbliiiing / unet-pytorch

这是一个unet-pytorch的源码，可以训练自己的模型
MIT License
1.39k stars 266 forks source link

泡泡哥请问一下制作数据集的问题 #17

Open yidimopozhibai opened 2 years ago

yidimopozhibai commented 2 years ago

可以看到视频用labelme标出来的是彩色图，猫狗分别是红色和绿色，但是训练的时候图又变成了二值图，中间是不是缺了什么流程！我用3060跑上面说的猫狗红色和绿色的数据集根本用不了，哭死

bubbliiiing commented 2 years ago

可以用的……你看注释

yidimopozhibai commented 2 years ago

可以用了 1 ， 我之前一直以为是数据及的问题，其实是这里，这个地方的numclass要改，不然没有办法用。

yidimopozhibai commented 2 years ago

顺便请问一下泡泡哥，如果我想换掉VGG这个backbone，大致需要修改unet里面的那些文件呢，非常感谢泡泡哥（视频次次都三连的

bubbliiiing commented 2 years ago

不是有resnet吗