yolov3输出坐标,yolov5输出预测框的坐标

  yolov3输出坐标,yolov5输出预测框的坐标

  这篇文章主要给大家介绍了关于yolov5返回坐标的相关资料,文中通过实例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下

  

目录
约洛夫5返回坐标(v6版)附参考:yolov5输出检测到的目标坐标信息(旧版本)总结

  

yolov5返回坐标(v6版)

  1 、从yolov5文件夹李找到detect.py,按打开输入annotator.box _ label

  如果保存_img或保存_裁剪或view_img: #将视频添加到图像

  c=int(cls) #整数类

  label=None if hide _ labels else(names[c]if hide _ conf else f { names[c]} { conf :2 f } )

  annotator.box_label(xyxy,label,color=colors(c,True))

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

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

  打印(左上点的坐标为:( 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)):

  #向带有标签的图像添加一个xyxy框

  如果self.pil是否为_ascii(标签):

  self.draw.rectangle(box,width=self.lw,outline=color) # box

  如果标签:

  w,h=self.font.getsize(label) #文本宽度,高度

  outside=box[1] - h=0 #标签适合盒子外部

  self.draw.rectangle([box[0],

  框[1] - h如果在框[1]之外,

  box[0] w 1,

  box[1]1 if else box[1]h 1],fill=color)

  # self.draw.text((box[0],box[1]),label,fill=txt_color,font=self.font,anchor=ls) #对于PIL8.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]))

  打印(左上点的坐标为:( 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

郑重声明:本文由网友发布,不代表盛行IT的观点,版权归原作者所有,仅为传播更多信息之目的,如有侵权请联系,我们将第一时间修改或删除,多谢。

留言与评论(共有 条评论)
   
验证码: