专栏介绍
这个专栏旨在分享我在进行深度学习算法前的一些脚本文件。包括一些常见的处理,如将单通道图片改为三通道、统一图片格式、将标签进行统一设置方便查找等等。
合并不同文件夹内的同名文件
训练YOLO系列网络时,标注文件格式需转成txt。当增添检测类别时,往往需要将两个同名txt文件合并,因此该程序解决将不同文件夹下的同名txt文件进行合并。
import os
def get(name info):
# 写入新文件夹
for info_one in info:
with open(r"D:/2/{}".format(name), "a") as f:
f.write(info_one +'\n')
path = r"D:/1"
lists = os.listdir(path)
for l in lists:
# 子文件夹路径
p = os.path.join(path, l)
ps =os.listdir(p)
for p1 in ps:
# 每个文件路径
p2 = os.path.join(p, p1)
# 读取每个文件内容
with open(p2,"r")as f:
lines =f.read().splitlines()
get(p1, lines)
这个程序的功能主要是将"D:/1/"下的文件夹下的txt进行合并,并将txt文件保存至"D:/2/"下。
结果展示
最后的结果展示是将两个文件夹(新建文件夹、新建文件夹(2))中的txt文件合并保存在"D:/2"中。