Skip to content

A convolutional neural network to classify mushroom species from the Common genus's images Kaggle dataset. Based on a pretained ResNet50 network, finetuned with fast.ai v2.

Notifications You must be signed in to change notification settings

valdas-v1/mushrooms-classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Mushrooms classifier

For this project, I have built a convolutional neural network to classify mushroom species from the Common genus's images Kaggle dataset. The network is based on a pretained ResNet50 network, finetuned with fast.ai v2.

This project includes:

  1. EDA of the initial Kaggle dataset
  2. Oulier analysis
  3. Image preprocessing
  4. Building and finetuning a pretrained ResNet50 convolutional neural network
  5. Model evaluation using confusion matrices and classification reports

Technology

About

A convolutional neural network to classify mushroom species from the Common genus's images Kaggle dataset. Based on a pretained ResNet50 network, finetuned with fast.ai v2.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published