how to loading a mmdetection model in C++
See original GitHub issueI train the cascade-rcnn by mmdetection and get a model, now I want to use this model to inference by C++.
So I convert this model by:
trace_script_module = torch.jit.trace(model, example_input) trace_script_module.save(result_file)
In general, example_input
is a array to simulate an image. But I find that mmdetection model doesn’t use numpy array as the input, it uses the following code to inference:
result = model(return_loss=False, rescale=True, **data)
How should I simulate example_input
?
Thanks
Issue Analytics
- State:
- Created 4 years ago
- Comments:11
Top Results From Across the Web
latest PDF - MMDetection's documentation!
When installing PyTorch, you need to specify the version of CUDA. If you are not clear on which to choose, follow our recommendations:....
Read more >MMDetection Tutorial - | notebook.community
Perform inference with a MMDet detector. MMDetection already provides high level APIs to do inference and training. In [4]: !mkdir checkpoints !wget -c...
Read more >MMDetection Tutorial — An End2End State-of-the-art Object ...
The next thing to do is to open up the repository files and navigate to the “configs” directory. In the configs directory, you...
Read more >Object Detection using mmdetection | by Ravi Singh | Medium
Mmdetection has three main folders:- Configs: It has all the configuration files, where you tell the model, location of the data, ...
Read more >Getting Started with MMDetection Training for Object Detection
Download the Pretrained Weights ; parse_meta_file() · Function to parse all the model meta files inside `mmdetection/configs` ; model_name) · Either ...
Read more >Top Related Medium Post
No results found
Top Related StackOverflow Question
No results found
Troubleshoot Live Code
Lightrun enables developers to add logs, metrics and snapshots to live code - no restarts or redeploys required.
Start FreeTop Related Reddit Thread
No results found
Top Related Hackernoon Post
No results found
Top Related Tweet
No results found
Top Related Dev.to Post
No results found
Top Related Hashnode Post
No results found
Top GitHub Comments
you can try https://github.com/ruinmessi/RFBNet ,this detection model is very easy to deploy in C++
Requires just a small patch to forward() to have default argument of img_metas=None and then if img_metas is None return simple_test