hinged-state-classifier-efficientnet-b0

EfficientNet-B0 image classification model for surgical instrument state recognition.

Task

Binary classification of hinged surgical instrument state:

  • closed
  • open

Model details

  • Architecture: EfficientNet-B0
  • Framework: PyTorch
  • Checkpoint: best.pt
  • Input image size: 224 × 224

Dataset

Trained on:

  • joonhaim/hinged-state-classifier-crops

Class mapping

  • 0: closed
  • 1: open

Notes

This is a custom PyTorch checkpoint for classifying the state of hinged surgical instruments from cropped images.

The class order must match the order used during training.

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

Dataset used to train joonhaim/hinged-state-classifier-efficientnet-b0

Collection including joonhaim/hinged-state-classifier-efficientnet-b0