pkl形式的数据集读取和可视化

最新推荐文章于 2025-05-12 11:07:01 发布

F_aF_a

最新推荐文章于 2025-05-12 11:07:01 发布

阅读量1.4w

点赞数 5

CC 4.0 BY-SA版权

分类专栏： Python # 代码文章标签： python

本文链接：https://blog.csdn.net/qq_37252519/article/details/120890053

Python 同时被 2 个专栏收录

41 篇文章

订阅专栏

代码

18 篇文章

订阅专栏

这篇博客介绍了如何在命令行中下载miniImagenet数据集，并利用Python的pickle模块进行读取。通过OpenCV展示了如何保存和显示数据集中图像的例子。此外，还提供了将图像显示在屏幕上而非保存为文件的代码片段。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在命令行下载miniImagenet数据集

wget https://data.deepai.org/miniimagenet.zip

数据是pkl形式的，需要用特殊的方法进行读取和可视化
在这里插入图片描述
参考了OpenCV读取和写入图像文件（详解版）和python——pkl文件

代码如下：

import os
import pickle
import cv2

path = "/media/idea/c9ae2b5e-60d5-4916-8e02-23f584456660/DataSet/mini-imagenet/miniimagenet_pkl"

file_list = {}
for file in os.listdir(path):
    f = open(os.path.join(path, file), 'rb')
    f = pickle.load(f)
    file_list[file.split('-')[-1].split('.')[0]] = f

print(file_list)

# 展示一幅图
img = file_list['train']['image_data'][0]
cv2.imwrite("img.png", img)