pytorch 训练maskrcnn 自定义数据集出错

问题描述:
RuntimeError: copy_if failed to synchronize: device-side assert triggered

这个问题是我在使用MaskRcnn做实例分割时遇到的,我要检测的目标有5种类别,所以我在data/config.py中的num_classes参数写了5,经过多方查找,发现了一个没注意到的细节,类别应该是5+1,那个1应该是背景。

还有一个原因就是标签的标号没有从0开始。

我的是两个都遇到了.

猜你喜欢

转载自blog.csdn.net/phker/article/details/109576341