EGNet实验中.lst文件的生成方法

一、训练集中的.lst的生成

import os
filePath1 = '/home/ubuntu/Downloads/datasets/DUTS-TR/DUTS-TR/DUTS-TR-Image'
filePath2 = '/home/ubuntu/Downloads/datasets/DUTS-TR/DUTS-TR/DUTS-TR-Mask'
filePath3 = '/home/ubuntu/Downloads/datasets/DUTS-TR/DUTS-TR/DUTS-TR-edge'

list_pic_1 = os.listdir(filePath1)
#list_pic_1.sort(key=lambda x: int(x[:-4]))  # 倒着数第四位'.'为分界线，按照.jpg的‘.’左边的数字从小到大排序
list_pic_1.sort(key=lambda x: x[:-4])
#print(type(list_pic_1))

list_pic_2 = os.listdir(filePath2)
list_pic_2.sort(key=lambda x: x[:-4])

list_pic_3 = os.listdir(filePath3)
list_pic_3.sort(key=lambda x: x[:-4])

if __name__ == '__main__':

    with open('lst_run.lst', 'w') as f:  # 你想生成的.lst文件名
        #for i in list_pic_1:
             #如果去掉图片后缀
             # i = i.split(".")[0]