TensorRT inference for DETR and Deformable DETR

Visual-Behavior / detr-tensorflow

Tensorflow implementation of DETR : Object Detection with Transformers

MIT License

168 stars 53 forks source link

Open nntrongnghia opened 3 years ago

nntrongnghia commented 3 years ago

In this pull request:

Custom TensorRT plugin Im2Col for Deformable attention
Export models to ONNX graph, then export ONNX graph to TensorRT serialized engine
A sample script for inference with TensorRT
Use Group Norm in Deformable DETR
Better visualization for bbox