import os
import matplotlib.pyplot as plt
import random
from PIL import Image
# https://blog.csdn.net/qq_44886601/article/details/135486707?spm=1001.2014.3001.5502
def main(path):
classes = [i for i in os.listdir(path)] # ['cat', 'dog']
# 将所有图片按照 类别:路径 字典形式保存
images_path = [] # [{'cat': './data/train\\cat\\Baidu_0000.jpeg'}, {'cat': './data/train\\cat\\Baidu_0002.jpeg'}]
for cla in classes:
for i in os.listdir(os.path.join(path, cla)):
dic = {} # 类别:图像路径
img_path = os.path.join(path, cla, i)
dic[cla] = img_path # {'cat': './data/train\\cat\\Baidu_0000.jpeg'}
images_path.append(dic)
# 随机展示4张图像
plt.figure(figsize=(12, 8))
for i in range(4):
r = random.randint(0, len(images_path) - 1) # 生成随机数
label, im_path = list(images_path[r].keys())[0], list(images_path[r].values())[0]
# cat , ./data/train\cat\Baidu_0049.jpeg
im = Image.open(im_path)
plt.subplot(2, 2, i + 1)
plt.title(label)
plt.imshow(im)
plt.savefig('show.png') # 保存图片
plt.show()
if __name__ == '__main__':
root = './data/train' # 传入目录
main(path=root)
没有合适的资源?快使用搜索试试~ 我知道了~
温馨提示
数据集包含5种大型哺乳动物图像分类数据集(5分类),数据按照文件夹储存,不需要处理可直接用作深度学习训练数据。 数据分为以下5类:大象、狮子、马、猫、狗 数据集总大小:468MB 下载解压后的图像目录:训练集(1386张图片)和测试集(342张图片) data-train 训练集-每个子文件夹放同类别的图像,文件夹名为分类类别 data-test 测试集-每个子文件夹放同类别的图像,文件夹名为分类类别 除此之外,提供了classes的json字典类别文件,以及可视化的脚本py文件
资源推荐
资源详情
资源评论
收起资源包目录
计算机视觉之分图像类数据集:5种大型哺乳动物图像分类数据集(5分类) (1731个子文件)
Baidu_0007.bmp 768KB
Baidu_0470.jpeg 4.32MB
Baidu_0148.jpeg 3.66MB
Baidu_0110.jpeg 2.96MB
Baidu_0365.jpeg 2.87MB
Baidu_0119.jpeg 2.84MB
Baidu_0121.jpeg 2.8MB
Baidu_0406.jpeg 2.76MB
Baidu_0138.jpeg 2.66MB
Baidu_0098.jpeg 2.55MB
Baidu_0104.jpeg 2.42MB
Baidu_0391.jpeg 2.29MB
Baidu_0121.jpeg 2.14MB
Baidu_0085.jpeg 2.08MB
Baidu_0111.jpeg 1.99MB
Baidu_0100.jpeg 1.94MB
Baidu_0017.jpeg 1.87MB
Baidu_0445.jpeg 1.86MB
Baidu_0094.jpeg 1.82MB
Baidu_0103.jpeg 1.81MB
Baidu_0071.jpeg 1.74MB
Baidu_0389.jpeg 1.73MB
Baidu_0374.jpeg 1.73MB
Baidu_0108.jpeg 1.71MB
Baidu_0132.jpeg 1.48MB
Baidu_0430.jpeg 1.43MB
Baidu_0087.jpeg 1.39MB
Baidu_0498.jpeg 1.36MB
Baidu_0416.jpeg 1.36MB
Baidu_0136.jpeg 1.3MB
Baidu_0231.jpeg 1.21MB
Baidu_0379.jpeg 1.2MB
Baidu_0174.jpeg 1.2MB
Baidu_0485.jpeg 1.19MB
Baidu_0417.jpeg 1.17MB
Baidu_0001.jpeg 1.16MB
Baidu_0302.jpeg 1.12MB
Baidu_0421.jpeg 1.11MB
Baidu_0128.jpeg 1.11MB
Baidu_0292.jpeg 1.1MB
Baidu_0433.jpeg 1.1MB
Baidu_0326.jpeg 1.08MB
Baidu_0084.jpeg 1.07MB
Baidu_0081.jpeg 1.06MB
Baidu_0250.jpeg 1.05MB
Baidu_0326.jpeg 1.04MB
Baidu_0366.jpeg 1.03MB
Baidu_0393.jpeg 1.02MB
Baidu_0336.jpeg 1.02MB
Baidu_0442.jpeg 1.02MB
Baidu_0194.jpeg 1MB
Baidu_0491.jpeg 1021KB
Baidu_0127.jpeg 1015KB
Baidu_0014.jpeg 1004KB
Baidu_0075.jpeg 1000KB
Baidu_0142.jpeg 982KB
Baidu_0453.jpeg 973KB
Baidu_0312.jpeg 953KB
Baidu_0072.jpeg 944KB
Baidu_0277.jpeg 938KB
Baidu_0336.jpeg 937KB
Baidu_0265.jpeg 932KB
Baidu_0082.jpeg 931KB
Baidu_0093.jpeg 926KB
Baidu_0452.jpeg 925KB
Baidu_0316.jpeg 919KB
Baidu_0458.jpeg 919KB
Baidu_0290.jpeg 913KB
Baidu_0329.jpeg 900KB
Baidu_0114.jpeg 899KB
Baidu_0330.jpeg 898KB
Baidu_0376.jpeg 887KB
Baidu_0278.jpeg 861KB
Baidu_0066.jpeg 858KB
Baidu_0299.jpeg 858KB
Baidu_0117.jpeg 853KB
Baidu_0375.jpeg 850KB
Baidu_0484.jpeg 843KB
Baidu_0481.jpeg 842KB
Baidu_0296.jpeg 840KB
Baidu_0133.jpeg 838KB
Baidu_0084.jpeg 833KB
Baidu_0211.jpeg 829KB
Baidu_0149.jpeg 827KB
Baidu_0388.jpeg 821KB
Baidu_0326.jpeg 819KB
Baidu_0216.jpeg 819KB
Baidu_0450.jpeg 814KB
Baidu_0149.jpeg 813KB
Baidu_0030.jpeg 813KB
Baidu_0086.jpeg 810KB
Baidu_0077.jpeg 797KB
Baidu_0113.jpeg 797KB
Baidu_0472.jpeg 791KB
Baidu_0442.jpeg 785KB
Baidu_0000.jpeg 783KB
Baidu_0144.jpeg 773KB
Baidu_0162.jpeg 769KB
Baidu_0409.jpeg 766KB
Baidu_0374.jpeg 766KB
共 1731 条
- 1
- 2
- 3
- 4
- 5
- 6
- 18
资源评论
听风吹等浪起
- 粉丝: 1w+
- 资源: 1239
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功