最近, 训练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中类别输入有误, 一个个对应检查好。立贴惊醒自己 , 一个小错误花了我半天时间。