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

在命令行下载miniImagenet数据集

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

数据是pkl形式的,需要用特殊的方法进行读取和可视化
pkl形式的数据集读取和可视化_第1张图片
参考了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)

pkl形式的数据集读取和可视化_第2张图片

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

cv2.imshow("img",img)

你可能感兴趣的:(Python,#,代码,python)