Skip to content

Anipro-10/Deep-Learning-based-Metal-Scrap-Classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

3 Commits
Β 
Β 
Β 
Β 

Repository files navigation

Metal Scrap Classification and Detection using Deep Learning

πŸ“Œ Overview

This project uses Convolutional Neural Networks (CNNs) to classify metallic scrap images into four categories β€” Aluminium, Copper, Steel, and Brass β€” based on shape, color, and other visual features. With proper preprocessing, data augmentation, and training strategies, the model achieved 96.5% accuracy on real-world test data.

πŸ› οΈ Technologies Used

  • Python 3.x
  • TensorFlow / Keras
  • Matplotlib, NumPy

πŸ§ͺ Performance

Class Precision Recall F1-Score
Aluminium 0.97 0.96 0.96
Copper 0.95 0.94 0.95
Steel 0.96 0.97 0.96
Brass 0.95 0.94 0.95
  • Overall Accuracy: 96.51%
  • Evaluated using Confusion Matrix and Real-World Test Images

🌍 Real-World Impact

  • Automates scrap sorting in industrial scrapyards
  • Enables vision-based inspection in recycling plants
  • Contributes to AI-driven circular economy solutions

πŸš€ Future Developments

  • πŸ” Train and fine-tune YOLOv8 for object detection with bounding boxes
  • 🧠 Integrate attention mechanisms to handle cluttered and overlapping scrap regions
  • πŸŽ₯ Enable real-time detection from video streams
  • πŸ§ͺ Use a hybrid dataset (synthetic + real) to improve generalizability

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors