Download as pdf or txt
Download as pdf or txt
You are on page 1of 2

11/25/21, 9:58 AM Gmail - (no subject)

VISHNU VARDHAN <421pvv@gmail.com>

(no subject)

2 messages

siddardha msd7183 <siddardhamsd7183@gmail.com> Wed, Nov 24, 2021 at 1:02 PM


To: 421pvv@gmail.com

# example of progressively loading images from file

from keras.preprocessing.image import ImageDataGenerator

from keras.applications.vgg16 import VGG16

from tensorflow.keras.utils import to_categorical

from tensorflow.keras.optimizers import RMSprop

from keras.layers import Reshape

from keras.models import Model

import tensorflow as tf

# create generator

datagen= ImageDataGenerator()

train = ImageDataGenerator(rescale=1/255)

validation= ImageDataGenerator(rescale=1/255)

# prepare an iterators for each dataset

train_it = train.flow_from_directory('data/train/', class_mode='binary',target_size=(200,200),batch_size=64)

val_it = validation.flow_from_directory('data/validate/', class_mode='binary',target_size=(200,200),batch_size=64)

test_it = datagen.flow_from_directory('data/test/', class_mode='binary',target_size=(200,200),batch_size=64)

# confirm the iterator works

trainX, trainy = train_it.next()

testX, testy=test_it.next()

trainy=to_categorical(trainy)

testy=to_categorical(testy)

print('Batch shape=%s, min=%.3f, max=%.3f' % (trainX.shape, trainX.min(), trainX.max()))

print("X=%s,y=%s"%(trainX.shape,trainy.shape))

print("X=%s,y=%s"%(testX.shape,testy.shape))

model=tf.keras.models.Sequantial([tf.keras.layers.Conv2D(16,(3,3),activation='relu',input_shape=(200,200,3)),

                                 tf.keras.layers.MaxPool2D(2,2),

                                  #

                                  tf.keras.layers.Conv2D(32,(3,3),activation='relu'),

                                 tf.keras.layers.MaxPool2D(2,2),

                                  #

                                  tf.keras.layers.Conv2D(64,(3,3),activation='relu'),

                                 tf.keras.layers.MaxPool2D(2,2),

                                  #

                                  tf.keras.layers.Flatten(),

                                  ##

                                  tf.keras.layers.Dense(512,activation='relu'),

                                  ##

                                  tf.keras.layers.Dense(1,activation='sigmoid')

      

                               ])

model.compile(loss='binary_crossentropy',

             optimizer = RMSprop(learning_rate = 0.001),

             metrics=['accuracy'])

model_fit = model.fit(train_it,

                     steps_per_epoch=3,

                     epochs = 2,

                     validation_data=val_it)

after this 
we have to predict using predict_it method. 

VISHNU VARDHAN <421pvv@gmail.com> Wed, Nov 24, 2021 at 5:37 PM


To: siddardha msd7183 <siddardhamsd7183@gmail.com>

Thank you so much


[Quoted text hidden]

https://mail.google.com/mail/u/0/?ik=586392e13b&view=pt&search=all&permthid=thread-f%3A1717293974368537850&simpl=msg-f%3A171729… 1/2
11/25/21, 9:58 AM Gmail - (no subject)

https://mail.google.com/mail/u/0/?ik=586392e13b&view=pt&search=all&permthid=thread-f%3A1717293974368537850&simpl=msg-f%3A171729… 2/2

You might also like