在命令行下载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)
另外如果没用用服务器,黄框代码可以改成
cv2.imshow("img",img)