Unet复现:遇到 block: [0,0,0], thread: [594,0,0] Assertion `t >= 0 && t < n_classes` failed.题

  • Post author:
  • Post category:其他


复现参考链接:

http://t.csdn.cn/sAPq1

在训练自己用labelme标注的图片时遇到上面提到的问题,经过网上的分享总共分下面几种情况,但是都还是无法解决我遇到的问题,最后排查代码才找到问题。

1、原作者说的一种情况;

http://t.csdn.cn/sAPq1

2、 见链接:

http://t.csdn.cn/faMAi

3、见链接:

http://t.csdn.cn/lKUuf

4、我遇到的情况:发现无论怎么改n_classes的数值,运行结果中输出通道还是2,因此,直接将train.py中的n_classes修改为3,因为调用可能运行失败。

现在就运行成功了,测试当然也是一样的改动。



版权声明:本文为seeyouwlx原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。