❀数据集❀数据集的处理---格式之间的转换

1：图片路径转txt文件

#coding:utf8
#txt保存的图片的路径信息
import os
import sys
def listfiles(rootDir, txtfile, label=0):
    ftxtfile = open(txtfile, 'w')
    list_dirs = os.walk(rootDir)
    count = 0
    dircount = 0
    for root,dirs,files in list_dirs:
        for d in dirs:
            print(os.path.join(root, d))
            dircount += 1
        for f in files:
             print(os.path.join(root, f))
             ftxtfile.write(os.path.join(root, f)+ ' ' + str(label) + '\n')
             count += 1
    print(rootDir + ' has ' + str(count) + ' files')

listfiles('E:\pythonProject\cv2\images', 'E:\\pythonProject\\FormatConversion\\txtfile.txt')#记得修改路径

2：图片转npy文件

#为了节省空间，有时会将.jpg文件转换为.npy文件，这样便于存储，然后需要图片的时候，再将其转为.jpg文件
import PIL
import numpy as np
from PIL import Image
'''
path = 'E:/pythonProject/cv2/images/2.jpg' #
# 将其保存为.npy文件
image = Image.open(path)
image = image.resize((342, 256))
image = np.array(image)
np.save('image.npy', image)
'''
#读取这个.npy文件，并将其转换为图片
path = 'image.npy' # 要转换为图片的.npy文件
data = np.load(path)
image = Image.fromarray(data)
image.show()#要一步一步来，要把图片转为npy文件，再将npy文件随机赋名称给转换回来的照片。
# 直接全部运行的话，同一张照片俩个名称不同，会报错
#直接用来把别的npy文件转照片也是有的可以，有的不可以，nyp通道数可能不同。