New
Detectron2 and torchscript support, attention and panoptic notebooks, code improvements
This release contains several improvements compared to the initial release of DETR.
Highlights
Detectron2 support
DETR now can be used with Detectron2 in a seamless way. It reuses the original abstractions of DETR, while making a compatibility layer so that it fits Detectron2's API. More information can be found in the README under the d2/ folder
TorchScript support for all models
DETR models can now be natively exported to torchscript. This enables DETR to be run in C++ via libtorch. Converting a model to torchscript is easy:
model = torch.hub.load('facebookresearch/detr', 'detr_resnet50', pretrained=True)
model = torch.jit.script(model)
# now the model run entirely in torchscript!