How to transfer this model to onnx?

bowenc0221 / panoptic-deeplab

This is Pytorch re-implementation of our CVPR 2020 paper "Panoptic-DeepLab: A Simple, Strong, and Fast Baseline for Bottom-Up Panoptic Segmentation" (https://arxiv.org/abs/1911.10194)

Apache License 2.0

590 stars 117 forks source link

How to transfer this model to onnx? #101

Open justsolo-smith opened 3 years ago

bowenc0221 commented 3 years ago

I'm not familiar with ONNX, but since my implementation is based on Detectron2, you can probably try this: https://github.com/facebookresearch/detectron2/blob/master/tools/deploy/export_model.py