Skip to content

Image Classification and Image Segmentation projects developed during the Artificial Neural Network & Deep Learning course at PoliMi

Notifications You must be signed in to change notification settings

GiorCocc/Deep-Learning-Uni-Projects

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

7 Commits
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

🧠 Deep Learning University Projects

This repository contains two comprehensive deep learning projects developed for the Artificial Neural Networks and Deep Learning course, showcasing advanced techniques in computer vision and neural network architectures.

🩸 Project 1: Blood Cell Image Classification

Objective: Multi-class classification of blood cell images into 8 categories
Architecture: MobileNetV3Large with custom classifier layers
Key Results:

  • βœ… 97.59% accuracy on internal test set
  • πŸ“Š 0.67 benchmark score
  • πŸ”¬ Processed 11,738 blood cell images (96x96 pixels)

πŸ“‚ View Project Details


πŸ”΄ Project 2: Mars Terrain Segmentation

Objective: Semantic segmentation of Martian terrain into 5 classes (background, soil, bedrock, sand, big rock)
Architecture: Dual U-Net with attention mechanisms and ensemble approach
Key Results:

  • 🎯 0.52828 benchmark score
  • πŸš€ Advanced U-Net architectures with custom modules
  • 🌍 Processed 2,505 Mars terrain images (64x128 pixels)

πŸ“‚ View Project Details


πŸ“‹ Repository Structure

Deep-Learning-Uni-Projects/
β”œβ”€β”€ πŸ“ Image Classification/           # Blood cell classification project
β”‚   β”œβ”€β”€ πŸ““ Notebook Homework 1.ipynb   # Main implementation notebook
β”‚   β”œβ”€β”€ πŸ“„ AN2DL_Homeworks_Report.pdf  # Detailed project report
β”‚   └── πŸ“Š training_set.npz            # Blood cell dataset
β”œβ”€β”€ πŸ“ Image Segmentation/             # Mars terrain segmentation project  
β”‚   β”œβ”€β”€ πŸ““ anndl-homework-2.ipynb      # Main implementation notebook
β”‚   β”œβ”€β”€ πŸ““ big-rock-specialized-model.ipynb # Specialized model for big rocks
β”‚   β”œβ”€β”€ πŸ““ ensemble-experiment.ipynb   # Ensemble approach experiments
β”‚   β”œβ”€β”€ πŸ“„ AN2DL_Homework_2_Report.pdf # Detailed project report
β”‚   └── πŸ“Š mars_for_students.npz       # Mars terrain dataset
└── πŸ“– README.md                       # This file

🎯 Key Highlights

  • State-of-the-art Architectures: Implementation of MobileNetV3Large and advanced U-Net models
  • Custom Modules: Squeeze-and-Excitation blocks, attention mechanisms, and cellular automata
  • Data Handling: Comprehensive preprocessing, augmentation, and class balancing strategies
  • Performance Optimization: Advanced techniques including focal loss, ensemble methods, and fine-tuning
  • Real-world Applications: Medical imaging and space exploration computer vision tasks

πŸ› οΈ Technologies Used

  • Deep Learning: TensorFlow/Keras
  • Computer Vision: OpenCV, PIL
  • Data Science: NumPy, Pandas, Matplotlib
  • Development: Jupyter Notebooks, Google Colab

Developed as part of the Artificial Neural Networks and Deep Learning course - Showcasing practical applications of advanced deep learning techniques in computer vision.

About

Image Classification and Image Segmentation projects developed during the Artificial Neural Network & Deep Learning course at PoliMi

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 2

  •  
  •