metadata
tags:
- image-classification
- pytorch
- computer-vision
- cup-detection
library_name: pytorch
Cup Detector - SmallCNN 2 Output
This model classifies a machine-specific ROI into:
- EMPTY
- CUP
- UNDEFINED
Model
- Architecture:
16 -> 32 -> 64 CNN - Input:
1 x 64 x 64 - Output layer:
2 - Loss:
BCEWithLogitsLoss
Output encoding
- EMPTY ->
[0, 0] - CUP ->
[1, 0] - UNDEFINED ->
[0, 1]
Output 0 represents CUP vs EMPTY.
Output 1 represents UNDEFINED.
Test Results
- Test Accuracy:
98.91% - Balanced Accuracy:
94.44% - CUP Precision:
100.00% - CUP Recall:
100.00% - EMPTY Specificity:
98.03% - CUP F1:
100.00%
Intended Pipeline
Camera / RTSP
↓
Machine-specific ROI
↓
Grayscale + Resize 64x64
↓
SmallCNN
↓
CUP / EMPTY / UNDEFINED
The ROI selection is handled outside the model. The model receives the cropped machine ROI.