LibreDeformableDETRr50twostage

The official box-refinement two-stage ResNet-50 Deformable DETR checkpoint (46.9 AP on COCO val2017), converted for LibreYOLO.

from libreyolo import LibreYOLO

model = LibreYOLO("LibreDeformableDETRr50twostage.pt")
results = model.predict("image.jpg")

LibreYOLO ships this museum family for inference only. Its portable deformable attention uses the upstream pure-PyTorch grid_sample reference path; the custom CUDA extension is not packaged. Native and ONNX inference use a fixed 800 x 800 PIL-bilinear stretch with ImageNet normalization. That deployment transform intentionally differs from upstream's short-side-800/max-1333 evaluation transform.

Source

Architecture derived from fundamentalvision/Deformable-DETR at commit 11169a60c33333af00a4849f1808023eba96a931. Copyright (c) 2020 SenseTime. All Rights Reserved. Modified from DETR, Copyright 2020 - present, Facebook, Inc. Licensed under Apache License 2.0.

The checkpoint comes from SenseTime/deformable-detr-with-box-refine-two-stage at revision e74bff70d69f3e825f6cefaf179bfba707f92054. Source model.safetensors SHA-256: 411bb4238a834d40fff651b1b5b7d6dd80c2dd28be1747eec7b6918674e85de6.

Modifications

Checkpoint keys are remapped to the original module layout, decoder Q/K/V projections are concatenated, and serialization aliases for the refinement and encoder-proposal heads are restored. Non-parameter BatchNorm tracking counters are omitted; learned tensor values are unchanged. The released 91-column COCO category-id head is retained and mapped to contiguous COCO-80 classes during postprocess.

The converted checkpoint SHA-256 is 7250d79708f53f2f5b8b0daec94011def1c262c4fc723993c1f58128c8002670. LibreYOLO's output tensors, including encoder proposals, are bit-exact against the pinned upstream pure-PyTorch path (max_abs_diff == 0.0); fixed-800 ONNX Runtime prediction parity is also verified.

See weights/convert_deformable_detr_weights.py and docs/provenance/deformable_detr.md in the LibreYOLO source repository.

License

Apache License 2.0. See the LICENSE and NOTICE files in this repository.

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Dataset used to train LibreYOLO/LibreDeformableDETRr50twostage

Collection including LibreYOLO/LibreDeformableDETRr50twostage