python目标检测给图画框,bbox画到图上并保存案例


Posted in Python onMarch 10, 2020

我就废话不多说了,还是直接上代码吧!

import os
import xml.dom.minidom
import cv2 as cv
 
ImgPath = 'C:/Users/49691/Desktop/gangjin/gangjin_test/JPEGImages/'
AnnoPath = 'C:/Users/49691/Desktop/gangjin/gangjin_test/Annotations/' #xml文件地址
save_path = ''
def draw_anchor(ImgPath,AnnoPath,save_path):
  imagelist = os.listdir(ImgPath)
  for image in imagelist:
 
    image_pre, ext = os.path.splitext(image)
    imgfile = ImgPath + image
    xmlfile = AnnoPath + image_pre + '.xml'
    # print(image)
    # 打开xml文档
    DOMTree = xml.dom.minidom.parse(xmlfile)
    # 得到文档元素对象
    collection = DOMTree.documentElement
    # 读取图片
    img = cv.imread(imgfile)
 
    filenamelist = collection.getElementsByTagName("filename")
    filename = filenamelist[0].childNodes[0].data
    print(filename)
    # 得到标签名为object的信息
    objectlist = collection.getElementsByTagName("object")
 
    for objects in objectlist:
      # 每个object中得到子标签名为name的信息
      namelist = objects.getElementsByTagName('name')
      # 通过此语句得到具体的某个name的值
      objectname = namelist[0].childNodes[0].data
 
      bndbox = objects.getElementsByTagName('bndbox')
      # print(bndbox)
      for box in bndbox:
        x1_list = box.getElementsByTagName('xmin')
        x1 = int(x1_list[0].childNodes[0].data)
        y1_list = box.getElementsByTagName('ymin')
        y1 = int(y1_list[0].childNodes[0].data)
        x2_list = box.getElementsByTagName('xmax')  #注意坐标,看是否需要转换
        x2 = int(x2_list[0].childNodes[0].data)
        y2_list = box.getElementsByTagName('ymax')
        y2 = int(y2_list[0].childNodes[0].data)
        cv.rectangle(img, (x1, y1), (x2, y2), (255, 255, 255), thickness=2)
        cv.putText(img, objectname, (x1, y1), cv.FONT_HERSHEY_COMPLEX, 0.7, (0, 255, 0),
              thickness=2)
        # cv.imshow('head', img)
        cv.imwrite(save_path+'/'+filename, img)  #save picture

补充知识:深度学习python之用Faster-rcnn 检测结果(txt文件) 在原图画出box

使用Faster-rcnn 的test_net.py 检测网络的mAP等精度会生成一个检测结果(txt文件),格式如下:

000004 0.972 302.8 94.5 512.0 150.0
000004 0.950 348.1 166.1 512.0 242.9
000004 0.875 1.0 25.7 292.6 126.3
000004 0.730 1.0 138.5 488.3 230.0
000004 0.699 1.0 120.9 145.5 139.9
000004 0.592 54.4 227.4 431.9 343.4
000004 0.588 1.0 159.8 18.8 231.6
000004 0.126 1.0 247.1 342.3 270.0
000004 0.120 1.0 225.4 185.7 309.3

每行分别为 名称 检测概率 xmin ymin xmax ymax

问题在于每一行只显示一个box数据,每幅图像可能包括多个box,需要判断提取的多行数据是不是属于同一图片

下面使用python提取这些数据,在原图上画出box并且保存起来

import os
import os.path
import numpy as np
import xml.etree.ElementTree as xmlET
from PIL import Image, ImageDraw
import cPickle as pickle 

txt_name = 'comp4_8a226fd7-753d-40fc-8013-f68d2a465579_det_test_ship.txt'
file_path_img = '/home/JPEGImages'
save_file_path = '/home/detect_results'


source_file = open(txt_name)

img_names = []
for line in source_file:
  staff = line.split()
  img_name = staff[0]
  img_names.append(img_name)

name_dict = {}
for i in img_names:
  if img_names.count(i)>0:
    name_dict[i] = img_names.count(i) 

source_file.close()

source_file = open(txt_name)
for idx in name_dict:
  img = Image.open(os.path.join(file_path_img, idx + '.jpg')) 
  draw = ImageDraw.Draw(img)
  for i in xrange(name_dict[idx]):
    line = source_file.readline()
    staff = line.split()
    score = staff[1]
    box = staff[2:6]
    draw.rectangle([int(np.round(float(box[0]))), int(np.round(float(box[1]))), 
          int(np.round(float(box[2]))), int(np.round(float(box[3])))], outline=(255, 0, 0))
  img.save(os.path.join(save_file_path, idx + '.jpg')) 

