Image Classification
timm
ONNX
Safetensors
mobile-screenshots
phone-screenshots
screenshot-analysis
content-safety
Instructions to use yapwithai/phone-screen-classifier with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- timm
How to use yapwithai/phone-screen-classifier with timm:
import timm model = timm.create_model("hf_hub:yapwithai/phone-screen-classifier", pretrained=True) - Notebooks
- Google Colab
- Kaggle
Upload phone screen classifier
Browse files- README.md +0 -8
- onnx/model.int8.onnx +0 -3
README.md
CHANGED
|
@@ -39,7 +39,6 @@ This model is built for routing and filtering mobile screenshot workflows. It se
|
|
| 39 |
| :------------------------ | :------------------------------------------------------------- |
|
| 40 |
| `onnx/model.onnx` | ONNX model for CPU/server inference. |
|
| 41 |
| `onnx/model.fp16.onnx` | Optional FP16 ONNX candidate. |
|
| 42 |
-
| `onnx/model.int8.onnx` | Optional INT8 ONNX candidate. |
|
| 43 |
| `onnx/model.onnx.data` | External ONNX weight data loaded beside `model.onnx`. |
|
| 44 |
| `model.safetensors` | PyTorch state dict for reproducibility and continued training. |
|
| 45 |
| `config.json` | Model identity, base model, output names, and label arrays. |
|
|
@@ -77,12 +76,6 @@ Inference uses `argmax` for both heads in this version.
|
|
| 77 |
| fp16 | screen-balanced test | safety | 0.9776 | 0.9285 | 0.9110 | 0.9978 | 6246 |
|
| 78 |
| fp16 | safety-balanced test | screen | 0.9580 | 0.7752 | 0.6547 | 0.9870 | 3000 |
|
| 79 |
| fp16 | safety-balanced test | safety | 0.8957 | 0.8957 | 0.8947 | 0.9847 | 3000 |
|
| 80 |
-
| int8 | full test | screen | 0.0018 | 0.0424 | 0.0024 | 0.0278 | 23615 |
|
| 81 |
-
| int8 | full test | safety | 0.1013 | 0.3502 | 0.0696 | 0.9431 | 23615 |
|
| 82 |
-
| int8 | screen-balanced test | screen | 0.0062 | 0.0424 | 0.0031 | 0.1039 | 6246 |
|
| 83 |
-
| int8 | screen-balanced test | safety | 0.1703 | 0.3686 | 0.1153 | 0.9241 | 6246 |
|
| 84 |
-
| int8 | safety-balanced test | screen | 0.0003 | 0.0417 | 0.0000 | 0.0140 | 3000 |
|
| 85 |
-
| int8 | safety-balanced test | safety | 0.3493 | 0.3493 | 0.1992 | 0.6667 | 3000 |
|
| 86 |
|
| 87 |
## CPU Timing
|
| 88 |
|
|
@@ -90,7 +83,6 @@ Inference uses `argmax` for both heads in this version.
|
|
| 90 |
| :-- | --: | --: | --: | --: | --: | --: | --: |
|
| 91 |
| fp32 | 20.0538 | 0.12 ms | 49.52 ms | 49.87 ms | 57.84 ms | 85.82 ms | onnxruntime:CPUExecutionProvider |
|
| 92 |
| fp16 | 20.8351 | 0.11 ms | 47.69 ms | 48.00 ms | 48.60 ms | 63.65 ms | onnxruntime:CPUExecutionProvider |
|
| 93 |
-
| int8 | 72.6504 | 0.14 ms | 13.36 ms | 13.76 ms | 14.07 ms | 18.01 ms | onnxruntime:CPUExecutionProvider |
|
| 94 |
|
| 95 |
Timing is measured with ONNX Runtime CPU execution on `Apple M4 Max (16 logical cores)`. Total latency includes image load/preprocess, model inference, and label decoding.
|
| 96 |
|
|
|
|
| 39 |
| :------------------------ | :------------------------------------------------------------- |
|
| 40 |
| `onnx/model.onnx` | ONNX model for CPU/server inference. |
|
| 41 |
| `onnx/model.fp16.onnx` | Optional FP16 ONNX candidate. |
|
|
|
|
| 42 |
| `onnx/model.onnx.data` | External ONNX weight data loaded beside `model.onnx`. |
|
| 43 |
| `model.safetensors` | PyTorch state dict for reproducibility and continued training. |
|
| 44 |
| `config.json` | Model identity, base model, output names, and label arrays. |
|
|
|
|
| 76 |
| fp16 | screen-balanced test | safety | 0.9776 | 0.9285 | 0.9110 | 0.9978 | 6246 |
|
| 77 |
| fp16 | safety-balanced test | screen | 0.9580 | 0.7752 | 0.6547 | 0.9870 | 3000 |
|
| 78 |
| fp16 | safety-balanced test | safety | 0.8957 | 0.8957 | 0.8947 | 0.9847 | 3000 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 79 |
|
| 80 |
## CPU Timing
|
| 81 |
|
|
|
|
| 83 |
| :-- | --: | --: | --: | --: | --: | --: | --: |
|
| 84 |
| fp32 | 20.0538 | 0.12 ms | 49.52 ms | 49.87 ms | 57.84 ms | 85.82 ms | onnxruntime:CPUExecutionProvider |
|
| 85 |
| fp16 | 20.8351 | 0.11 ms | 47.69 ms | 48.00 ms | 48.60 ms | 63.65 ms | onnxruntime:CPUExecutionProvider |
|
|
|
|
| 86 |
|
| 87 |
Timing is measured with ONNX Runtime CPU execution on `Apple M4 Max (16 logical cores)`. Total latency includes image load/preprocess, model inference, and label decoding.
|
| 88 |
|
onnx/model.int8.onnx
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:f1acefd4c5a1204c6c7b1db7bd8e31b59117982136379a60c4dae05d5798e155
|
| 3 |
-
size 9542131
|
|
|
|
|
|
|
|
|
|
|
|