Skip to content

ADITYASHAH-IITP/Image-Classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

16 Commits
 
 
 
 

Repository files navigation

Image-Classification

CIFAR-10 Photo Classification

CIFAR-10_Dataset_Classfication


CIFAR-10 Photo Classification Dataset CIFAR is an acronym that stands for the Canadian Institute For Advanced Research and the CIFAR-10 dataset was developed along with the CIFAR-100 dataset by researchers at the CIFAR institute.

The dataset is comprised of 60,000 32×32 pixel color photographs of objects from 10 classes, such as frogs, birds, cats, ships, etc. The class labels and their standard associated integer values are listed below.

0: airplane
1: automobile
2: bird
3: cat
4: deer
5: dog
6: frog
7: horse
8: ship
9: truck

Model Summary




image


Result


image
image

Saving the entire mode



from tensorflow.keras.callbacks import ModelCheckpoint
checkpoint_path='/content/model_checkpoints'
checkpoint=ModelCheckpoint(filepath=checkpoint_path,frequency='epoch',save_weights_only=False,verbose=2)
to delete model --> del model
from tensorflow.keras.models import load_model
model=load_model(checkpoint_path)
manual save --> model.save('my_model') or model.save('my_model.h5')