--- 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 ```