weixin_44961126
迎儿2329
2019-05-25 07:39

交通标志图像处理,已知感兴趣区域.csv文件,如何循环使图片与.csv中每一行对应,从而进行mask处理?

已知感兴趣区域.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')
  • 点赞
  • 写回答
  • 关注问题
  • 收藏
  • 复制链接分享
  • 邀请回答

为你推荐