Yolov3所需数据预处理python文件

图片-xml匹配、重命名、读取xml并重新生成xml

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import os
import re
import shutil

origin_dir = ‘D:\\ocr\\five_categories\\5_class_icon5\\5_class_icon4_labeled\\collect‘   #xml和jpg数量不对的目录
target_dir = ‘D:\\ocr\\five_categories\\5_class_icon5\\5_class_icon4_labeled\\collect-after‘    #转移后的目录

pattern = re.compile（‘（.*）\.（.*）‘）


def to_move（）:
    cursor = None
    last_name = None
    for f in sorted（os.listdir（origin_dir））:
        try:
            name postfix = re.match（pattern f）.groups（）
            if name == cursor:
                yield last_name
                yield f
            else:
                cursor = name
                last_name = f
        except Exception as e:
            print（e）


for i in to_move（）:
    origin = os.path.join（origin_dir i）
    if os.path.isfile（origin）:
        target = os.path.join（target_dir i）
        print（‘move {origin} to {target}‘）
        shutil.move（origin target）

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     文件        5090  2019-03-28 21:05  my_icdar13_to_voc.py
     文件        3305  2019-04-11 21:34  voc_to_icdar13.py
     文件         995  2019-04-13 22:22  match_quantity.py
     目录           0  2019-04-13 22:22  __MACOSX\
     文件         176  2019-04-13 22:22  __MACOSX\._match_quantity.py
     文件         274  2019-03-28 20:28  rename.py