yolov5只检测单一类别或者特定的类

最新推荐文章于 2024-08-16 17:37:33 发布

雪可问春风

最新推荐文章于 2024-08-16 17:37:33 发布

阅读量7.9k

点赞数 5

本文链接：https://blog.csdn.net/BruceBorgia/article/details/123103804

版权

深度学习同时被 2 个专栏收录

18 篇文章 1 订阅

订阅专栏

目标检测

5 篇文章 0 订阅

订阅专栏

在detect.py的第170行，探讨了如何将检测到的边界框添加到图片中，针对类别‘name_sign’，进行了隐藏标签处理，并提供了裁剪和保存选项。代码详细展示了如何根据分类创建标签和进行相应的图像操作。

摘要由CSDN通过智能技术生成

在detect.py中
170行

                    if save_img or save_crop or view_img:  # Add bbox to image
                        c = int(cls)  # integer class
                        # FIXME:
                        if names[int(cls)] == "name_sign":
                        		label = None if hide_labels else (names[c] if hide_conf else f'{names[c]} {conf:.2f}')
                        		annotator.box_label(xyxy, label, color=colors(c, True),counta = counta)
                        		if save_crop:
                           				 save_one_box(xyxy, imc, file=save_dir / 'crops' / names[c] / f'{p.stem}.jpg', BGR=True)