006_切割图片根据xml
# 切割图片按名称切割放入文件夹```pythonimport xml.etree.ElementTree as ETimport cv2import osfrom lxml import etreefrom tqdm import tqdmclass GEN_Annotations:def __init__(self, filename):self.root = etree.Element("a
·
# 切割图片 按名称切割放入文件夹
```python
import xml.etree.ElementTree as ET
import cv2
import os
from lxml import etree
from tqdm import tqdm
class GEN_Annotations:
def __init__(self, filename):
self.root = etree.Element("annotation")
child1 = etree.SubElement(self.root, "folder")
child1.text = "VOC2007"
child2 = etree.SubElement(self.root, "filename")
child2.text = filename
child3 = etree.SubElement(self.root, "source")
child4 = etree.SubElement(child3, "annotation")
child4.text = "PASCAL VOC2007"
child5 = etree.SubElement(child3, "database")
child5.text = "Unknown"
child6 = etree.SubElement(child3, "image")
child6.text = "flickr"
child7 = etree.SubElement(child3, "flickrid")
child7.text = "35435"
def set_size(self, witdh, height, channel):
size = etree.SubElement(self.root, "size")
widthn = etree.SubElement(size, "width")
widthn.text = str(witdh)
heightn = etree.SubElement(size, "height")
heightn.text = str(height)
channeln = etree.SubElement(size, "depth")
channeln.text = str(channel)
def savefile(self, filename):
tree = etree.ElementTree(self.root)
tree.write(filename, pretty_print=True, xml_declaration=False, encoding='utf-8')
def add_pic_attr(self, label, xmin, ymin, xmax, ymax):
object = etree.SubElement(self.root, "object")
namen = etree.SubElement(object, "name")
namen.text = label
bndbox = etree.SubElement(object, "bndbox")
xminn = etree.SubElement(bndbox, "xmin")
xminn.text = str(xmin)
yminn = etree.SubElement(bndbox, "ymin")
yminn.text = str(ymin)
xmaxn = etree.SubElement(bndbox, "xmax")
xmaxn.text = str(xmax)
ymaxn = etree.SubElement(bndbox, "ymax")
ymaxn.text = str(ymax)
def get_ext(w,h,b):
# up down 100 left right 50
b1 = []
# print(b[0],b[1],b[2],b[3])
b1.append(max(0, b[0] - 300))
b1.append(max(0, b[1] - 300 ))
b1.append(min(w, b[2] + 300))
b1.append(min(h, b[3] + 300))
return b1
# 原图
path_img = r'D:\biaoji\JPEGImages'
path_xml = r'D:\biaoji\an'
# 切割后
out_path_img =r'D:\01_biaoji\_jpg'
out_path_xml = r'D:\01_biaoji\_ann'
def makedir(*new_dir):
for dir in new_dir:
if not os.path.exists(dir):
os.makedirs(dir)
makedir(out_path_img, out_path_xml)
img_dir = os.listdir(path_img)
xml_dir = os.listdir(path_xml)
idx = 0
#clss = ['jueyuanzi','junyahuan','zhishipai','bileiqi','hengdan','tonggan','fangzhenchui','baogu']#niaochao,xianjia
clss = ["biaoji"] #按照切割的名字
for i in tqdm(range(len(img_dir))):
xml_path = os.path.join(path_xml, img_dir[i][:-3]+'xml')
jpg_file = os.path.join(path_img, img_dir[i])
if not os.path.exists(xml_path):
continue
im = cv2.imread(jpg_file)
tree = ET.parse(xml_path) # 解析xml
root = tree.getroot()
size = root.find('size') # 图片尺寸<Element 'size' at 0x000002A6DF8DB778>
w = int(size.find('width').text) # 图片的宽
# print(w)
h = int(size.find('height').text) # 图片的高
# print(h)
# is_show = False
is_show = True
for obj in root.iter('object'):
cls = obj.find('name').text
if cls in clss:
# is_show = False
xmlbox = obj.find('bndbox')
b = [float(xmlbox.find('xmin').text), float(xmlbox.find('ymin').text), float(xmlbox.find('xmax').text),
float(xmlbox.find('ymax').text)]
b1 = get_ext(w, h, b)
xmin, ymin, xmax, ymax = int(b[0]-b1[0] ), int(b[1]-b1[1]), int(b[2]-b1[0]), int(b[3] - b1[1])
new_img = im[int(b1[1]):int(b1[3]), int(b1[0]):int(b1[2])].copy()
idx += 1
new_img_path = os.path.join(out_path_img, img_dir[i][:-4]+'_'+str(idx).zfill(6)+".jpg")
new_xml_path = os.path.join(out_path_xml, img_dir[i][:-4]+'_'+str(idx).zfill(6)+".xml")
# 写图片
# print('图片',new_img_path)
cv2.imwrite(new_img_path,new_img)
# 写标签
anno = GEN_Annotations(new_img_path)
nh, nw, nc = new_img.shape
anno.set_size(nw, nh, nc)
anno.add_pic_attr(cls, xmin, ymin, xmax, ymax)
# print('xml', new_xml_path)
anno.savefile(new_xml_path)
# cv2.waitKey(0)
更多推荐



所有评论(0)