bmp图像，jpg图像转npy

一.bmp文件转jpg时，可以稍作改动，就可以转成其他图像格式了

def bmpToJpg(file_path):
    for fileName in os.listdir(file_path):
        oldpath=os.path.join(file_path,fileName)
        #newFileName 中的jpg,改成其他格式，就可以实现上述操作，例如png
        newFileName = os.path.join(file_path,fileName.split('.')[0]+".jpg")

        img = cv2.imread(oldpath)
        cv2.imwrite(newFileName, img)
        os.remove(oldpath)

二.jpg转npy

def jpgtonpy(path):
    for file in os.listdir(path):
        sourceDir1 = os.path.join(path,file)
        im1 = cv2.imread(sourceDir1)
        im2 = np.array(im1)
        file_name = os.path.join(path,file.split('.')[0]+'.npy')
        np.save(file_name, im2)
        os.remove(sourceDir1)

三.改名

def newname(root_path):
    filelist = os.listdir(root_path)  # 遍历文件夹
    i = 0
    for item in filelist:

        if item.endswith('.npy'):
            src = os.path.join(os.path.abspath(root_path), item)  # 原本的名称
            dst = os.path.join(os.path.abspath(root_path),
                               os.path.basename('amp_sample') + str(i) + '.npy')  # 这里我把格式统一改成了 .jpg
                               # os.path.basename('sample') + str(i) + '.npy')  # 这里我把格式统一改成了 .jpg

        try:
            os.rename(src, dst)  # 意思是将 src 替换为 dst
            i += 1
            print('rename from %s to %s' % (src, dst))
        except:
            continue

四.合集

# 尝试将转化的npy文件展示其中的内容。
import numpy as np
import os
import cv2

# bmp 转换为jpg
def bmpToJpg(file_path):
    for fileName in os.listdir(file_path):
        oldpath=os.path.join(file_path,fileName)
        newFileName = os.path.join(file_path,fileName.split('.')[0]+".jpg")

        img = cv2.imread(oldpath)
        cv2.imwrite(newFileName, img)
        os.remove(oldpath)
# jpg 转换为npy
def jpgtonpy(path):
    for file in os.listdir(path):
        sourceDir1 = os.path.join(path,file)
        im1 = cv2.imread(sourceDir1)
        im2 = np.array(im1)
        file_name = os.path.join(path,file.split('.')[0]+'.npy')
        np.save(file_name, im2)
        os.remove(sourceDir1)

def newname(root_path):
    filelist = os.listdir(root_path)  # 遍历文件夹
    i = 0
    for item in filelist:

        if item.endswith('.npy'):
            src = os.path.join(os.path.abspath(root_path), item)  # 原本的名称
            dst = os.path.join(os.path.abspath(root_path),
                                #这里是修改数据集和标签的文件
                               os.path.basename('amp_sample') + str(i) + '.npy')  
                               # os.path.basename('sample') + str(i) + '.npy')  

        try:
            os.rename(src, dst)  # 意思是将 src 替换为 dst
            i += 1
            print('rename from %s to %s' % (src, dst))
        except:
            continue

if __name__ == '__main__':
    # path = r'../dataset/client1/data_npy'
    # jpgtonpy(path)
    path = r'../dataset/client1/freq_amp_npy'
    bmpToJpg(path)