Python:データセットのピクチャ名を読み出す、順次記憶する.txtファイル内

前言:mAP計算をするときに、テストサンプル名(順序が一致する)を含むtxtドキュメントが必要であることを覚えておいて、今このコードを整理してください.コード:

import os
import glob
import argparse

parser = argparse.ArgumentParser(description="read img name to write in txt")
parser.add_argument("--input_path",type=str,help="The path of the img path")
parser.add_argument("--save_path",type=str,help="The path of the txt save path")
parser.add_argument("--add_name_prefix",type=str,help="The name of the prefix of the file")
args = parser.parse_args()

IMAGE_LIST_FILE = "test.txt"


def read_img_name_to_txt(img_path,txt_path,prefix):
    IMAGE_NUM = len(glob.glob(img_path + '*.jpg'))
    f = open( os.path.join(txt_path, IMAGE_LIST_FILE), 'w')

    for i in range(1,IMAGE_NUM+1):
        print(i)
        f.write(str(prefix)+'_'+str(i)+'
')

    f.close()
    print ('FINISHED!')

if __name__ == "__main__":
    read_img_name_to_txt(args.input_path,args.save_path,args.add_name_prefix)


# python 04_read_img_to_txt.py --input_path ./test/JPEGImages/ --save_path ./test/ImageSets/Main/ --add_name_prefix new