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
新版数据处理
最新推荐文章于 2024-07-10 22:17:05 发布