---
license: mit
tags:
- computer-vision
- image-classification
- pytorch
---

# TrashNet PoC Best Model

Two candidates were trained on TrashNet:
- ResNet50
- MobileNetV2

Selected best by validation accuracy.

Classes: ['cardboard', 'glass', 'metal', 'paper', 'plastic', 'trash']

Files:
- `best_model.pt`: checkpoint (state_dict + metadata)
- `labels.json`: class names