# -*- 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')
根据图片改名,并改其对应txt label名
猜你喜欢
转载自blog.csdn.net/weixin_38740463/article/details/89669656
今日推荐
周排行