import os
target_path='/content/drive/MyDrive/srntt-pytorch-master/data/CUFED/ref'
target_path2='/content/drive/MyDrive/srntt-pytorch-master/data/CUFED/input'
all_content=os.listdir(target_path)
all_content2=os.listdir(target_path2)
print('All ref numbers is',len(all_content))
print('All input numbers is',len(all_content2))
# count_num=1
# file = open('/content/drive/MyDrive/srntt-pytorch-master/data/CUFED/b.txt','w')
# for content in all_content2:
# file.write(str(count_num) + '---------'+content + '\n')
# count_num+=1