yolov5返回坐标的方法实例


Posted in Python onMarch 17, 2022

yolov5返回坐标(v6版)

1 、从yolov5文件夹李找到detect.py,按Ctrl+F 输入annotator.box_label;

if save_img or save_crop or view_img:  # Add bbox to image
                        c = int(cls)  # integer class
                        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))

2、找到这个代码后按住ctrl键,鼠标点击box_label,就会跳到plots.py文件并定位到box_label定义的地方;

3、找到p1, p2 = (int(box[0]), int(box[1])), (int(box[2]), int(box[3])),在这行代码下面新增:

print("左上点的坐标为:(" + str(p1[0]) + "," + str(p1[1]) + "),右下点的坐标为(" + str(p2[0]) + "," + str(p2[1]) + ")")

4、完成后的代码如下:

def box_label(self, box, label='', color=(128, 128, 128), txt_color=(255, 255, 255)):
        # Add one xyxy box to image with label
        if self.pil or not is_ascii(label):
            self.draw.rectangle(box, width=self.lw, outline=color)  # box
            if label:
                w, h = self.font.getsize(label)  # text width, height
                outside = box[1] - h >= 0  # label fits outside box
                self.draw.rectangle([box[0],
                                     box[1] - h if outside else box[1],
                                     box[0] + w + 1,
                                     box[1] + 1 if outside else box[1] + h + 1], fill=color)
                # self.draw.text((box[0], box[1]), label, fill=txt_color, font=self.font, anchor='ls')  # for PIL>8.0
                self.draw.text((box[0], box[1] - h if outside else box[1]), label, fill=txt_color, font=self.font)
        else:  # cv2
            p1, p2 = (int(box[0]), int(box[1])), (int(box[2]), int(box[3]))
            print("左上点的坐标为:(" + str(p1[0]) + "," + str(p1[1]) + "),右下点的坐标为(" + str(p2[0]) + "," + str(p2[1]) + ")")
            
            cv2.rectangle(self.im, p1, p2, color, thickness=self.lw, lineType=cv2.LINE_AA)

5、测试情况:回到命令行,cd到yolov5文件夹,输入指令:python detect.py --source ../mask.1.jpg,其中mask.1.jpg应改为你yolov5文件夹下的图片名称;按回车键后运行就发现输出的信息多了刚才添加的一行

(venv) (base) rongxiao@rongxiao:~/PycharmProjects/yolococo/yolov5$ python detect.py --source ../mask.1.jpg
detect: weights=yolov5s.pt, source=../mask.1.jpg, 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
YOLOv5 ? v6.0-147-g628817d torch 1.8.2+cpu CPU

Fusing layers... 
Model Summary: 213 layers, 7225885 parameters, 0 gradients
左上点的坐标为:(982,384),右下点的坐标为(1445,767)
左上点的坐标为:(724,237),右下点的坐标为(770,277)
左上点的坐标为:(711,226),右下点的坐标为(1689,938)
image 1/1 /home/rongxiao/PycharmProjects/yolococo/mask.1.jpg: 384x640 2 persons, 1 airplane, Done. (0.182s)
Speed: 1.1ms pre-process, 181.7ms inference, 1.0ms NMS per image at shape (1, 3, 640, 640)
Results saved to runs/detect/exp15

附参考:yolov5输出检测到的目标坐标信息(旧版本)

找到detect.py,在大概113行,找到plot_one_box

                # Write results
                for *xyxy, conf, cls in reversed(det):
                    if save_txt:  # Write to file
                        xywh = (xyxy2xywh(torch.tensor(xyxy).view(1, 4)) / gn).view(-1).tolist()  # normalized xywh
                        with open(txt_path + '.txt', 'a') as f:
                            f.write(('%g ' * 5 + '\n') % (cls, *xywh))  # label format

                    if save_img or view_img:  # Add bbox to image
                        label = '%s %.2f' % (names[int(cls)], conf)
                        plot_one_box(xyxy, im0, label=label, color=colors[int(cls)], line_thickness=3)

ctr+鼠标点击,进入general.py,并自动定位到plot_one_box函数,修改函数为

def plot_one_box(x, img, color=None, label=None, line_thickness=None):
    # Plots one bounding box on image img
    tl = line_thickness or round(0.002 * (img.shape[0] + img.shape[1]) / 2) + 1  # line/font thickness
    color = color or [random.randint(0, 255) for _ in range(3)]
    c1, c2 = (int(x[0]), int(x[1])), (int(x[2]), int(x[3]))
    cv2.rectangle(img, c1, c2, color, thickness=tl, lineType=cv2.LINE_AA)
    print("左上点的坐标为:(" + str(c1[0]) + "," + str(c1[1]) + "),右下点的坐标为(" + str(c2[0]) + "," + str(c2[1]) + ")")

