处理数据集:多行信息合并为一行,目的是与图片对应,方便对单张图片进行预测
import pandas as pd
import numpy as np
import os
import re
for filename in os.listdir("kvasir-dataset-v2-features"):
for filename2 in os.listdir("kvasir-dataset-v2-features/" + filename):
with open("kvasir-dataset-v2-features/"+ filename + "/"+ filename2) as f:
for line in f.readlines():
line=line.strip('\n')
with open("xinxi/" + filename + "/"+ filename2,"a") as all:
all.write(line + ",")
6000个文件只需几秒,真不错