python如何将txt坐标批量打印到原图上

 更新时间:2023年08月16日 16:50:23   作者:慕容洛凝  
这篇文章主要介绍了python如何将txt坐标批量打印到原图上的问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教

python将txt坐标批量打印到原图上

# -*- coding: utf-8 -*-
# liufangtao
#批量处理img和xml文件,根据xml文件中的坐标把img中的目标标记出来,并保存到指定文件夹。
import xml.etree.ElementTree as ET
import os, cv2
import numpy 
# from tqdm import tqdm
from PIL import Image, ImageDraw, ImageFont
annota_dir = r'\ann\atxt' #txt文件夹
origin_dir = r'\testjyz\ann\bimages'  #图片文件夹
target_dir1= r'\testjyz\ann\cresult'  #保存地址
class_dist = {0:'盘式绝缘子', 1:'棒式绝缘子', 2:'复合绝缘子', 
                        3:'柱上开关绝缘子', 4:'针式绝缘子', 5:'玻璃绝缘子',6:'悬式棒绝缘子'}
def divide_img(oriname):
    img_file = os.path.join(origin_dir, oriname + '.jpg')
    im = cv2.imread(img_file)
    h, w = im.shape[:2]
    # print(w)
    # 图像从OpenCV格式转换成PIL格式 
    img_PIL = Image.fromarray(cv2.cvtColor(im, cv2.COLOR_BGR2RGB))
    draw = ImageDraw.Draw(img_PIL)
    #字体路径和字体大小
    font = ImageFont.truetype('SimHei.ttf', 40) 
    xml_file = open(os.path.join(annota_dir, oriname + '.txt'))  # 读取每个原图像的xml文件
    # print(xml_file)
    for bbox in xml_file.readlines():
        id, x1, y1, x2, y2 = float(bbox.split(' ')[0]), float(bbox.split(' ')[1]), float(bbox.split(' ')[2]), float(bbox.split(' ')[3]), float(bbox.split(' ')[4])
        xmin = ((x1*2*w)-x2*w)/2
        xmax = ((x1*2*w)+x2*w)/2
        ymin = ((y1*2*h)-y2*h)/2
        ymax = ((y1*2*h)+y2*h)/2
        # print(xmin, ymin, xmax, ymax)
        # cv2.rectangle(image, (x1, y1), (x2, y2), (0, 0, 255), 2)
        # 在边界框的两点(左上角、右下角)画矩形,无填充,边框红色,边框像素为5
        draw.rectangle(((xmin, ymin), (xmax, ymax)), fill=None, outline='red', width=5)
        draw.text((xmin, ymin-40), class_dist[id], font=font, fill=(0, 255, 0))       
    # 转换回OpenCV格式 
    img_OpenCV = cv2.cvtColor(numpy.asarray(img_PIL),cv2.COLOR_RGB2BGR) 
    img_name = oriname + '.jpg'
    print(img_name)
    to_name = os.path.join(target_dir1, img_name)
    cv2.imwrite(to_name, img_OpenCV)
img_list = os.listdir(origin_dir)
for name in img_list:
    divide_img(name.rstrip('.jpg'))

python根据txt文件批量提取图片

Python代码:

根据txt文件批量提取图片并保存至另一文件夹,用于深度学习-图片数据预处理

import os
import shutil
# 根据 标签列表/txt文件 批量提取图片
name_list = open('val.txt', 'r')
# print(name_list)
# 图片路径
tu_dir = 'label'
# 保存路径
save = '/val/label'
dir_name = []
# 获取文件名
for i in name_list:
    dir_name.append(os.path.basename(i.replace('\n', '')))
# print(dir_name)
# 批量复制
for i in dir_name:
    shutil.copy(tu_dir + '/' + i, save + '/' + i)

总结

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

相关文章

  • python读取图像矩阵文件并转换为向量实例

    python读取图像矩阵文件并转换为向量实例

    这篇文章主要介绍了python读取图像矩阵文件并转换为向量实例,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2020-06-06
  • python爬虫今日热榜数据到txt文件的源码

    python爬虫今日热榜数据到txt文件的源码

    这篇文章主要介绍了python爬虫今日热榜数据到txt文件的源码,本文给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2021-02-02
  • 一文详解Python定时任务触发

    一文详解Python定时任务触发

    这篇文章主要为大家详细介绍了Python定时任务触发的相关知识,文中的示例代码讲解详细,对我们学习Python有一定的帮助,需要的可以参考一下
    2023-04-04
  • 教你如何使用Python selenium

    教你如何使用Python selenium

    今天教大家如何使用Python selenium,本文会以艺龙旅游网为对象,进行selenium的学习,目的:爬取艺龙网中南阳市唐河县的酒店信息,包括:名字,电话,标间价格,地址,介绍,图片,需要的朋友可以参考下
    2021-06-06
  • Python利用多线程枚举实现获取wifi信息

    Python利用多线程枚举实现获取wifi信息

    这篇文章主要为大家详细介绍了Python如何利用枚举字典的方式来实现获取wifi信息,文中的示例代码讲解详细,感兴趣的小伙伴可以了解一下
    2022-12-12
  • Python实战之设计一个多功能办公小工具

    Python实战之设计一个多功能办公小工具

    这篇文章主要介绍了通过Python制作一个多功能的办公小工具,可以实现应用、网页直达以及天气查询,代码具有一定学习价值,需要的小伙伴可以了解一下
    2021-12-12
  • python 寻找离散序列极值点的方法

    python 寻找离散序列极值点的方法

    今天小编就为大家分享一篇python 寻找离散序列极值点的方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2019-07-07
  • Python异常处理如何才能写得优雅(retrying模块)

    Python异常处理如何才能写得优雅(retrying模块)

    异常就是程序运行时发生错误的信号,下面这篇文章主要给大家介绍了关于Python异常处理的相关资料,文中通过实例代码介绍的非常详细,需要的朋友可以参考下
    2022-03-03
  • Python脚本实现抓取指定网站上的所有图片

    Python脚本实现抓取指定网站上的所有图片

    对于开发者、数据分析师以及研究人员而言,从网页中提取有价值的信息是一项至关重要的技能,本文将详细介绍如何使用Python编写一个脚本来自动抓取指定网站上的所有图片,需要的可以参考下
    2024-10-10
  • Python 实现简单的客户端认证

    Python 实现简单的客户端认证

    这篇文章主要介绍了Python 如何实现简单的客户端认证,文中讲解非常细致,代码帮助大家更好的理解和学习,感兴趣的朋友可以了解下
    2020-07-07

最新评论