search

lyuwenyu / RT-DETR

[CVPR 2024] Official RT-DETR (RTDETR paddle pytorch), Real-Time DEtection TRansformer, DETRs Beat YOLOs on Real-time Object Detection. 🔥 🔥 🔥
Apache License 2.0
2.31k stars 258 forks source link

can i disable postprocess in onnx-export? #434

Closed ksv87 closed 1 month ago

ksv87 commented 1 month ago

Can i disable postprocess when i export onnx model?

lyuwenyu commented 1 month ago

Of course, but you need to modify this file. 

# TODO remove postprocessor
# TODO modify forward args
# TODO modify torch.onnx.export

https://github.com/lyuwenyu/RT-DETR/blob/main/rtdetrv2_pytorch/tools/export_onnx.py#L33-L42

ksv87 commented 1 month ago

thanks, can you say, how its will affect to performance (fps)?

lyuwenyu commented 1 month ago

Without the postprocessor part, the speed will increase