source_file.close()

运行完即可在保存文件夹中得到效果图。

以上这篇python目标检测给图画框,bbox画到图上并保存案例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python字符串格式化
Jun 15 Python
在Python中使用正则表达式的方法
Aug 13 Python
Python简单获取自身外网IP的方法
Sep 18 Python
详解Python中的相对导入和绝对导入
Jan 06 Python
python list元素为tuple时的排序方法
Apr 18 Python
使用Python进行QQ批量登录的实例代码
Jun 11 Python
python进程和线程用法知识点总结
May 28 Python
pytorch模型预测结果与ndarray互转方式
Jan 15 Python
Django重设Admin密码过程解析
Feb 10 Python
python 项目目录结构设置
Feb 14 Python
Python双链表原理与实现方法详解
Feb 22 Python
Python3.8.2安装包及安装教程图文详解(附安装包)
Nov 28 Python
python opencv 检测移动物体并截图保存实例
Mar 10 #Python
Python标准库json模块和pickle模块使用详解
Mar 10 #Python
Python xlrd excel文件操作代码实例
Mar 10 #Python
python图片剪裁代码(图片按四个点坐标剪裁)
Mar 10 #Python
python 截取XML中bndbox的坐标中的图像,另存为jpg的实例
Mar 10 #Python
Python代码一键转Jar包及Java调用Python新姿势
Mar 10 #Python
Python读取VOC中的xml目标框实例
Mar 10 #Python
You might like
php 图片上传类代码
2009/07/17 PHP
php下过滤html代码的函数 提高程序安全性
2010/03/02 PHP
用PHP实现递归循环每一个目录
2010/08/08 PHP
Php Ctemplate引擎开发相关内容
2012/03/03 PHP
PHP把小数转成整数3种方法
2014/06/30 PHP
函数中使用require_once问题深入探讨 优雅的配置文件定义方法推荐
2014/07/02 PHP
安装ImageMagick出现error while loading shared libraries的解决方法
2014/09/23 PHP
浅谈PHP错误类型及屏蔽方法
2017/05/27 PHP
PHP实现双链表删除与插入节点的方法示例
2017/11/11 PHP
PHP递归实现汉诺塔问题的方法示例
2017/11/25 PHP
tp5(thinkPHP5)框架实现多数据库查询的方法
2019/01/10 PHP
tp5框架基于Ajax实现列表无刷新排序功能示例
2020/02/10 PHP
js 页面输出值
2008/11/30 Javascript
jQuery 性能优化指南(3)
2009/05/21 Javascript
Javascript 日期对象Date扩展方法
2009/05/30 Javascript
输入密码检测大写是否锁定js实现代码
2012/12/03 Javascript
jquery获取tr中控件值并操作tr实现思路
2013/03/27 Javascript
JQuery onload、ready概念介绍及使用方法
2013/04/27 Javascript
JavaScript加强之自定义event事件
2013/09/21 Javascript
使用jquery实现鼠标滑过弹出更多相关信息层附源码下载
2015/11/23 Javascript
Vue.js动态添加、删除选题的实例代码
2016/09/30 Javascript
深入理解Vue生命周期、手动挂载及挂载子组件
2017/09/27 Javascript
微信小程序wx:for循环的实例详解
2018/10/07 Javascript
基于vue.js实现分页查询功能
2018/12/29 Javascript
js Array.slice的8种不同用法示例
2019/07/10 Javascript
layui 解决富文本框form表单提交为空的问题
2019/10/26 Javascript
[01:14]英雄,所敬略同——2018完美盛典宣传视频4K
2018/12/05 DOTA
windows下python模拟鼠标点击和键盘输示例
2014/02/28 Python
Python中使用dom模块生成XML文件示例
2015/04/05 Python
Python引用类型和值类型的区别与使用解析
2017/10/17 Python
python web基础之加载静态文件实例
2018/03/20 Python
HTML5 CSS3给网站设计带来出色效果
2009/07/16 HTML / CSS
销售总监岗位职责
2014/01/04 职场文书
学校花圃的标语
2014/06/18 职场文书
机电系毕业生求职信
2014/07/11 职场文书
Windows Server 2012 修改远程默认端口3389的方法
2022/04/28 Servers