问题描述:在跑Faster R-CNN时,使用单GPU训练,计算mAP时,报错:
logs/logs_1/ep150-loss1.017-val_loss1.106.pth model, anchors, and classes loaded.
0%| | 0/4952 [00:00<?, ?it/s]Load model done.
Get predict result.
0%| | 0/4952 [00:00<?, ?it/s]
Traceback (most recent call last):
File "/home/PycharmProjects/compare-programs/faster-rcnn-pytorch-master/get_map.py", line 76, in <module>
frcnn.get_map_txt(image_id, image, class_names, map_out_path)
File "/home/PycharmProjects/compare-programs/faster-rcnn-pytorch-master/frcnn.py", line 278, in get_map_txt
roi_cls_locs, roi_scores, rois, _ = self.net(images)
File "/home/.conda/envs/ultralytics_yolov3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 722, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/.conda/envs/ultralytics_yolov3/lib/py