前言:
记得在做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