该博客介绍了一个Python脚本change_dir.py，其功能是将TXT文件中保存的本地图片路径替换为服务器上的路径。脚本读取原始TXT文件，解析每行中的最后一个文件名，并在新路径前添加'./images/train2017/'，然后将更新后的路径写入新的TXT文件。这个过程对于在服务器上训练模型时确保正确引用文件路径非常有用。
在自己电脑上生成txt文件保存的是本机的路径，但是需要放到服务器上训练，此时就需要修改txt文件中的路径为服务器上的路径，用以下代码即可解决。
change_dir.py
import os
txt_path1 = 'train_all.txt'
txt_path2 = 'train_all_2.txt'
txt_new = []
f1 = open(txt_path1, 'r',encoding='utf-8')
for line in f1:
img_name = line.split('/')[-1]
line_new = './images/train2017/'+ img_name
txt_new.append(line_new)
f1.close()
f2 = open(txt_path2, 'w', encoding='utf-8')
for line in txt_new:
f2.writelines(line)
f2.close()
636
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
