python-----将图片与标注的xml坐标水平翻转

左手的ㄟ右手 2021-12-01 10:46 255阅读 0赞

我们做机器学习的时候，总会用到很多训练集，然后我们的数据比较少的时候，就可以将图片翻转标注。代码如下：

#!/usr/bin/env python
    # -*- coding: utf-8 -*-
    # @Time    : 2019/6/28 13:59
    # @Author  : xiaodai
    import cv2
    from PIL import Image
    import xml.etree.ElementTree as ET
    from tqdm import tqdm
    import os
    
    def flip_horizontal(jpg_file,new_jpg):
        im = Image.open(jpg_file)
        # out = im.transpose(Image.ROTATE_180)
        out = im.transpose(Image.FLIP_LEFT_RIGHT)
        out.save(new_jpg)
    
    
    def flip_xml(xml_file,new_xml):
        tree = ET.parse(xml_file)
        objs = tree.findall('object')
        for ix, obj in enumerate(objs):
            name = obj.find('name').text
            if name == 'mouse':
                print(xml_file)
                obj_new = obj.find('bndbox')
                xmin = str( 1280 - int(obj_new.find('xmin').text))
                xmax = str( 1280 - int(obj_new.find('xmax').text))
                obj_new.find('xmin').text = xmin
                obj_new.find('xmax').text = xmax
                tree.write(new_xml)
    
    
    if __name__ == '__main__':
        path = r'F:\temp\123\55_38\0625\mouse'
        xml_files = [os.path.join(rootdir, file) for rootdir, _, files in os.walk(path) for file in files if
                     (file.endswith('.xml'))]
        jpg_files =[xmlfile.replace('Annotations','JPEGImages').replace('xml','jpg') for xmlfile in xml_files]
        files = zip(jpg_files,xml_files)
        for jpg_file,xml_file in tqdm(files):
            new_jpg = jpg_file.replace(jpg_file.split('\\')[-1].split('.')[0],jpg_file.split('\\')[-1].split('.')[0] + '_filp')
            new_xml = new_jpg.replace('JPEGImages', 'Annotations').replace('jpg', 'xml')
            flip_xml(xml_file,new_xml)
            flip_horizontal(jpg_file, new_jpg)

转载于:https://www.cnblogs.com/xiaodai0/p/11153672.html