python如何将txt坐标批量打印到原图上
作者:慕容洛凝
这篇文章主要介绍了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)
总结
以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。