读取文件名,生成txt目录。
import os
# get .txt document
rootdir=os.path.join('C:/Users/lishe/Desktop/data/shapenetcore_partanno_segmentation_benchmark_v0_normal/val')
# read
write_path=open('C:/Users/lishe/Desktop/write.txt','w')
for (dirpath,dirnames,filenames) in os.walk(rootdir):
for filename in filenames:
if os.path.splitext(filename)[1]=='.txt':
write_path.write('"'+'shape_data/val/'+filename.split(".")[0]+'",')
write_path.close()