导入必要的模块
import os
import cv2
定义要裁剪的图片的路径变量
DATADIR = r"E:\系统默认\桌面\pytorch-CycleGAN-and-pix2pix123\datasets\RGB2ToF\\"
data_k = "trainA"
path = os.path.join(DATADIR, data_k)
os.listdir()函数:返回指定的文件夹包含的文件或文件夹的名字的列表。
img_list = os.listdir(path)
循环遍历文件夹中的文件
for i in img_list:
img1 = cv2.imread(path + '/' + i )
cv2.imshow("Original Image",img1 )
start_row,start_col=150,150
end_row,end_col=1150,1150
cropped=img1[start_row:end_row,start_col:end_col]
cv2.imshow("Cropped_Image",cropped)
'''生成图片存储的目标路径'''
img_name = str(i)
save_path = path + '3/'
if not os.path.exists(save_path):
os.makedirs(save_path)
save_img = save_path + img_name
cv2.imwrite(save_img, cropped)
cv2.waitKey(1)
cv2.destroyAllWindows()
将RGB属性的图片转换为HSV属性
data_k1 = "trainA3"
path1 = os.path.join(DATADIR, data_k1)
img_list1 = os.listdir(path1)
for i in img_list1:
img1 = cv2.imread(path1 + '/' + i)
cv2.imshow('img0',img1)
img2 = cv2.cvtColor(img1,cv2.COLOR_BGR2HSV)
cv2.imshow('img2_2',img2)
img_name1 = str(i)
save_path = path1 + '3_3/'
if not os.path.exists(save_path):
os.makedirs(save_path)
save_img1 = save_path + img_name1
cv2.imwrite(save_img1, img2)
cv2.waitKey(1)
cv2.destroyAllWindows()
完整代码
import cv2
import random
import numpy as np
import os
DATADIR = r"E:\系统默认\桌面\pytorch-CycleGAN-and-pix2pix123\datasets\RGB2ToF\\"
data_k = "trainA"
path = os.path.join(DATADIR, data_k)
img_list = os.listdir(path)
for i in img_list:
img1 = cv2.imread(path + '/' + i )
cv2.imshow("Original Image",img1 )
start_row,start_col=150,150
end_row,end_col=1150,1150
cropped=img1[start_row:end_row,start_col:end_col]
cv2.imshow("Cropped_Image",cropped)
'''生成图片存储的目标路径'''
img_name = str(i)
save_path = path + '3/'
if not os.path.exists(save_path):
os.makedirs(save_path)
save_img = save_path + img_name
cv2.imwrite(save_img, cropped)
cv2.waitKey(1)
cv2.destroyAllWindows()
data_k1 = "trainA3"
path1 = os.path.join(DATADIR, data_k1)
img_list1 = os.listdir(path1)
for i in img_list1:
img1 = cv2.imread(path1 + '/' + i)
cv2.imshow('img0',img1)
img2 = cv2.cvtColor(img1,cv2.COLOR_BGR2HSV)
cv2.imshow('img2_2',img2)
img_name1 = str(i)
save_path = path1 + '3_3/'
if not os.path.exists(save_path):
os.makedirs(save_path)
save_img1 = save_path + img_name1
cv2.imwrite(save_img1, img2)
cv2.waitKey(1)
cv2.destroyAllWindows()