dert_head.py bug with mmdet2trt version 0.6.0

grimoire / mmdetection-to-tensorrt

convert mmdetection model to tensorrt, support fp16, int8, batch input, dynamic shape etc.

Apache License 2.0

590 stars 85 forks source link

dert_head.py bug with mmdet2trt version 0.6.0 #132

Open KaneiGi opened 1 month ago

KaneiGi commented 1 month ago

Describe the bug On line 15 of mmdet2trt/models/dense_heads/detr_head.py:self.positional_encoding = build_wrapper(module.positional_encoding) Since mmdet version 3, the parameter positional_encoding is moved out from DETRHead's scope, so the module does not the attribution of positional_encoding.

environment: torch=2.2.0 tensorrt=8.6.1 mmdetection=3.3.0 cuda=11.7 mmdet2trt=0.6.0

grimoire commented 4 weeks ago

Thanks for the report. I will fix it soon.