运行test.py报错

[hopesala]

下载了训练好的yolov3_ckpt_195.pth 修改了yolov3-custom.cfg里面 [yolo]节点下面的classes=2 不然int（空字符串的时候）报错 之后运行加载权重文件报错

Namespace(batch_size=8, class_path='data/custom/classes.names', conf_thres=0.001, data_config='config/custom.data', img_size=416, iou_thres=0.5, model_def='config/yolov3-custom.cfg', n_cpu=8, nms_thres=0.5, weights_path='checkpoints/yolov3_ckpt_195.pth') Traceback (most recent call last): File "/Applications/PyCharm.app/Contents/helpers/pydev/pydevd.py", line 1741, in main() File "/Applications/PyCharm.app/Contents/helpers/pydev/pydevd.py", line 1735, in main globals = debugger.run(setup['file'], None, None, is_module) File "/Applications/PyCharm.app/Contents/helpers/pydev/pydevd.py", line 1135, in run pydev_imports.execfile(file, globals, locals) # execute the script File "/Applications/PyCharm.app/Contents/helpers/pydev/_pydev_imps/_pydev_execfile.py", line 18, in execfile exec(compile(contents+"\n", file, 'exec'), glob, loc) File "/Users/caochenghua/Documents/GitHub/Pytorch-Yolov3---Remote-sensing-image-master/test.py", line 83, in model = Darknet(opt.model_def).to(device) File "/Users/caochenghua/Documents/GitHub/Pytorch-Yolov3---Remote-sensing-image-master/models.py", line 241, in init self.yolo_layers = [layer[0] for layer in self.module_list if hasattr(layer[0], "metrics")] File "/Users/caochenghua/Documents/GitHub/Pytorch-Yolov3---Remote-sensing-image-master/models.py", line 241, in self.yolo_layers = [layer[0] for layer in self.module_list if hasattr(layer[0], "metrics")] File "/anaconda3/lib/python3.7/site-packages/torch/nn/modules/container.py", line 68, in getitem return self._get_item_by_idx(self._modules.values(), idx) File "/anaconda3/lib/python3.7/site-packages/torch/nn/modules/container.py", line 60, in _get_item_by_idx raise IndexError('index {} is out of range'.format(idx)) IndexError: index 0 is out of range 请问这是什么原因呢？

[zigangzhao-ai]

可以把之前的config文件删掉，自己再重新生成一下试试