Val batch: 0%| | 0/12 [00:06<?, ?imgs/s]
Traceback (most recent call last):
File "/home/桌面/YOLOv4-pytorch/train.py", line 260, in
Trainer(log_dir,resume= args.resume).train()
File "/home/桌面/YOLOv4-pytorch/train.py", line 171, in train
coco_stat = evaluator.evaluate(self.yolov4)
File "/home/桌面/YOLOv4-pytorch/eval/cocoapi_evaluator.py", line 61, in evaluate
info_img = [float(info) for info in info_img]
File "/home/桌面/YOLOv4-pytorch/eval/cocoapi_evaluator.py", line 61, in
info_img = [float(info) for info in info_img]
ValueError: only one element tensors can be converted to Python scalars
Val batch: 0%| | 0/12 [00:06<?, ?imgs/s] Traceback (most recent call last): File "/home/桌面/YOLOv4-pytorch/train.py", line 260, in
Trainer(log_dir,resume= args.resume).train()
File "/home/桌面/YOLOv4-pytorch/train.py", line 171, in train
coco_stat = evaluator.evaluate(self.yolov4)
File "/home/桌面/YOLOv4-pytorch/eval/cocoapi_evaluator.py", line 61, in evaluate
info_img = [float(info) for info in info_img]
File "/home/桌面/YOLOv4-pytorch/eval/cocoapi_evaluator.py", line 61, in
info_img = [float(info) for info in info_img]
ValueError: only one element tensors can be converted to Python scalars