轻松搭建Yolov5 GPU运行环境

  • 运行测试命令
(yolov5) D:\ptwork\git\yolov5>python detect.py
detect: weights=yolov5s.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
Parse error at "'-i https'": Expected W:(abcd...)
YOLOv5  v6.1-207-g5774a15 Python-3.9.0 torch-1.11.0 CUDA:0 (NVIDIA GeForce MX350, 2048MiB)
Downloading https://github.com/ultralytics/yolov5/releases/download/v6.1/yolov5s.pt to yolov5s.pt...

第一次运行会下载yolov5s.pt文件,下载可能会很慢,可以自行下载,放置在 yolov5/文件夹下即可,也可以创建一个权重文件夹, yolov5/weights/yolov5s.pt(git上还有其他预训练好的模型,可以自行下载)。 可以使用如下命令

(yolov5) D:\ptwork\git\yolov5>python detect.py --weights weights/yolov5s.pt
detect: weights=['weights/yolov5s.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
Parse error at "'-i https'": Expected W:(abcd...)
YOLOv5  v6.1-207-g5774a15 Python-3.9.0 torch-1.11.0 CUDA:0 (NVIDIA GeForce MX350, 2048MiB)

Fusing layers...

YOLOv5s summary: 213 layers, 7225885 parameters, 0 gradients
image 1/2 D:\ptwork\git\yolov5\data\images\bus.jpg: 640x480 4 persons, 1 bus, Done. (0.034s)
image 2/2 D:\ptwork\git\yolov5\data\images\zidane.jpg: 384x640 2 persons, 2 ties, Done. (0.025s)
Speed: 1.5ms pre-process, 29.5ms inference, 3.5ms NMS per image at shape (1, 3, 640, 640)
Results saved to runs\detect\exp2

运行的结果在 runs\detect\exp2中。
在这里插入图片描述

轻松搭建Yolov5 GPU运行环境

Original: https://blog.csdn.net/bbs165198646/article/details/124901156
Author: 家有一亩三分地
Title: 轻松搭建Yolov5 GPU运行环境

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/710697/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球