技術標籤：EGNet遇到問題

import os


filePath = '/home/xx/data/ECSSD/image'
list_jpg = os.listdir(filePath)

if __name__ == '__main__':
    '''
    => 用於完成test.txt, 要在此檔案目錄下事先建立test.txt檔案
    與後面完成train.txt分開單獨使用時, 直接註釋掉另一個
    '''
    # /=== ===> test.txt 寫入pic_name & label <=== ===\
    with open('test_pre.txt', 'w') as f:  # train.txt或者test.txt 本檔案生成在本程式碼所在位置
        for i in list_jpg:
            # 如果去掉圖片字尾
            # i = i.split(".")[0]
            
            f.writelines(i + '\n')

    # \=== ===> test.txt 寫入pic_name & label <=== ===/
 

字尾可去可新增