YOLOv5实战记录02 模型检测

最新推荐文章于 2024-05-03 12:34:38 发布

原创

最新推荐文章于 2024-05-03 12:34:38 发布 · 548 阅读

5 ·

CC 4.0 BY-SA版权

文章标签：

#YOLO

本文介绍了如何使用YOLOv5进行图像检测，包括关键参数如置信度阈值和IOU阈值的解释，以及利用torch.hub简化模型加载和图像识别的过程。

本人记录打卡，不够自习，慎看。

今天主要学了计组和计网，YOLO简单打个卡。

指路大佬：【手把手带你实战YOLOv5-入门篇】YOLOv5 模型检测_哔哩哔哩_bilibili

1. 主要讲了几个关键参数：

图源你可是处女座

运行示例：

 python detect.py --weights yolov5s.pt   #--weights可以指定训练好的模型文件。

 python detect.py --weights yolov5s.pt --source data/images/bus.jpg
#source可以指定检测的目标，此处为指定单张图片。如果--source 0 即指定摄像头。 --source screen对整个屏幕进行检测。

python detect.py  --weights yolov5s.pt  --conf-thres 0.8
#--conf-thres代表置信度阈值，数值越低框越多。

python detect.py  --weights yolov5s.pt  --iou-thres 0.8
#