使用Tensorflow2.0和Keras训练分类模型实例

本文通过实例详细介绍了如何利用Python的Tensorflow2.0和Keras库训练一个分类模型,涵盖了深度学习和计算机视觉的基本应用。
摘要由CSDN通过智能技术生成
1.训练过程:

from tensorflow.keras.preprocessing.image import ImageDataGenerator
from tensorflow.keras.applications.xception import Xception
from tensorflow.keras.layers import GlobalAveragePooling2D, GlobalMaxPooling2D, Dense, Flatten, Input, Concatenate, Dropout
from tensorflow.keras.models import Model

import tensorflow as tf
from tensorflow.keras.callbacks import ModelCheckpoint,ReduceLROnPlateau,TensorBoard, EarlyStopping
from sklearn.utils import class_weight
import numpy as np
# 下面的一部分是进行数据预处理
import os
import shutil

def image_number(dir_path):
    path = []
    for dir, file, images in os.walk(dir_path):
        for image in images:
            if image.endswith(".bmp") or image.endswith(".jpg")or image.endswith(".jpeg"):
                imagepath = os.path.join(dir, image)
                path.append(imagepath)
    return path

training_path=r".\training"
eval_path =r".\validation"
test_path=r".\test"
MODEL_path=r".\model"
log_path=r".\log"
training_image_path = image_number(training_path)  # 训练集数据路径列表

val_image_path = image_number(eval_path)
#test_image_path=image_number(test_path)
label=["defect","dust","ok"]
#label=["defect","ok"]
training_image_number = len(training_image_path)  # 训练集数据量
val_image_number = len(val_image_path)
#test_image_number=len(test_image_path)
classfication_number = len(label)  # 分类数
batch=16
train_epoch = (training_image_number // batch)

val_epoch = (val_ima
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值