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)
没有合适的资源?快使用搜索试试~ 我知道了~
温馨提示
数据集包含正方形、圆形、三角形数学图案分类数据集(3分类),数据按照文件夹储存,不需要处理可直接用作深度学习训练数据。 数据分为以下3类正方形、圆形、三角形(每个类别均有100张图片) 数据集总大小:170KB 下载解压后的图像目录:训练集(240张图片)和测试集(60张图片) data-train 训练集-每个子文件夹放同类别的图像,文件夹名为分类类别 data-test 测试集-每个子文件夹放同类别的图像,文件夹名为分类类别 除此之外,提供了classes的json字典类别文件,以及可视化的脚本py文件
资源推荐
资源详情
资源评论
收起资源包目录
深度学习之图像分类数据集:正方形、圆形、三角形数学图案分类数据集(3分类) (303个子文件)
class_indices.json 68B
show.png 23KB
drawing(31).png 766B
drawing(88).png 748B
drawing(100).png 729B
drawing(9).png 728B
drawing(29).png 725B
drawing(33).png 713B
drawing(22).png 712B
drawing(68).png 704B
drawing(86).png 692B
drawing(24).png 664B
drawing(62).png 664B
drawing(53).png 654B
drawing(7).png 654B
drawing(94).png 647B
drawing(61).png 644B
drawing(92).png 641B
drawing(13).png 640B
drawing(11).png 635B
drawing(67).png 633B
drawing(97).png 633B
drawing(96).png 632B
drawing(32).png 627B
drawing(73).png 627B
drawing(71).png 625B
drawing(55).png 625B
drawing(53).png 622B
drawing(26).png 620B
drawing(56).png 620B
drawing(65).png 619B
drawing(13).png 619B
drawing(34).png 619B
drawing(89).png 617B
drawing(5).png 617B
drawing(16).png 614B
drawing(8).png 613B
drawing(64).png 612B
drawing(28).png 611B
drawing(54).png 610B
drawing(44).png 608B
drawing(52).png 606B
drawing(93).png 604B
drawing(1).png 603B
drawing(69).png 603B
drawing(90).png 599B
drawing(66).png 597B
drawing(78).png 593B
drawing(59).png 592B
drawing(82).png 592B
drawing(3).png 590B
drawing(11).png 584B
drawing(58).png 583B
drawing(43).png 582B
drawing(10).png 580B
drawing(48).png 579B
drawing(72).png 575B
drawing(57).png 571B
drawing(58).png 571B
drawing(7).png 570B
drawing(26).png 570B
drawing(48).png 570B
drawing(68).png 570B
drawing(4).png 569B
drawing(86).png 569B
drawing(80).png 568B
drawing(13).png 567B
drawing(85).png 566B
drawing(4).png 565B
drawing(95).png 565B
drawing(89).png 564B
drawing(42).png 563B
drawing(59).png 562B
drawing(78).png 561B
drawing(25).png 561B
drawing(5).png 561B
drawing(23).png 559B
drawing(25).png 559B
drawing(90).png 558B
drawing(71).png 558B
drawing(26).png 558B
drawing(74).png 556B
drawing(87).png 556B
drawing(17).png 555B
drawing(84).png 555B
drawing(93).png 554B
drawing(91).png 554B
drawing(70).png 554B
drawing(94).png 553B
drawing(77).png 552B
drawing(51).png 551B
drawing(30).png 551B
drawing(79).png 550B
drawing(91).png 550B
drawing(97).png 550B
drawing(42).png 547B
drawing(75).png 547B
drawing(90).png 547B
drawing(28).png 547B
drawing(76).png 546B
共 303 条
- 1
- 2
- 3
- 4
资源评论
听风吹等浪起
- 粉丝: 1w+
- 资源: 1289
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功