【已解决】目标检测模型报错:Assertion `srcIndex < srcSelectDimSize` failed.
被困扰了很久,但后来发现这是一个很蠢的问题,原因是:dotav1数据集中有15个类别,而diorr中有20个,但config文件中不会自动转换类别数量,需要手动输入,于是。将config文件中的以下两项num_classes改成20,问题就解决了。当我使用ars-detr训练模型,将数据集从dotav1切换成diorr时,发生了类似于下面的错误。
原创
2025-04-04 15:30:35 ·
345 阅读 ·
0 评论