AITS Journal
Published in

AITS Journal


Photo by July PJuxa (Puchkova) on Dribbble

Fishes also known as Ichthyology , accounts for the majority of Sea Life on our planet . They range from small centimetre to many meters .It can be even said that is one of the most diversified species on the planet that we mere humans have not explored .Distinguishing them is a task that even puzzle experts let alone a normal person .But that’s Exactly what we will try to accomplish here .


  • Problem Statement
  • Library and Concepts
  • Methodology
  • Modelling
  • Kernel link


We will be making a model to predict the fish breed of the image passed to it .We will be classifying the fish into one of 9 different categories .


Convolutional neural network

Image Augmentation



Matplotlib / Seaborn


Our dataset contains 3 channels images of shape 288x384 . We will be using multiple Convolutional layers followed by a Max Pool layer with Dropout layer to keep our model generalized and prevent overfitting .We will be also using Image Augmentation on features such as rescaling ,horizontal and vertical flips . This increases our dataset size and adds a bit of randomness to our model .


We will be using Convolution layers of 32 filters of size 3x3 at start followed by MaxPooling of size 2x2 accompanied by Dropout layer . As we go down deep in the model we increase the number of filter to 64 to allow our model to learn complex features .

To pass the Images into the Model we will be using Keras flow from dataframe .This allows our model to handle images easily and apply transformation with ease rather than getting exhausted by applying the whole dataset in one go .

Accuracy over training
Loss over Training

We are achieving about 80 % accuracy .


Kernel Link

Credit : Devansh Chowdhury




AI content from AITS associates with 💝. AITS is a deep learning company and lead developer of open source deep learning compiler.

Recommended from Medium

Building an AI-Powered Searchable Video Archive

Detection of Autism Spectrum Disorder in Newborns Using Biomarkers Early After Birth and Logistic…

Prediction of Surface Roughness Using Artificial Neural Network(ANN).

Role of Data in Implicit Regularization

Introduction to Reinforcement Learning (Coding Q-Learning) — Part 3

Simple scalable graph neural networks

Understanding how LIME explains predictions

Well… let's start with Machine Learning (using Bigquery?

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store
AI Technology & Systems

AI Technology & Systems

Simplifying AI development for EDGE devices

More from Medium

Detecting Malaria with Deep Learning

Decision Boundaries using Support Vector Machine (SVM)

Neural Networks

Understanding Image Classification — Basics