YOLOv5读取摄像头实时检测目标并将裁剪后的部分保存

一个来自三本院校的底层工人

已于 2023-01-11 09:50:47 修改

阅读量4.4k

点赞数 4

分类专栏：图像实例分割文章标签： python 开发语言

于 2022-10-21 19:37:10 首次发布

本文链接：https://blog.youkuaiyun.com/weixin_41837701/article/details/127452514

版权

图像实例分割专栏收录该内容

3 篇文章

订阅专栏

该博客介绍了如何利用Python处理USB相机实时视频流。首先，在datasets.py中注释掉与YouTube相关的代码，然后在detect.py中修改默认参数，使程序从相机（设备ID 0）读取视频。接下来，通过annotator.result()获取裁剪后的图片，并将其作为后续模型的输入。最后，更新了保存裁剪图片的代码，先进行裁剪操作再保存，确保保存的是裁剪后的图像。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1.输入usb相机实时拍摄视频

1.在datasets.py文件中，按住Ctrl+F查找www.youtubo.com，把这个if语句模块注释掉；

2.在detect.py文件中，将默认参数default改为0，parser.add_argument( ‘–source’, type=str, default=‘0’, help=‘’)

3.此时的im0 = annotator.result(）为裁剪后的图片。可以作为后续模型的输入使用。

2.保存裁剪图片

在detect.py文件中找到如下代码

if save_crop:
	save_one_box(xyxy, imc, file=save_dir / 'crops' / names[c] / f'{p.stem}.jpg', BGR=True)

改为如下代码：

if save_crop:
	crop_img = crop_one_box(xyxy, imc, BGR = True)
	save_one_box(xyxy, crop_img , file=save_dir / 'crops' / names[c] / f'{p.stem}.jpg', BGR=True)