graph.collapse_namespaces(namespace_plugin_map)

AI & Data Science Deep Learning (Training & Inference) TensorRT
1

https://github.com/AastaNV/TRT_object_detection/blob/master/config/model_ssd_mobilenet_v2_coco_2018_03_29.py

namespace_plugin_map = {
“MultipleGridAnchorGenerator”: PriorBox,
“Postprocessor”: NMS,
“Preprocessor”: Input,
“ToFloat”: Input,
“image_tensor”: Input,
“Concatenate”: concat_priorbox,
“concat”: concat_box_loc,
“concat_1”: concat_box_conf
}

graph.collapse_namespaces(namespace_plugin_map)

The corresponding namespace of NMS is Postprocessor, but under the Postprocessor, there are other operations, such as the Preprocessor/map/while/Enter, unified into NMS.
Will this lead to problems in node connection？

2

Hi,

There should not be any issue with node connection in this case.
The config file should be slightly modified for different models.

If possible could you please share your model file so we can better help?

Thanks