import os
import matplotlib.pyplot as plt
import random
from PIL import Image
# https://blog.csdn.net/qq_44886601/article/details/135486707
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)
没有合适的资源?快使用搜索试试~ 我知道了~
温馨提示
数据集包含动物图像分类数据集(10分类),数据按照文件夹储存,不需要处理可直接用作深度学习训练数据。 数据分为以下10类:狗、猫、马、蜘蛛、蝴蝶、鸡、羊、牛、松鼠、大象(每个类别有100-200张图片) 数据集总大小:830MB 下载解压后的图像目录:训练集(2650张图片)和测试集(656张图片) data-train 训练集-每个子文件夹放同类别的图像,文件夹名为分类类别 data-test 测试集-每个子文件夹放同类别的图像,文件夹名为分类类别 除此之外,提供了classes的json字典类别文件,以及可视化的脚本py文件
资源推荐
资源详情
资源评论
收起资源包目录
深度学习之图像分类数据集:动物图像分类数据集(10分类) (2000个子文件)
Baidu_0429.bmp 425KB
Baidu_0402.jpeg 7.59MB
Baidu_0325.jpeg 7.55MB
Baidu_0324.jpeg 7.41MB
Baidu_0489.jpeg 6.82MB
Baidu_0148.jpeg 6.05MB
Baidu_0322.jpeg 6.01MB
Baidu_0143.jpeg 5.88MB
Baidu_0294.jpeg 3.32MB
Baidu_0363.jpeg 3.28MB
Baidu_0292.jpeg 3.16MB
Baidu_0460.jpeg 2.99MB
Baidu_0394.jpeg 2.89MB
Baidu_0422.jpeg 2.85MB
Baidu_0101.jpeg 2.84MB
Baidu_0145.jpeg 2.8MB
Baidu_0020.jpeg 2.76MB
Baidu_0060.jpeg 2.66MB
Baidu_0433.jpeg 2.56MB
Baidu_0344.jpeg 2.48MB
Baidu_0333.jpeg 2.29MB
Baidu_0121.jpeg 2.14MB
Baidu_0262.jpeg 1.9MB
Baidu_0132.jpeg 1.88MB
Baidu_0027.jpeg 1.87MB
Baidu_0069.jpeg 1.82MB
Baidu_0087.jpeg 1.81MB
Baidu_0131.jpeg 1.74MB
Baidu_0457.jpeg 1.73MB
Baidu_0414.jpeg 1.7MB
Baidu_0094.jpeg 1.49MB
Baidu_0441.jpeg 1.48MB
Baidu_0125.jpeg 1.48MB
Baidu_0381.jpeg 1.43MB
Baidu_0469.jpeg 1.3MB
Baidu_0371.jpeg 1.28MB
Baidu_0320.jpeg 1.21MB
Baidu_0049.jpeg 1.21MB
Baidu_0106.jpeg 1.2MB
Baidu_0302.jpeg 1.2MB
Baidu_0443.jpeg 1.14MB
Baidu_0324.jpeg 1.13MB
Baidu_0364.jpeg 1.09MB
Baidu_0113.jpeg 1.07MB
Baidu_0137.jpeg 1.06MB
Baidu_0373.jpeg 1.05MB
Baidu_0120.jpeg 1.05MB
Baidu_0447.jpeg 1.05MB
Baidu_0001.jpeg 1.04MB
Baidu_0421.jpeg 1.04MB
Baidu_0409.jpeg 1.03MB
Baidu_0468.jpeg 1.02MB
Baidu_0102.jpeg 1.01MB
Baidu_0197.jpeg 1MB
Baidu_0144.jpeg 1MB
Baidu_0046.jpeg 1021KB
Baidu_0090.jpeg 1015KB
Baidu_0016.jpeg 1004KB
Baidu_0106.jpeg 982KB
Baidu_0440.jpeg 973KB
Baidu_0098.jpeg 971KB
Baidu_0102.jpeg 953KB
Baidu_0101.jpeg 949KB
Baidu_0256.jpeg 947KB
Baidu_0132.jpeg 944KB
Baidu_0025.jpeg 939KB
Baidu_0223.jpeg 938KB
Baidu_0270.jpeg 932KB
Baidu_0139.jpeg 931KB
Baidu_0362.jpeg 926KB
Baidu_0489.jpeg 925KB
Baidu_0337.jpeg 922KB
Baidu_0164.jpeg 913KB
Baidu_0434.jpeg 905KB
Baidu_0258.jpeg 903KB
Baidu_0066.jpeg 887KB
Baidu_0423.jpeg 874KB
Baidu_0267.jpeg 861KB
Baidu_0271.jpeg 859KB
Baidu_0126.jpeg 858KB
Baidu_0099.jpeg 840KB
Baidu_0096.jpeg 838KB
Baidu_0130.jpeg 835KB
Baidu_0265.jpeg 833KB
Baidu_0184.jpeg 829KB
Baidu_0329.jpeg 827KB
Baidu_0061.jpeg 819KB
Baidu_0483.jpeg 819KB
Baidu_0095.jpeg 818KB
Baidu_0415.jpeg 810KB
Baidu_0141.jpeg 810KB
Baidu_0303.jpeg 802KB
Baidu_0107.jpeg 798KB
Baidu_0135.jpeg 797KB
Baidu_0094.jpeg 797KB
Baidu_0444.jpeg 791KB
Baidu_0000.jpeg 787KB
Baidu_0439.jpeg 786KB
Baidu_0046.jpeg 785KB
Baidu_0437.jpeg 785KB
共 2000 条
- 1
- 2
- 3
- 4
- 5
- 6
- 20
资源评论
听风吹等浪起
- 粉丝: 1w+
- 资源: 1012
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功