Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Feature] Support TAO dataset of MOT field #415

Merged
merged 15 commits into from
Feb 17, 2022
Merged

[Feature] Support TAO dataset of MOT field #415

merged 15 commits into from
Feb 17, 2022

Conversation

Seerkfang
Copy link
Collaborator

No description provided.

@codecov
Copy link

codecov bot commented Feb 3, 2022

Codecov Report

Merging #415 (c8a347e) into master (b62250d) will increase coverage by 0.36%.
The diff coverage is 84.04%.

Impacted file tree graph

@@            Coverage Diff             @@
##           master     #415      +/-   ##
==========================================
+ Coverage   71.31%   71.68%   +0.36%     
==========================================
  Files         117      118       +1     
  Lines        6534     6697     +163     
  Branches     1250     1285      +35     
==========================================
+ Hits         4660     4801     +141     
- Misses       1490     1499       +9     
- Partials      384      397      +13     
Flag Coverage Δ
unittests 71.58% <84.04%> (+0.32%) ⬆️

Flags with carried forward coverage won't be shown. Click here to find out more.

Impacted Files Coverage Δ
mmtrack/datasets/tao_dataset.py 83.95% <83.95%> (ø)
mmtrack/datasets/__init__.py 100.00% <100.00%> (ø)
mmtrack/models/reid/base_reid.py 78.26% <0.00%> (-8.70%) ⬇️
mmtrack/datasets/sot_train_dataset.py 82.82% <0.00%> (-3.04%) ⬇️
mmtrack/models/trackers/base_tracker.py 84.73% <0.00%> (-1.53%) ⬇️
mmtrack/datasets/pipelines/transforms.py 87.84% <0.00%> (-0.92%) ⬇️
mmtrack/datasets/coco_video_dataset.py 59.18% <0.00%> (+0.81%) ⬆️
mmtrack/models/trackers/tracktor_tracker.py 79.06% <0.00%> (+15.11%) ⬆️

Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update b62250d...c8a347e. Read the comment docs.

@Seerkfang Seerkfang requested a review from GT9505 February 3, 2022 12:32
mmtrack/datasets/tao_dataset.py Show resolved Hide resolved
mmtrack/datasets/tao_dataset.py Show resolved Hide resolved
mmtrack/datasets/tao_dataset.py Outdated Show resolved Hide resolved
mmtrack/datasets/tao_dataset.py Outdated Show resolved Hide resolved
mmtrack/datasets/tao_dataset.py Outdated Show resolved Hide resolved
tests/test_data/test_datasets/test_tao_dataset.py Outdated Show resolved Hide resolved
tools/convert_datasets/tao/merge_coco_with_lvis.py Outdated Show resolved Hide resolved
tools/convert_datasets/tao/merge_coco_with_lvis.py Outdated Show resolved Hide resolved
tools/convert_datasets/tao/merge_coco_with_lvis.py Outdated Show resolved Hide resolved
tools/convert_datasets/tao/tao2coco.py Outdated Show resolved Hide resolved
.github/workflows/build.yml Outdated Show resolved Hide resolved
docs/en/dataset.md Outdated Show resolved Hide resolved
docs/en/dataset.md Outdated Show resolved Hide resolved
docs/en/install.md Show resolved Hide resolved
docs/en/dataset.md Show resolved Hide resolved
docs/en/install.md Outdated Show resolved Hide resolved
tests/data/demo_tao_data/tao_classes.txt Outdated Show resolved Hide resolved
docs/en/dataset.md Outdated Show resolved Hide resolved
mmtrack/datasets/tao_dataset.py Show resolved Hide resolved
mmtrack/datasets/tao_dataset.py Show resolved Hide resolved
docs/en/dataset.md Show resolved Hide resolved
docs/en/dataset.md Outdated Show resolved Hide resolved
docs/en/dataset.md Outdated Show resolved Hide resolved
docs/en/dataset.md Show resolved Hide resolved

- The `annotations` under `lvis` contains the official annotations of lvis-v0.5 which can be downloaded according to [here](https://github.com/lvis-dataset/lvis-api/issues/23#issuecomment-894963957)。

- The synset mapping file `coco_to_lvis_synset.json` used in `lvis` training can be found [here](https://github.com/lvis-dataset/lvis-api/issues/23#issuecomment-894963957)。
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

wrong link.

Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

And merge line 38 with line 40

Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
- The synset mapping file `coco_to_lvis_synset.json` used in `lvis` training can be found [here](https://github.com/lvis-dataset/lvis-api/issues/23#issuecomment-894963957)
- The synset mapping file `coco_to_lvis_synset.json` used in `./tools/convert_datasets/tao/merge_coco_with_lvis.py` script can be found [here](xxx)

@@ -197,6 +234,14 @@ python ./tools/convert_datasets/mot/mot2reid.py -i ./data/MOT17/ -o ./data/MOT17
# CrowdHuman
python ./tools/convert_datasets/mot/crowdhuman2coco.py -i ./data/crowdhuman -o ./data/crowdhuman/annotations

# LVIS
# Merge annotations from LVIS and COCO for training QDTrack
python ./tools/convert_datasets/tao/merge_coco_with_lvis.py --lvis ./data/lvis/annotations/lvis_v0.5_train.json --coco ./data/coco/annotations/instances_train2017.json --mapping ./data/coco_to_lvis_synset.json --output-json ./data/lvis/annotations/lvisv0.5+coco_train.json
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

put ./data/coco_to_lvis_synset.json into ./data/lvis/annotations/coco_to_lvis_synset.json

@GT9505 GT9505 changed the title [Add] support TAO dataset [Add] support TAO dataset in MOT field Feb 17, 2022
@GT9505 GT9505 changed the title [Add] support TAO dataset in MOT field [Add] Support TAO dataset of MOT field Feb 17, 2022
@GT9505 GT9505 changed the title [Add] Support TAO dataset of MOT field [Feature] Support TAO dataset of MOT field Feb 17, 2022
@GT9505 GT9505 merged commit 938eaa7 into open-mmlab:master Feb 17, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants