cifar-10 图像转为jpg

dir_file目录下需有这几个文件

源代码

#coding=utf-8
import cv2
import numpy as np
import os#文件夹名
str_2 = './train_cifar10'
str_1 = './test_cifar10'#判断文件夹是否存在，不存在的话创建文件夹
if os.path.exists(str_1) == False:
os.mkdir(str_1)
if os.path.exists(str_2) == False:
os.mkdir(str_2)# 解压缩，返回解压后的字典，f,encoding='bytes'
def unpickle(file):
import pickle
fo = open(file, 'rb')
dict = pickle.load(fo, encoding='bytes')
fo.close()
return dictdef cifar_jpg(dir_file):
# 生成训练集图片，如果需要png格式，只需要改图片后缀名即可。
for j in range(1, 6):
dataName = dir_file + '/' + "data_batch_" + str(j) # 读取当前目录下的data_batch12345文件，dataName其实也是data_batch文件的路径，本文和脚本文件在同一目录下。
Xtr = unpickle(dataName)
#print(Xtr)
print(dataName + " is loading...")for i in range(0, 10000):
img = np.reshape(Xtr[b'data'][i], (3, 32, 32)) # Xtr['data']为图片二进制数据
img = img.transpose(1, 2, 0) # 读取image
picName = './train_cifar10/' + str(Xtr[b'labels'][i]) + '_' + str(
i + (j - 1) * 10000) + '.jpg' # Xtr['labels']为图片的标签，值范围0-9，本文中，train文件夹需要存在，并与脚本文件在同一目录下。
cv2.imwrite(picName, img)
print(dataName + " loaded.")print("test_batch is loading...")# 生成测试集图片
testName = dir_file + '/' + 'test_batch'
testXtr = unpickle(testName)
for i in range(0, 10000):
img = np.reshape(testXtr[b'data'][i], (3, 32, 32))
img = img.transpose(1, 2, 0)
picName = './test_cifar10/' + str(testXtr[b'labels'][i]) + '_' + str(i) + '.jpg'
cv2.imwrite(picName, img)
print("test_batch loaded.")
return#标签与名字的对应关系
def label_name():
label_name_dict = {
'airplane': "0",
'automobile': "1",
'bird': "2",
'cat': "3",
'deer': "4",
'dog': "5",
'frog': "6",
'horse': "7",
'ship': "8",
'truck': "9"
}
return label_name_dictif __name__ == '__main__':
dir_file = './cifar-10-batches-py'
try:
cifar_jpg(dir_file)
except:
print('函数报错')

cifar-10 图像转为jpg相关推荐

彩色RGB图像转为灰度图像
将彩色RGB图像转为灰度图像,其中像素值的转换为灰度值=0.2989 * R + 0.5870 * G + 0.1140 * B 原彩色RGB图像是三通道的,转换成单通道的灰度图像自己写了一个Py ...
python输出jpg_利用python实现.dcm格式图像转为.jpg格式
如下所示: import pydicom import matplotlib.pyplot as plt import scipy.misc import pandas as pd import nu ...
【OpenCV 例程200篇】10. 图像的拼接（np.hstack）
[OpenCV 例程200篇]10. 图像的拼接(np.hstack) 欢迎关注『OpenCV 例程200篇』系列,持续更新中欢迎关注『Python小白的OpenCV学习课』系列,持续更新中 ...
python bmp转jpg_利用python实现.dcm格式图像转为.jpg格式
如下所示: import pydicom import matplotlib.pyplot as plt import scipy.misc import pandas as pd import nu ...
将图像转为特征值_用K均值进行图像分割
个人学习笔记:采用聚类方法对图像进行分割,以下内容纯粹个人理解,如有错误请帮我指出!多谢! 图像分割就是把图像按照某些条件分成不同的区域,并提取出感兴趣的区域.传统的分割方法包括基于阈值的分割.基于区 ...
【ENVI入门系列】10.图像裁剪
[ENVI入门系列]10.图像裁剪 (2014-09-26 10:08:15) 转载▼ 标签: 杂谈分类: ENVI 版权声明:本教程涉及到的数据仅供练习使用,禁止用于商业用途. 目录图像裁剪 1 ...
python图片保存jpg、show变成bmp_利用python实现.dcm格式图像转为.jpg格式
如下所示: import pydicom import matplotlib.pyplot as plt import scipy.misc import pandas as pd import nu ...
用python将.dcm格式图像转为.jpg格式
用python将.dcm格式图像转为.jpg格式 import pydicom import matplotlib.pyplot as plt import scipy.misc import pan ...
如何将深度学习的float32图像转为Unit8格式以方便cv2使用
在使用Pyside2中的 QImage处理深度学习模型生成的图片时,需要将float32的图像转为Unit8格式,再使用cv2处理. 一开始使用网上的其他教程,如下: # 模型生成G_recon = ...
OpenCV-Python 中文教程10——图像阈值
OpenCV-Python 中文教程10--图像阈值目标 • 学习简单阈值,自适应阈值, Otsu's 二值化等 • 学习函数 cv2.threshold, cv2.adaptiveThresho ...

cifar-10 图像转为jpg

cifar-10 图像转为jpg相关推荐

最新文章

热门文章