Skip to content

Unsupervised learning approach for waste classification using contrastive learning and clustering-based voting

License

Notifications You must be signed in to change notification settings

CryAndRRich/r4bbish

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

31 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

rubbish4/
├── data/                      
│   └── trashnet/              # Directory containing raw waste images
│
├── src/                       # Main source code
│   ├── visualize/             # Functions for visualizing embeddings, clusters, etc.
│   │
│   ├── encoders.py            # Defines the Dual-Encoder contrastive model and loss functions
│   ├── dataset.py             # PyTorch Dataset and DataLoader for loading waste images
│   ├── utils.py               # Utility functions (feature extraction, checkpoint saving, etc.)
│   ├── config.py              # Configuration for hyperparameters, paths, and global settings
│   └── multi_cluster.py       # Scripts for running clustering algorithms (KMeans, BIRCH, Agglomerative)
│
├── checkpoints/               # Stores model checkpoints (.pth) during training
│
├── test.py                    # Script to run the full unsupervised classification pipeline
├── requirements.txt           # Required Python packages (timm, torch, torchvision, etc.)
├── README.md                  # Project overview and usage instructions
└── .gitignore

About

Unsupervised learning approach for waste classification using contrastive learning and clustering-based voting

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 4

  •  
  •  
  •  
  •