Determine how to make multiple network outputs flexible
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 18
- Forks
- 28
- Avg merge
- 2d 2h
- Merged PRs (30d)
- 7
Description
The current export and inference pipeline assumes a single output tensor (detection statistic). Supporting merger time requires routing a second heatmap tensor through the TRT ensemble in export/main.py, receiving and unpacking multiple output arrays in the inference client, and dispatching to the correct postprocessor strategy based on which outputs are present. The goal is to make this configurable (e.g. via export_heatmap: bool) rather than requiring separate code paths.
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start with export/main.py and trace how the export and inference pipeline handles the current detection-statistic tensor. Then inspect the inference client’s output unpacking and the postprocessor dispatch strategy. Done means a configurable export_heatmap option supports routing and processing either one output or both outputs without separate code paths.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- backend, machine-learning
- Issue type
- Feature
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Quiet
- Clarity
- Mostly clear
- Newbie friendliness
- 48/100