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

在命令行下载miniImagenet数据集

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

数据是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)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18

另外如果没用用服务器，黄框代码可以改成

cv2.imshow("img",img)
1