图像存储位置如下所示：
原图与gt文件在同一文件夹下，为其创建csv文件用于保存路径，便于程序读取。
2.csv的创建import csv
import os
path1 = '/data1/wmw/dataset/weixing/test/' # 所需修改文件夹所在路径
dirs1 = os.listdir(path1)
path2 = '/data1/wmw/dataset/weixing/gt/'
dirs2 = os.listdir(path2)
with open("metadata.csv", "w") as csvfile:
writer = csv.writer(csvfile)
writer.writerow(["image_id", "sat_image_path", "mask_path"])
j = 0
for dir in dirs1:
for dir0 in dirs2:
if dir.split('_')[0]==dir0.split('_')[0]:
writer.writerow([j, 'test/'+dir, 'gt/'+dir0])
j=j+1
csv文件内容展示：
3.csv的读取（片段）
DATA_DIR = '/data1/wmw/dataset/weixing/test/'
metadata_df = pd.read_csv('/data1/wmw/dataset/weixing/metadata.csv')
metadata_df = metadata_df[metadata_df['split'] == 'test']
metadata_df = metadata_df[['image_id', 'sat_image_path', 'mask_path']]
metadata_df['sat_image_path'] = metadata_df['sat_image_path'].apply(lambda img_pth: os.path.join(DATA_DIR, img_pth))
metadata_df['mask_path'] = metadata_df['mask_path'].apply(lambda img_pth: os.path.join(DATA_DIR, img_pth))
# Shuffle Dataframe
metadata_df = metadata_df.sample(frac=1).reset_index(drop=True)
print(metadata_df)
