根据图片改名，并改其对应txt label名

# -*- coding: utf-8 -*-
"""
Created on Sun Jun 11 18:04:24 2017
@author: wuhui
用来对图片进行重命名，并初始化目录结构
"""

import cv2
import os

if __name__=='__main__':
    path="C:\\Users\\chenyu03\\PycharmProjects\\test1\\venv\\src\\src\\EastTrain\\data\\test"
    imgs = os.listdir(path+'\\img')
    # if os.path.exists('JPEGImages') == False:
    #     os.mkdir('JPEGImages')
    # if os.path.exists('Annotations') == False:
    #     os.mkdir('Annotations')
    # if os.path.exists('ImageSets') == False:
    #     os.mkdir('ImageSets')
    #     os.mkdir('ImageSets/Main')
    cnt = 1
    prename = "000000"
    for img in imgs:
        print(img)
        temp=cv2.imread(path+'\\img\\'+img)
        if  os.path.exists(path+'\\ann\\'+img[:-4]+'.txt'):
            print('22')
            cv2.imwrite(path+"\\JPEGImages\\"+prename[0:len(prename)-len(str(cnt))]+str(cnt)+".jpg",temp)
            os.rename(path+'\\ann\\'+img[:-4]+'.txt',path+'\\annt\\'+prename[0:len(prename)-len(str(cnt))]+str(cnt)+".txt")
        print('111')
            #print ("renamed "+img+" to "+prename[0:len(prename)-len(str(cnt))]+str(cnt)+".jpg")
        cnt+=1
    print('done')