from pathlib import Path
if __name__ == '__main__':
img_files = [
Path("C:/Users/ytc79/Desktop/CrowdCounting-P2PNet-main/vis/part_B_final/train_data/images/")]#图片路径
txt_files = [
Path("C:/Users/ytc79/Desktop/CrowdCounting-P2PNet-main/vis/part_B_final/train_data/txt/")]#txt路径
result_png_list = r"C:/Users/ytc79/Desktop/CrowdCounting-P2PNet-main/vis/part_B_final/train_data/train.txt"#要写到哪个文件中
write_png = open(result_png_list, "w", encoding="utf-8")#打开要写入的txt文件
for img_file, txt_file in zip(img_files, txt_files):
for img_pth, txt_pth in list(zip(img_file.glob("IMG_*.jpg"), txt_file.glob("GT_IMG_*.txt"))):
#我这里是IMG_开头的jpg和GT_IMG_开头的txt对应,如果只有后缀名不同,直接img_file.glob("*.jpg"), txt_file.glob("*.txt")即可
write_png.write(str(img_pth) + "\x20" + str(txt_pth)+'\n')#路径拼接,加一个空格\x20,,再换行
write_png.close()#关闭文件
结果如图所示:
参考了这位老师的代码:https://blog.csdn.net/xiangzun0155/article/details/128470042