JosephCatrambone/badcrop

By JosephCatrambone · ↓ 0 · ♥ 0 · Updated 2025-12-14

What's in this repo

This page lists the files in JosephCatrambone/badcrop, including bad_crop.onnx.

1 ONNX models (.onnx files)
3 files, including 1 .md, 1 .onnx
Last updated 2025-12-14

ONNX files

File	Size	Download
bad_crop.onnx	—	Download

How to use

These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.

README

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:

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

Model info

onnx base_model:google/efficientnet-b2 base_model:quantized:google/efficientnet-b2 license:apache-2.0 region:us

↓ 0 ♥ 0

Created: 2025-12-14

Updated: 2025-12-14

View on Hugging Face

Files (3)

.gitattributes

README.md

bad_crop.onnx ONNX

Related models

More models

No task type declared.