换成自己的数据集运行predict.py结果是全黑的

bubbliiiing / hrnet-pytorch

这是一个hrnet-pytorch的库，可用于训练自己的语义分割数据集

MIT License

104 stars 16 forks source link

Open gaozimeng21 opened 10 months ago

gaozimeng21 commented 10 months ago

代码结构没改，只改了必要的参数，为什么数据集格式正确，训练了10个epoch后再运行predict.py，结果图片没有进行分割，是全黑的，是不是网络本身或者predict.py文件有问题呢？

Refresh-today commented 1 month ago

你解决了吗？我运行原版代码时候也出现这问题了