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日

悬赏问题

  • ¥20 西门子S7-Graph,S7-300,梯形图
  • ¥50 用易语言http 访问不了网页
  • ¥50 safari浏览器fetch提交数据后数据丢失问题
  • ¥15 matlab不知道怎么改,求解答!!
  • ¥15 永磁直线电机的电流环pi调不出来
  • ¥15 用stata实现聚类的代码
  • ¥15 请问paddlehub能支持移动端开发吗?在Android studio上该如何部署?
  • ¥20 docker里部署springboot项目,访问不到扬声器
  • ¥15 netty整合springboot之后自动重连失效
  • ¥15 悬赏!微信开发者工具报错,求帮改