训练ssd300时网络报错

最近, 训练ssd300.运行源代码的时候,遇到各种各样的错误。其中一个如下:

img, boxes, labels = self.transform(img, target[:, :4], target[:, 4])
IndexError: too many indices for array: array is 1-dimensional, but 2 were indexed

查了发现是类别的原因,可能和0712.py中VOC_CLASSES中类别输入有误, 一个个对应检查好。立贴惊醒自己 , 一个小错误花了我半天时间。


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