roots="/DATACENTER4/hao.yang/project/Qin/data/preProcess/extractFrame/" rootsName=['0706_checkout'] for rootName in rootsName: rootdir=os.path.join(roots,rootName) tardir=os.path.join(roots,rootName+"All") #不用创建 print(rootdir,tardir) i=1 if os.path.exists(rootdir): FILES.mkdir(tardir) alldirs=FILES.get_sub_dirs(rootdir) for dirName in alldirs: if 'test' in dirName: continue dirpath=os.path.join(rootdir,dirName) # print(dirpath) alljpg=sorted([x for x in FILES.get_files(dirpath) if ".jpg" in x])#审核为.jpg形成列表 for jpg in alljpg: # rename # newJpg=dirpath.split("/")[-1]+str("%06d"%i)+".jpg" newJpg='{}_{}'.format(dirName,jpg) # newJpg=jpg i+=1 oldpath=os.path.join(dirpath,jpg) newpath=os.path.join(tardir,newJpg) shutil.copy(oldpath,newpath) print(time.clock()-timeStart) # i=1 # dir1=r"E:\BUS\BUS1" # filelist=os.listdir(dir1)
#encoding=utf8 #由于文件夹中只标注了部分refer,因此按照xx.refer复制已标注的xx.jpg和xx.refer到新的文件夹 #单文件夹 import os import shutil import time from basicFun import FILES if __name__=="__main__": count=0 start=time.time() operators=['gs','hs','px','tsy','zxy','yy','wzx','lf','dxj'] referDir1="/DATACENTER6/hao.yang/dataRoom/Qin/trade/labelTask_phone/labels_phone/" referForm=".xml" allRefers=[x for x in FILES.get_files(referDir1) if referForm in x]#审核为.refer形成列表 # dirName='safe_close' for operator in operators: count=0 # referDir2=referDir1.replace('door_close','door_open') sourDir="/DATACENTER6/hao.yang/dataRoom/Qin/trade/labelTask_phone/{}".format(operator) sourForm=".jpg" tarDir='/DATACENTER6/hao.yang/dataRoom/Qin/trade/labelTask_phone/abandon/' # tarDir=os.path.join(tarDir,operator) tarForm=".jpg" FILES.mkdir(tarDir) allImgs=[x for x in FILES.get_files(sourDir) if sourForm in x] for img in allImgs: if img.replace(sourForm,referForm) not in allRefers: sour=img.split('.')[0]+sourForm tar=img.split('.')[0]+tarForm sourPath=os.path.join(sourDir,sour) tarPath=os.path.join(tarDir,tar)
# font_label = ImageFont.truetype('/DATACENTER2/yh/resources/heiti.TTF',50) # img_PIL = Image.fromarray(cv2.cvtColor(image,cv2.COLOR_BGR2RGB)) # cv2==>PIL # draw = ImageDraw.Draw(img_PIL) # draw.text((xmin,ymax-50),label_name,font_color,font=font_label) # image = cv2.cvtColor(numpy.asarray(img_PIL),cv2.COLOR_RGB2BGR) # PIL==>cv2 # img_PIL = Image.fromarray(cv2.cvtColor(image,cv2.COLOR_BGR2RGB)) # cv2==>PIL # draw = ImageDraw.Draw(img_PIL) # image = cv2.cvtColor(numpy.asarray(img_PIL),cv2.COLOR_RGB2BGR) # PIL==>cv2 return image if __name__ == "__main__": start = time.time() jpgDir = r"/disk2/hao.yang/project/Qin/data/xmls/isle/complete/imgs_overlook_16000_test/" xmlDir = r"/disk2/hao.yang/project/Qin/data/xmls/isle/complete/isle_FMXX_test_16000/" outDir = r"/disk2/hao.yang/project/Qin/data/xmls/isle/complete/imgs_overlook_16000_test_truth/" FILES.rm_mkdir(outDir) allJpgs = FILES.get_files(jpgDir) for jpg in allJpgs: # print(jpg) if '.jpg' in jpg: jpgPath = os.path.join(jpgDir, jpg) xmlPath = os.path.join(xmlDir, jpg.split('.')[0] + '.xml') objs = XML.read_objects(xmlPath) image = cv2.imread(jpgPath) img = drawXml(image, objs) desImg = os.path.join(outDir, jpg) cv2.imwrite(desImg, img) # extraFrames(davDir,mp4Dir)