新版数据处理

import cv2 as cv
import os

img_path =r'C:\Users\86152\Desktop\344\JPEGImages\\'
new_path = r'C:\Users\86152\Desktop\the new imageDatasets\\'
if os.path.exists(new_path):
    pass
else:
    os.mkdir(new_path)
# img = cv.imread(img_path)
# print(img.shape)

w = 1024
h = 600
aspectRation = w/h
dim = (w,h)

for i in range(0,len(os.listdir(img_path)),1):
    img_path_new = img_path + os.listdir(img_path)[i]
    img = cv.imread(img_path_new)
    img_row,img_col,img_weidu = img.shape

    aspectRation_test = img_row/img_col

    #condition1:
    if img_row < w * 0.8 or img_col < h * 0.8:
        continue
    #condition2:
    if aspectRation_test <= aspectRation * 0.8 and aspectRation_test > aspectRation:
        continue
    #condition3:
    for k in range(0,img_row-w+1,w):
        for j in range(0,img_col-h+1,h):
            print(i,k,j,end='\n当前符合条件的照片名称为{}\n'.format(os.listdir(img_path)[i]))
            img_test = img[k:k+w,j:j+h]
            if img_test.shape[0] == w and img_test.shape[1] == h:
                pass
            else:
                img_test = cv.resize(img_test,dim,interpolation=cv.INTER_AREA)    #i指代的是第几个照片 k j 表示当前位置
            cv.imwrite(r'C:\Users\86152\Desktop\the new imageDatasets\test{}_{}_{}.jpg'.format(i,k,j),img_test)


# 将名字问题归结为统一的形式，便于管理
filelist = os.listdir(new_path) #该文件夹下所有的文件（包括文件夹）
count=1
for file in filelist:
    print(file)
for file in filelist:   #遍历所有文件
    Olddir=os.path.join(new_path,file)   #原来的文件路径
    if os.path.isdir(Olddir):   #如果是文件夹则跳过
        continue
    filename=os.path.splitext(file)[0]   #文件名
    filetype=os.path.splitext(file)[1]   #文件扩展名
    Newdir=os.path.join(new_path,str(count).zfill(6)+'.jpg')  #用字符串函数zfill 以0补全所需位数
    os.rename(Olddir,Newdir)#重命名
    count+=1