microsoft / microsoft/onnxruntime

[RFC] Add torchvision/faster-rcnn inference example on cpp backend

Open
#6,208 0 comments 1 reaction 0 assignees View on GitHub

Nobody has claimed this yet.

contributions welcome
Dominant language
C++
Stars
21.9k
Forks
4.2k
Avg merge
4d 11h
Merged PRs (30d)
184

Description

**Is your feature request related to a problem? Please describe.**

[TorchVision](https://github.com/pytorch/vision) have supported `faster-rcnn` and `retinanet`, refer to [here](https://github.com/pytorch/vision/blob/7b9d30eb7c4d92490d9ac038a140398e0a690db6/test/test_onnx.py#L366), this two models can be export to `onnx` and infer with `onnxruntime` python backend. I wanne know whether there is a plan to add an inference example of this two models on cpp backend?

**System information**
- ONNX Runtime version (you are using): build from master branch

**Describe the solution you'd like**

Add an inference example of torchvision's `retinanet` or `faster-rcnn` on C++ backend

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start with the linked TorchVision test/test_onnx.py reference for faster-rcnn and retinanet, then review how ONNX Runtime represents C++ inference examples. Decide which model the example should cover and what the C++ backend entry point requires; done means the selected model has a working C++ inference example.

Written by the indexing model from the issue text.

Assessment

Tech stack
cpp, pytorch
Domain
backend, machine-learning
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.