|
|
--- |
|
|
license: apache-2.0 |
|
|
base_model: |
|
|
- google/efficientnet-b2 |
|
|
--- |
|
|
|
|
|
## Overview |
|
|
|
|
|
Detects bad crop jobs and misalignments, like screenshots from phones that should be trimmed down to relevant details. |
|
|
|
|
|
Version 8. |
|
|
|
|
|
Training progress: https://wandb.ai/josephc/bad_crop/overview |
|
|
|
|
|
## Usage: |
|
|
|
|
|
```python |
|
|
import numpy |
|
|
import onnxruntime as ort |
|
|
session = ort.InferenceSession("./bad_crop.onnx") |
|
|
blank = numpy.zeros((1, 3, 224, 224), dtype=numpy.float32) |
|
|
out = torch.softmax(torch.tensor(session.run(["output"], {"input": blank,})[0][0]), axis=0) |
|
|
print(out) # tensor([9.9941e-01, 5.8832e-04]) # Bad crop, good crop |
|
|
``` |
|
|
|
