import os
import matplotlib.pyplot as plt
import random
from PIL import Image
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,cmap='gray')
plt.savefig('show.png') # 保存图片
# plt.show()
if __name__ == '__main__':
root = './data/train' # 传入目录
main(path=root)
没有合适的资源?快使用搜索试试~ 我知道了~
温馨提示
数据包含:金属表面缺陷识别10分类【包括划分好的数据、类别字典文件、python数据可视化脚本】 【分类个数:10分类】冲孔、熔接线、月牙形间隙、水斑、油斑、丝斑、夹杂物、轧坑、折痕、腰部折叠 【数据集详情】data目录下分为2个目录,训练集和验证集,存放各自的同一类数据图片。训练集图片总数1620,验证集图片总数686。可以用作yolov5的分类数据集. 【json文件】10 种分类的字典文件 为了方便查看数据,提供了可视化py文件,随机传入4张图片即可展示,并且保存在当前目录。 脚本无需更改,可以直接运行!
资源推荐
资源详情
资源评论
收起资源包目录
图像分类数据集:金属表面缺陷识别10分类【包括划分好的数据、类别字典文件、python数据可视化脚本】 (2000个子文件)
img_02_3402618000_00007.jpg 875KB
img_03_3402618000_00013.jpg 765KB
img_02_4402329100_00006.jpg 744KB
img_03_3402618000_00007.jpg 741KB
img_07_4404897500_00875.jpg 720KB
img_02_436068400_00001.jpg 720KB
img_02_3402618000_00002.jpg 718KB
img_02_3437009400_00001.jpg 704KB
img_07_4405377700_00912.jpg 699KB
img_07_3403335200_00806.jpg 685KB
img_07_4402625300_00105.jpg 681KB
img_07_4406743300_00001.jpg 678KB
img_02_SIS002356_00001.jpg 666KB
img_06_3403334000_00839.jpg 660KB
img_02_4402116500_00005.jpg 659KB
img_02_425505100_00022.jpg 658KB
img_02_4402329000_00001.jpg 655KB
img_02_4402116700_00002.jpg 647KB
img_02_4403955000_00001.jpg 646KB
img_07_3403406200_01076.jpg 642KB
img_07_4405381400_00773.jpg 638KB
img_06_4406743300_00001.jpg 637KB
img_02_436068500_00002.jpg 633KB
img_03_3437006900_00002.jpg 627KB
img_07_436164700_01555.jpg 625KB
img_02_3402618000_00010.jpg 625KB
img_07_3403405500_00670.jpg 622KB
img_06_4404997500_01006.jpg 614KB
img_04_425501800_00017.jpg 609KB
img_03_425502200_00017.jpg 609KB
img_07_4404711100_00787.jpg 607KB
img_06_3403333000_00659.jpg 606KB
img_07_436164700_01552.jpg 605KB
img_07_4404893200_00890.jpg 605KB
img_07_3403338100_00785.jpg 599KB
img_07_436164700_01551.jpg 598KB
img_07_3403405200_00719.jpg 596KB
img_07_4404893900_00568.jpg 596KB
img_02_431854300_00002.jpg 592KB
img_06_4402625300_01263.jpg 590KB
img_07_3403334600_00682.jpg 589KB
img_02_427146700_00002.jpg 588KB
img_07_436164700_01537.jpg 587KB
img_07_3436785900_00004.jpg 586KB
img_02_425508400_00140.jpg 585KB
img_02_436152900_00001.jpg 584KB
img_02_425508400_00141.jpg 584KB
img_07_3403406000_00825.jpg 583KB
img_07_436164700_01538.jpg 583KB
img_07_436164700_01545.jpg 582KB
img_02_SIS001522_00028.jpg 582KB
img_02_4406743300_00001.jpg 581KB
img_07_430103100_01150.jpg 581KB
img_07_3403335600_00795.jpg 580KB
img_07_4405168700_00710.jpg 579KB
img_07_4406645900_00531.jpg 579KB
img_06_4402625300_01262.jpg 579KB
img_07_3436814000_00684.jpg 579KB
img_02_4403466600_00003.jpg 578KB
img_06_3403400800_00948.jpg 577KB
img_06_3436642500_00002.jpg 577KB
img_02_SIS001522_00026.jpg 577KB
img_07_4404710300_00877.jpg 576KB
img_02_SIS001577_00072.jpg 574KB
img_06_4402852500_01045.jpg 574KB
img_02_4403466200_00006.jpg 574KB
img_07_4406645900_00530.jpg 573KB
img_02_SIS001577_00067.jpg 573KB
img_07_3403403800_00910.jpg 571KB
img_02_436149500_00939.jpg 571KB
img_07_3436786500_00002.jpg 571KB
img_07_4406645900_00527.jpg 571KB
img_06_4402625300_00109.jpg 569KB
img_07_4406645900_00526.jpg 569KB
img_07_3403338900_00683.jpg 569KB
img_07_3403334300_00854.jpg 568KB
img_07_3436814000_00692.jpg 568KB
img_07_4406645900_00539.jpg 568KB
img_07_4404374100_01367.jpg 568KB
img_02_4402818600_00001.jpg 567KB
img_07_4405168700_00711.jpg 567KB
img_07_4406645900_00523.jpg 567KB
img_02_4404612700_00054.jpg 567KB
img_07_3436814000_00693.jpg 566KB
img_07_4406645900_00522.jpg 566KB
img_01_3436789500_00004.jpg 566KB
img_07_3403405800_00821.jpg 566KB
img_02_SIS001522_00036.jpg 566KB
img_02_425507000_00931.jpg 565KB
img_07_4406645900_00001.jpg 564KB
img_07_4406444100_00004.jpg 564KB
img_02_425502300_00338.jpg 562KB
img_07_436164700_01536.jpg 562KB
img_07_4402719500_00001.jpg 562KB
img_02_425507000_00872.jpg 562KB
img_07_4406645900_00518.jpg 562KB
img_07_4404374100_01348.jpg 561KB
img_02_425508400_00382.jpg 561KB
img_02_425507000_00871.jpg 561KB
img_08_4405381400_00773.jpg 561KB
共 2000 条
- 1
- 2
- 3
- 4
- 5
- 6
- 20
资源评论
听风吹等浪起
- 粉丝: 1w+
- 资源: 1329
下载权益
C知道特权
VIP文章
课程特权
开通VIP
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
最新资源
- 基于单片机的便携式粮食水分测试仪的研究
- 测绘基坑支护工程变形监测报告1.pdf
- 基于PHP+swoole实现的微信机器人,依赖vbot和微信网页版的功能,帮助管理微信群/聊天/踢人等+源码+开发文档+运行教程
- com.xunmeng.pinduoduo_Release_cd290ca9_ARM64.apk
- 2788727d-25a0-41b2-b6b4-265d193edb95.doc
- 基于AVR单片机的伺服电机系统研究
- Lab-Electronic Craft Practicum-2-Simulation of a Single Tube Com
- 贪吃蛇基于TypeScript
- CS-CP1-2C3WF固件
- 软件测试测试用例设计方法大全
资源上传下载、课程学习等过程中有任何疑问或建议,欢迎提出宝贵意见哦~我们会及时处理!
点击此处反馈
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功