CIFAR-10 数据转为图片-python

  • CIFAR-10 数据集
"""
CIFAR-10 是 32X32 的彩色图片,共有10个类别,每个类别6000张图片,50000张训练图片(均分为5个batch),10000张测试图片(每个类别选1000张)
将 CIFAR-10 转为 png
"""import os
import pickleimport numpy as np
from imageio import imwrite# 数据存放的根目录
base_dir = r'H:\DataStore'
# cifar-10 存放位置
data_dir = os.path.join(base_dir, 'cifar-10-batches-py')
# 训练图片目录
train_dir = os.path.join(base_dir, 'cifar-10-train-png')
# 测试图片目录
test_dir = os.path.join(base_dir, 'cifar-10-test-png')# 这里不进行训练图片的生成
Train = False
Test = True# 反序列化
def unpickle(file_path):with open(file_path, 'rb') as f:_obj = pickle.load(f, encoding='bytes')return _obj# 目录不存在时创建一个
def create_dir(dir_path):if not os.path.isdir(dir_path):os.makedirs(dir_path)def get_label_names():_label_names_obj = unpickle(os.path.join(data_dir, 'batches.meta'))return _label_names_obj[b'label_names']def save_images(i, obj, class_num, label_names, dir_path):# 红、绿、蓝 (channel, width, height)# 一定要使用 b'' 的方式,因为 obj 是 bytes 编码的img = np.reshape(obj[b'data'][i], (3, 32, 32))# 保存为图片使用 (width, height, channel) 格式img = img.transpose(1, 2, 0)# 获取当前图片的类别下标 0-9label_idx = obj[b'labels'][i]# 获取当前图片的名称_label_name: str = label_names[label_idx].decode()train_dir_label_name_path = os.path.join(dir_path, _label_name)create_dir(train_dir_label_name_path)# 图片对应的类别数量+1class_num[label_idx] += 1_image_name = str(class_num[label_idx]) + '.png'image_path = os.path.join(train_dir_label_name_path, _image_name)# 写入图片imwrite(image_path, img)if __name__ == '__main__':_label_names = get_label_names()if Train:# 累计每个类别的数量train_class_num = [0] * 10for i in range(1, 6):data_batch_path = os.path.join(data_dir, 'data_batch_' + str(i))# k: data、labelstrain_batch_obj = unpickle(data_batch_path)print("{} is loading...".format(data_batch_path))# 每个batch中有10000张图片for j in range(0, 10000):save_images(j, train_batch_obj, train_class_num, _label_names, train_dir)print('train loaded')if Test:test_class_num = [0] * 10test_data_path = os.path.join(data_dir, 'test_batch')test_obj = unpickle(test_data_path)for i in range(10000):save_images(i, test_obj, test_class_num, _label_names, test_dir)print('test loaded')

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/diannao/21208.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

【STL】C++ queue队列(包含优先级队列) 基本使用

目录 一 queue 1 常见构造 1 空容器构造函数 2. 使用指定容器构造 3 拷贝构造函数 2 empty 3 size 4 front && back 5 push && pop 6 emplace 7 swap 二 优先级队列( priority_queue) 1 常见构造 2 其他操作 3 大堆和小堆 1. 大小堆切换 2 自…

Python知识点10---函数

提前说一点:如果你是专注于Python开发,那么本系列知识点只是带你入个门再详细的开发点就要去看其他资料了,而如果你和作者一样只是操作其他技术的Python API那就足够了。 Python的函数和Scala的函数很像,语法很简单,注…

scGPT实验解读

本篇内容为发表在Nature Methods上的scGPT的部分实验内容 来自:scGPT: toward building a foundation model for single-cell multi-omics using generative AI, Nature Methods, 2024 目录 scGPT揭示特定细胞状态的基因网络缩放法则和迁移学习中的上下文效应 scGP…

微信小程序 数据请求

在微信小程序中想要对其他接口进行数据请求,首先在【微信公众平台】中登录小程序, 微信公众平台的地址为:微信公众平台 (qq.com),在服务器域名中配置request合法域名,配置完后,在自己的微信小程序右侧的【…

寄存器、缓存、内存(虚拟、物理地址)、DDR、RAM的关系

寄存器、缓存、内存、DDR、RAM的关系 1. 主要概念内部存储器:2.1 寄存器,register2.2 主存储器,内存,memory2.3 缓存,高速缓冲存储器,cache 外部存储器2.4 快闪存储器,闪存,flash Me…

Python微服务架构的15个设计模式

对于初踏入微服务领域的Python开发者来说,掌握一系列设计模式是构建健壮、可扩展系统的关键。今天,我们将一起揭开这10个设计模式的神秘面纱,让你的微服务之旅更加顺畅。 目标读者:如果你是Python初学者,对微服务架构…

【风云】中国互联网发展的8个阶段

