Skip to content

Instantly share code, notes, and snippets.

View HengJayWang's full-sized avatar

HengJay HengJayWang

View GitHub Profile
@HengJayWang
HengJayWang / Transfer_Learning_Keras_01.py
Created September 20, 2017 05:44 — forked from prakashjayy/Transfer_Learning_Keras_01.py
Transfer Learning using Keras
from keras import applications
from keras.preprocessing.image import ImageDataGenerator
from keras import optimizers
from keras.models import Sequential, Model
from keras.layers import Dropout, Flatten, Dense, GlobalAveragePooling2D
from keras import backend as k
from keras.callbacks import ModelCheckpoint, LearningRateScheduler, TensorBoard, EarlyStopping
img_width, img_height = 256, 256
train_data_dir = "data/train"