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
Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support