Open flyingbird9 opened 1 year ago
谢谢,请问我运行pytorch版本遇到了这个问题,请问是什么原因呢
自己的数据还是? 另外torch的版本需要 torch==2.01
我用的自己的数据,但是在paddle下能跑,我用的cpu 版本,torch是2.0.1
自己的数据需要保证处理成coco格式,其中image_id需要是int类型,并且和图像文件名一一对应,否则使用torchvision.datasets.CocoDetection在按索引读取数据的时候会抛出异常。
另外,使用自己的数据注意一下这两个变量有没有正确设置,比如自定义数据集有7个非背景类别,那么num_classes需要设置为8(+背景类),remap_mscoco_category需要设置为False.
在代码的哪个部分呢