Megvii-BaseDetection / Megvii-BaseDetection/YOLOX

Understand label conversion

Open
#1,457 0 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Dominant language
Python
Stars
10.6k
Forks
2.5k
PR merge metrics
No merged PRs in 30d

Description

Hi @FateScript and others, thanks for this great contribution! I'm writing a dataloader for a custom dataset and I've having a bit of trouble with the format of labels at different steps and how the conversions happen. It could be that some comments in the code are correct, would be great if we could confirm a few things.

For coco dataset, it looks like the pull_item should be returning labels in the xyxy format resulting from:
https://github.com/Megvii-BaseDetection/YOLOX/blob/d78fe47802d3987a915645179d87f5c1a96e646f/yolox/data/datasets/coco.py#L149

Now in MosaicDetection:

Is it safe to assume that pull_item for my custom dataset should be returning xyxy format for everything else to work correctly?

Contributor guide

No contributing guide indexed for this repository

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

Read yolox/data/datasets/coco.py around pull_item and yolox/data/datasets/mosaicdetection.py around the mosaic and preprocessing paths. Trace the label format at each referenced step, then confirm the expected custom-dataset contract and correct the misleading comment or documentation. Done means the conversion sequence and pull_item format are unambiguous.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
data, machine-learning
Issue type
Documentation
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.