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)+'\n')

    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