图像处理问题,已知图片集和对应的excel表,每一行存有一个图片感兴趣区域,我想同时遍历图和表中每一行,实现与运算,从而留下感兴趣区域,使不感兴趣区域变成黑色,但是此程序循环不对?
def Image_Resize(dir):
for parent,dirnames,filenames in os.walk(dir):
for filename in filenames:
save_path = os.path.join('D:/Testing_mask/00000', filename)
csv_data = pd.read_csv('D:/BLS_Test_Images_jpg/00000/GT-00000.csv')
csv_data_array = np.array(csv_data)
for i in range(csv_data_array.shape[0]):
csv_data_list = np.array(csv_data)[i, :].tolist()[0].split(";")
sample_dir = os.path.join(save_path, csv_data_list[0])
img = cv2.imread(sample_dir)
mask = np.zeros(img.shape[:2], np.uint8)
mask[int(csv_data_list[3]):int(csv_data_list[5]), int(csv_data_list[4]):int(csv_data_list[6])] = 255
cv2.imwrite(sample_dir, masked_img)
Image_Resize('D:/Testing_mask/00000')