My first project in AI/ML! Hii Everyone I am diving deep into AI and ML. This project on Multi Class Animal Classification is created on jupyter notebook. I gained valuable experiences into the various Python Libraries. I gained hands-on Python libraries including: -> Numpy -> Matplotlib -> TensorFlow and its Keras API. The dataset was gathered from Kaggle which helped me in the project. I also used the Pyplot module of the Matplotlib Library to plot the images and show them on the screen. Then for the Data Preprocessing part -> I defined the image size and the batch size of the images. ImageDataGenerator is used for rescaling and validation splitting. Created Training and Validation Generators. Mapped the classes using the train_generator and extracted the class names Then I used a pre-trained model MobileNetV2 that helped in the project Used 20 epochs to train the model Then plotted the training and validation Accuracy and Loss At last I used the Keras library for the image generation
-
Notifications
You must be signed in to change notification settings - Fork 0
dhruvXcode247/Multi_class_animal_classification
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
My first project in AI/ML from the INTERNSHIP
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published