【054】Keras 垃圾分类图片识别项目

最新推荐文章于 2024-04-23 09:42:35 发布

小麦粒

最新推荐文章于 2024-04-23 09:42:35 发布

阅读量3.8k

点赞数 5

本文链接：https://blog.csdn.net/u010986753/article/details/105463469

版权

整个项目使用tensorflow-GPU跑的

1、数据集下载

链接：https://pan.baidu.com/s/1ywEV3_CXEIgxvPh6g3BX5Q

提取码：39zm

2、随机展示6张图片

import time
start = time.time()
import numpy as np
import matplotlib.pyplot as plt
import keras
import os,glob,random
from keras.models import Sequential
from keras.layers import Conv2D,Flatten,MaxPooling2D,Dense,AveragePooling2D
from keras.preprocessing.image import ImageDataGenerator,load_img,img_to_array,array_to_img

#1、数据位置
base_path = 'D:\A\AI-master\garbage_class\dataset-resized'
#glob.glob获取指定目录下的所有图片
img_list = glob.glob(os.path.join(base_path,'*/*.jpg')) #加上r让字符串不转义

print(len(img_list))#2527

#2、随机展示六张图片

for i ,img_path in enumerate(random.sample(img_list,6)):
    img = load_img(img_path)
    plt.subplot(2,3,i+1)

    plt.imshow(img)
    plt.show()

3、对数据进行分组

train_datagen = ImageDataGenerator(rescale=1./255,shear_range=0.1,zoom_range=0.1,
                                   width_shift_range=0.1,height_shift_range=0.1,horizontal_flip=True,
                                   vertical_flip=True