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
-
Notifications
You must be signed in to change notification settings - Fork 0
Unsupervised learning approach for waste classification using contrastive learning and clustering-based voting
License
CryAndRRich/r4bbish
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
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 0
No packages published