facebookresearch / facebookresearch/detectron2
origin of if operator in onnx export
Open
- Dominant language
- Python
- Stars
- 34.7k
- Forks
- 7.9k
- PR merge metrics
- No merged PRs in 30d
Description
When exporting a panoptic_fpn from detectron 2, there is the following subgraph in the onnx graph. The subgraph follows the NMS in the RPN. For my application, I would like to understand where the `If` operator comes from in the source code.
Code used to export the model: `python3 tools/deploy/export_model.py --config-file configs/COCO-PanopticSegmentation/panoptic_fpn_R_50_1x.yaml --output onnx_panoptic_fpn_R_50_1x/ --export-method tracing --format onnx`
Contributor guide
Assessment
This issue has not been assessed yet.