pytorch / pytorch/vision

[feature request] detection/roi_heads.py:postprocess_detections to also return keep indicator

Open
#4,309 2 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

enhancement module: models needs discussion topic: object detection
Dominant language
Python
Stars
17.9k
Forks
7.3k
Avg merge
1d 15h
Merged PRs (30d)
13

Description

With this it's easier to hack and retrieve the necesssary box_features for kept boxes without having to reimplement the whole pipeline

cc @datumbox

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 in detection/roi_heads.py at postprocess_detections and trace its callers to understand the existing return values and how kept boxes are represented. Define the additional keep indicator requested by the issue, then verify that callers and relevant detection behavior remain consistent; the work is done when kept-box selection can be recovered without reimplementing the pipeline.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
computer-vision, machine-learning
Issue type
Feature
Difficulty
3/5
Estimated time
1-2 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.