hinged-state-classifier-yolo26n-cls

YOLO26n classification model for surgical instrument state recognition.

Task

Binary classification of hinged surgical instrument state:

  • closed
  • open

Model details

  • Architecture: YOLO26n-cls
  • Framework: Ultralytics YOLO
  • Checkpoint: best.pt
  • Input image size: 224 × 224
  • Epochs: 40
  • Batch size: 32
  • Device used for training: MPS
  • Pretrained: yes
  • Optimizer: auto
  • Initial learning rate (lr0): 0.01
  • Weight decay: 0.0005

Dataset

Trained on:

  • joonhaim/hinged-state-classifier-crops

Notes

This is a custom Ultralytics YOLO classification 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
51
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Dataset used to train joonhaim/hinged-state-classifier-yolo26n-cls

Collection including joonhaim/hinged-state-classifier-yolo26n-cls