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

[Enhancement] Refactor TrackingNetDataset, SOTCocoDataset and SOTImageNetVIDDataset based on BaseSOTDataset #402

Merged
merged 11 commits into from
Jan 25, 2022

Conversation

JingweiZhang12
Copy link
Collaborator

@JingweiZhang12 JingweiZhang12 commented Jan 21, 2022

Refactor TrackingNetDataset, SOTCocoDataset and SOTImageNetVIDDataset based on BaseSOTDataset.
Please refer to #393 for more details.

@codecov
Copy link

codecov bot commented Jan 21, 2022

Codecov Report

Merging #402 (f6445e8) into master (1d32dc3) will increase coverage by 0.23%.
The diff coverage is 92.35%.

Impacted file tree graph

@@            Coverage Diff             @@
##           master     #402      +/-   ##
==========================================
+ Coverage   70.25%   70.48%   +0.23%     
==========================================
  Files         114      116       +2     
  Lines        5981     6113     +132     
  Branches     1157     1181      +24     
==========================================
+ Hits         4202     4309     +107     
- Misses       1428     1448      +20     
- Partials      351      356       +5     
Flag Coverage Δ
unittests 70.40% <92.35%> (+0.25%) ⬆️

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

Impacted Files Coverage Δ
mmtrack/datasets/got10k_dataset.py 86.07% <ø> (+2.53%) ⬆️
mmtrack/datasets/trackingnet_dataset.py 80.82% <78.94%> (-12.93%) ⬇️
mmtrack/datasets/__init__.py 100.00% <100.00%> (ø)
mmtrack/datasets/base_sot_dataset.py 61.60% <100.00%> (+1.44%) ⬆️
mmtrack/datasets/sot_coco_dataset.py 100.00% <100.00%> (ø)
mmtrack/datasets/sot_imagenet_vid_dataset.py 100.00% <100.00%> (ø)
mmtrack/models/trackers/tracktor_tracker.py 63.95% <0.00%> (-15.12%) ⬇️
mmtrack/datasets/pipelines/transforms.py 85.92% <0.00%> (-1.50%) ⬇️
mmtrack/models/trackers/byte_tracker.py 89.68% <0.00%> (ø)
... and 1 more

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 1d32dc3...f6445e8. Read the comment docs.

mmtrack/datasets/sot_coco_dataset.py Outdated Show resolved Hide resolved
mmtrack/datasets/sot_coco_dataset.py Outdated Show resolved Hide resolved
mmtrack/datasets/sot_imagenet_vid_dataset.py Outdated Show resolved Hide resolved
mmtrack/datasets/sot_imagenet_vid_dataset.py Outdated Show resolved Hide resolved
mmtrack/datasets/sot_imagenet_vid_dataset.py Show resolved Hide resolved
mmtrack/datasets/sot_imagenet_vid_dataset.py Show resolved Hide resolved
num_chunks (int, optional): the number of chunks. Some methods may
only use part of the dataset. Default to all chunks, 12.
"""
self.num_chunks = num_chunks
Copy link
Collaborator

Choose a reason for hiding this comment

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

The num_chunks should not be a int, since the user may want to specify the index of chunks. It should be a list with optional values in [1,2,3,4,'all']

tests/test_data/test_datasets/test_sot_dataset.py Outdated Show resolved Hide resolved
tests/data/demo_sot_data/lasot/create_assets.py Outdated Show resolved Hide resolved
mmtrack/datasets/sot_coco_dataset.py Outdated Show resolved Hide resolved
Copy link
Collaborator

@GT9505 GT9505 left a comment

Choose a reason for hiding this comment

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

Please test the performance of SiameseRPN++ on TrackingNet after completing the reviewing comments. Then, we can merge this PR.

mmtrack/datasets/sot_imagenet_vid_dataset.py Show resolved Hide resolved
mmtrack/datasets/trackingnet_dataset.py Outdated Show resolved Hide resolved
mmtrack/datasets/trackingnet_dataset.py Show resolved Hide resolved
@GT9505 GT9505 changed the title [Enhancement] Refactor trackingnet, sot coco and sot imagenetvid dataset class based on BaseSOTDataset [Enhancement] Refactor TrackingNetDataset, SOTCocoDataset nad SOTImageNetVIDDataset based on BaseSOTDataset Jan 25, 2022
@GT9505 GT9505 changed the title [Enhancement] Refactor TrackingNetDataset, SOTCocoDataset nad SOTImageNetVIDDataset based on BaseSOTDataset [Enhancement] Refactor TrackingNetDataset, SOTCocoDataset and SOTImageNetVIDDataset based on BaseSOTDataset Jan 25, 2022
@GT9505 GT9505 merged commit bb0a643 into open-mmlab:master Jan 25, 2022
@JingweiZhang12 JingweiZhang12 deleted the trackingnet branch April 29, 2022 02:30
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