已知感兴趣区域.csv文件,想通过循环使图片与.csv中每一行对应,从而进行mask处理,将不感兴趣区域变成黑色,但是好像循环的不对,恳请大家帮助
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')