即可输出目标坐标信息了

yolov5返回坐标的方法实例

总结

到此这篇关于yolov5返回坐标的文章就介绍到这了,更多相关yolov5返回坐标内容请搜索三水点靠木以前的文章或继续浏览下面的相关文章希望大家以后多多支持三水点靠木!

Python 相关文章推荐
python正则表达式re模块详解
Jun 25 Python
Python中operator模块的操作符使用示例总结
Jun 28 Python
win系统下为Python3.5安装flask-mongoengine 库
Dec 20 Python
利用标准库fractions模块让Python支持分数类型的方法详解
Aug 11 Python
在Python web中实现验证码图片代码分享
Nov 09 Python
一些Centos Python 生产环境的部署命令(推荐)
May 07 Python
分析经典Python开发工程师面试题
Apr 08 Python
Python: 传递列表副本方式
Dec 19 Python
Pytorch 神经网络—自定义数据集上实现教程
Jan 07 Python
Python3中configparser模块读写ini文件并解析配置的用法详解
Feb 18 Python
python由已知数组快速生成新数组的方法
Apr 08 Python
Python网络爬虫四大选择器用法原理总结
Jun 01 Python
PyTorch中的torch.cat简单介绍
Mar 17 #Python
Python Pygame实战在打砖块游戏的实现
python超详细实现完整学生成绩管理系统
Mar 17 #Python
Python Pygame实战之塔防游戏的实现
pytorch中的 .view()函数的用法介绍
Mar 17 #Python
Python绘画好看的星空图
opencv深入浅出了解机器学习和深度学习
You might like
国内咖啡文化
2021/03/03 咖啡文化
PHP字符转义相关函数小结(php下的转义字符串)
2007/04/12 PHP
PHP Pear 安装及使用
2009/03/19 PHP
php中实现记住密码自动登录的代码
2011/03/02 PHP
php使用gd2绘制基本图形示例(直线、圆、正方形)
2017/02/15 PHP
php实现微信企业付款到个人零钱功能
2018/10/09 PHP
PHP随机数函数rand()与mt_rand()的讲解
2019/03/25 PHP
phpQuery采集网页实现代码实例
2020/04/02 PHP
新增加的内容是如何将div的scrollbar自动移动最下面
2014/01/02 Javascript
jquery、js调用iframe父窗口与子窗口元素的方法整理
2014/07/31 Javascript
javascript学习笔记(一)基础知识
2014/09/30 Javascript
javascript实现倒计时N秒后网页自动跳转代码
2014/12/11 Javascript
关于Javascript加载执行优化的研究报告
2014/12/16 Javascript
jQuery中:eq()选择器用法实例
2014/12/29 Javascript
基于jQuery实现交互体验社会化分享代码附源码下载
2016/01/04 Javascript
vue-cli中的babel配置文件.babelrc实例详解
2018/02/22 Javascript
nodejs使用async模块同步执行的方法
2019/03/02 NodeJs
Vue+Koa2+mongoose写一个像素绘板的实现方法
2019/09/10 Javascript
vue3.0 项目搭建和使用流程
2021/03/04 Vue.js
python检测远程udp端口是否打开的方法
2015/03/14 Python
详解Python的Django框架中inclusion_tag的使用
2015/07/21 Python
python pandas 对series和dataframe的重置索引reindex方法
2018/06/07 Python
详解python中自定义超时异常的几种方法
2019/07/29 Python
python实现输入的数据在地图上生成热力图效果
2019/12/06 Python
Python ORM编程基础示例
2020/02/02 Python
python3.8.3安装教程及环境配置的详细教程(64-bit)
2020/11/28 Python
利用CSS3实现平移动画效果示例代码
2016/10/12 HTML / CSS
YSL圣罗兰美妆英国官网:Yves Saint Laurent Beauty UK
2019/08/03 全球购物
精彩的大学生自我评价
2013/11/17 职场文书
教师求职自荐信
2014/03/09 职场文书
公司年终奖分配方案
2014/06/16 职场文书
超市督导岗位职责
2015/04/10 职场文书
基石观后感
2015/06/12 职场文书
2016年暑期教师培训心得体会
2016/01/09 职场文书
Nginx四层负载均衡的配置指南
2021/06/11 Servers
Windows Server 2016 配置 IIS 的详细步骤
2022/04/28 Servers