中国互联网软件的发展经历了多个阶段,每个阶段都有其标志性的事件和产品。总结下关键的发展阶段和标志: 起步阶段(1994年前后) 1994年,中国实现与国际互联网的第一条TCP/IP全功能链接,成为互联网大家庭中…

AI图书推荐:使用GPT-4和ChatGPT开发AI应用APP

这本书是面向想要学习如何使用大型语言模型构建应用程序的 Python 开发人员的全面指南。作者 Olivier Caelen 和 Marie-Alice Blete 涵盖了 GPT-4 和 ChatGPT 的主要特征和好处,并解释了它们的工作原理。您还将获得使用 GPT-4 和 ChatGPT Python 库开发应用程序的逐…

多台相同IP产品如何组网?

🏆本文收录于「Bug调优」专栏,主要记录项目实战过程中的Bug之前因后果及提供真实有效的解决方案,希望能够助你一臂之力,帮你早日登顶实现财富自由🚀;同时,欢迎大家关注&&收藏&&…

Unity协程详解

什么是协程 协程,即Coroutine(协同程序),就是开启一段和主程序异步执行的逻辑处理,什么是异步执行,异步执行是指程序的执行并不是按照从上往下执行。如果我们学过c语言,我们应该知道&#xff0…

【零基础AI】Windows下最易懂的Cuda+cudnn+PyTorch配置教程

【零基础AI】最易懂的Windows下CUDAcuDNNPyTorch配置教程 本文面向零基础的AI初学者以及迁移DL开发环境的相关人员,旨在通过最直观、最简洁的一条龙式图文配置教程,最大程度地降低学习门槛与迁移环境的成本。在配置工作开始之前,请务必检查确…

U-boot、linux内核、根文件系统移植以及程序

终于这几天把这个移植的流程过了一遍,所以特此回来总结。 U-boot移植 首先是U-boot移植。Linux 系统要启动就必须需要一个 bootloader 程序,也就说芯片上电以后先运行一段bootloader 程序。这段bootloader程序会先初始化DDR等外设,然后将Li…

零钱兑换——使用动态规划,求解——Java

322. 零钱兑换 - 力扣(LeetCode) —— 凑成总金额的最少硬币个数 使用动态规划,二维数组缓存每种价值使用的硬币数,最后得到amount价值时最少的硬币数 0 1 2 3 4 5 --> 钱值 1 0 1 …

算法-堆结构和堆排序

文章目录 本节大纲1. 堆结构2. 堆排序本节的代码实现合集 本节大纲 1. 堆结构 堆结构是为集合类里面的优先级队列来服务的 优先级队列其实就是顺序存储的二叉树结构, 我们的底层的源码里面是没有链式存储的二叉树的,二叉树的实现的细节是通过我们的数组来模拟实现的 底层的实现…

Python基础:Python中类型注解的详细介绍

Python 中的类型注解是 Python 3.5 引入的功能,主要用于提供关于变量、函数参数和函数返回值类型的额外信息。类型注解不影响 Python 程序的运行时行为,因为 Python 仍然是一种动态类型语言,注解不会强制类型检查。但它们对于代码维护、可读性、以及通过静态类型检查工具(如…

TFTP服务器

软件,客户端,服务器。是简单的文件传输文件。 1.TFTP服务器介绍 是简单的文件传输协议,是tcp/IP协议的一个用来在客户端与服务器之间进行简单文件传输的协议。端口号为69。每个服务器都有自己都端口号。 2.TFTP文件传输特点 3. 二:TFTP环境…

c++异常处理exception

// c中的异常处理 // 1.throw : 专门用于抛出异常,做出提示 // 2.try : 尝试运行可能会异常的代码 // 3.catch : 用于接收前面跑出来的异常并进行解决// 执行循序为: // try // { // throw ...; // 执行的代码中必须直接或者…

VRTK4教程 二:基本追踪

文章目录 untiyXR和UnityXRPluginFramwork使用方法: TrackedAlias使用方法使用技巧 untiyXR和UnityXRPluginFramwork 这两个用于跟踪头盔位置,其中UnityXR使用的是旧版API,另一个是新版API,两个我我们选一个即可 使用方法&#…

6.3 Go 结构体(Struct)

💝💝💝欢迎莅临我的博客,很高兴能够在这里和您见面!希望您在这里可以感受到一份轻松愉快的氛围,不仅可以获得有趣的内容和知识,也可以畅所欲言、分享您的想法和见解。 推荐:「stormsha的主页」…

【计算机毕设】SpringBoot校园资料分享平台的设计与实现 - 源码免费(私信领取)

免费领取源码 | 项目完整可运行 | v:chengn7890 诚招源码校园代理! 1. 研究目的 本项目旨在设计并实现一个基于SpringBoot的校园资料分享平台,以满足学生在学习过程中对资料分享和获取的需求。具体目标包括&#xff1a…