Pan da da 2023-03-30 22:09 采纳率: 60%
浏览 23
已结题

yolov5 分割训练detect.py有错误得不到结果

yolov5 分割标记训练,训练很顺利但之后把权重加入到detect.py时就出现了以下错误,根本得不到结果,有没有哪位大佬能帮帮我,真的十分感谢!

img

"D:\python huangjing\anoconda\envs\yolo5\python.exe" "D:\python huangjing\yolov5\yolov5-master\detect.py" 
detect: weights=D:\python huangjing\yolov5\yolov5-master\exp14\weights\best.pt, source=data\images, data=data\coco128.yaml, imgsz=[640, 640], conf_thres=0.25, iou_thres=0.45, max_det=1000, device=, view_img=False, save_txt=False, save_conf=False, save_crop=False, nosave=False, classes=None, agnostic_nms=False, augment=False, visualize=False, update=False, project=runs\detect, name=exp, exist_ok=False, line_thickness=3, hide_labels=False, hide_conf=False, half=False, dnn=False, vid_stride=1
YOLOv5  2023-3-10 Python-3.8.0 torch-1.11.0+cu113 CUDA:0 (NVIDIA GeForce GTX 1650, 4096MiB)

Fusing layers... 
Model summary: 165 layers, 7401119 parameters, 0 gradients, 25.7 GFLOPs
Traceback (most recent call last):
  File "D:\python huangjing\yolov5\yolov5-master\detect.py", line 261, in <module>
    main(opt)
  File "D:\python huangjing\yolov5\yolov5-master\detect.py", line 256, in main
    run(**vars(opt))
  File "D:\python huangjing\anoconda\envs\yolo5\lib\site-packages\torch\autograd\grad_mode.py", line 27, in decorate_context
    return func(*args, **kwargs)
  File "D:\python huangjing\yolov5\yolov5-master\detect.py", line 160, in run
    s += f"{n} {names[int(c)]}{'s' * (n > 1)}, "  # add to string
KeyError: 2

进程已结束,退出代码1


  • 写回答

1条回答 默认 最新

  • 爱晚乏客游 2023-03-31 09:22
    关注

    分割运行segment/predict.py这个文件而不是detect.py

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

报告相同问题?

问题事件

  • 系统已结题 4月8日
  • 已采纳回答 3月31日
  • 创建了问题 3月30日

悬赏问题

  • ¥15 AttributeError: 'NoneType' object has no attribute 'drop_duplicates'报错
  • ¥15 编写c++计算器,用类和栈来做,要求如下
  • ¥15 以下代码,运行结果报错
  • ¥15 Vivado仿真数据出错
  • ¥15 银河麒麟不支持vant等前端框架怎么办?
  • ¥15 vue3的子组件在父页面调用不显示不生效问题
  • ¥15 cadence PEX
  • ¥15 phython创建了文件 显示在项目下面,但运行不了,运行时还是运行main文件,并且说main文件不允许并行运行
  • ¥20 C++哈希表的设计构造哈希表
  • ¥15 FutureWarning:不推荐使用空或全 NA 条目的 DataFrame 串联行为。怎么改呢?