Keras-常用代码

简介

本文主要介绍一些使用Keras过程中使用频率较高的常用代码段。

表格文件快速生成数据集

from keras.preprocessing.image import ImageDataGeneratortrain_gen = ImageDataGenerator(rescale=1/255., validation_split=0.2, horizontal_flip=False, shear_range=0.2, width_shift_range=0.1)
test_gen = ImageDataGenerator(rescale=1/255.)
img_size = (224, 224)
batch_size = 32
train_generator = train_gen.flow_from_dataframe(dataframe=df_train,directory='data/train',x_col='file_id',y_col='accent',batch_size=batch_size,class_mode='categorical',target_size=img_size, subset='training')
valid_generator = train_gen.flow_from_dataframe(dataframe=df_train,directory="data/train",x_col="file_id",y_col="accent",batch_size=batch_size,class_mode="categorical",    target_size=img_size,subset='validation')
test_generator = test_gen.flow_from_dataframe(dataframe=df_test,directory = "data/test",x_col="file_id",target_size=img_size,batch_size=1,shuffle=False,class_mode=None)

预训练模型使用

from keras.applications.densenet import DenseNet121
from keras.models import Model
from keras.layers import GlobalAveragePooling2D, Input, Dropout, Dense, BatchNormalization
from keras.optimizers import Adamdef build_densenet(input_shape=(224, 224, 3), n_classes=3):input_layer = Input(shape=input_shape)densenet121 = DenseNet121(include_top=False, weights='imagenet', input_tensor=input_layer)x = GlobalAveragePooling2D()(densenet121.output)x = Dropout(0.5)(x)x = Dense(n_classes, activation='softmax')(x)model = Model(input_layer, x)model.compile(loss='categorical_crossentropy', optimizer=Adam(lr=3e-4), metrics=['accuracy'])return modeldensenet = build_densenet()

Keras-常用代码相关推荐

pytorch常用代码
20211228 https://mp.weixin.qq.com/s/4breleAhCh6_9tvMK3WDaw 常用代码段本文代码基于 PyTorch 1.x 版本,需要用到以下包: impo ...
收藏！PyTorch常用代码段合集
↑↑↑关注后"星标"Datawhale 每日干货 & 每月组队学习,不错过 Datawhale干货作者:Jack Stark,来源:极市平台来源丨https://zhu ...
PyTorch常用代码段合集
↑ 点击蓝字关注视学算法作者丨Jack Stark@知乎来源丨https://zhuanlan.zhihu.com/p/104019160 极市导读本文是PyTorch常用代码段合集,涵盖基本 ...
正则化的通俗解释_干货|深度学习中的正则化技术概述（附Python+keras实现代码）...
点击上方"AI遇见机器学习",选择"星标"公众号重磅干货,第一时间送达介绍数据科学研究者们最常遇见的问题之一就是怎样避免过拟合.你也许在训练模型的时候也遇 ...
【深度学习】PyTorch常用代码段合集
来源 | 极市平台,机器学习算法与自然语言处理本文是PyTorch常用代码段合集,涵盖基本配置.张量处理.模型定义与操作.数据处理.模型训练与测试等5个方面,还给出了多个值得注意的Tips,内容非常 ...
收藏 | PyTorch常用代码段合集
点上方蓝字计算机视觉联盟获取更多干货在右上方 ··· 设为星标 ★,与你不见不散仅作学术分享,不代表本公众号立场,侵权联系删除转载于:作者丨Jack Stark@知乎来源丨https://zh ...
神经网络学习小记录45——Keras常用学习率下降方式汇总
神经网络学习小记录45--Keras常用学习率下降方式汇总 2020年5月19日更新前言为什么要调控学习率下降方式汇总 1.阶层性下降 2.指数型下降 3.余弦退火衰减 4.余弦退火衰减更新版 ...
TensorFlow—Keras常用API
TensorFlow-Keras常用API 文章目录 TensorFlow-Keras常用API 前言一.模型自定义 1. keras.Sequential 2.keras.layers.Layer ...
深度盘点：PyTorch常用代码段合集
本文是PyTorch常用代码段合集,涵盖基本配置.张量处理.模型定义与操作.数据处理.模型训练与测试等5个方面,还给出了多个值得注意的Tips,内容非常全面. PyTorch最好的资料是官方文档.本文 ...
PyTorch 常用代码段整理合集
PyTorch 常用代码段整理合集来源:知乎作者:张皓众所周知,程序猿在写代码时通常会在网上搜索大量资料,其中大部分是代码段.然而,这项工作常常令人心累身疲,耗费大量时间.所以,今天小编转载了知 ...

Keras-常用代码

简介

表格文件快速生成数据集

预训练模型使用

Keras-常用代码相关推荐

最新文章

热门文章