Python code for making predictions with a traced torchscript Mask R-CNN model #2905
fracpete
started this conversation in
Show and tell
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
Similar to the torchscript_traced_mask_rcnn.cpp, the following Python code can be used for making predictions with a traced torchscript Mask R-CNN model:
https://github.com/fracpete/detectron2/blob/master/tools/deploy/torchscript_traced_mask_rcnn.py
I was missing an example on how to use a torchscript model from Python, so it took me a while figuring that out (see #2781). Other people might benefit from this example as well. Either grab the code directly or allow me to do a PR.
Beta Was this translation helpful? Give feedback.
All reactions