jingtianyilong / YOLOv4-pytorch

This is a pytorch repository of YOLOv4 that can be trained with custom dataset.
31 stars 12 forks source link

在模型的验证中出现了以下问题,应怎样解决呢? 谢谢!ValueError: only one element tensors can be converted to Python scalars #11

Open lifafang opened 3 years ago

lifafang commented 3 years ago

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

lifafang commented 3 years ago

@jingtianyilong @

kimsjpk1 commented 2 years ago

you should validation parameter: batch size is 1. change validation batch size parameter