Add automatic model download for 01 tracking notebook

abuendia commented 4 years ago

Description

Adds automatic model download for tracking 01 notebook. The FairMOT baseline all_dla34.pth has been zipped into a baselines folder and uploaded to the public cvbp blob container.

Uses the existing unzip_url util and utils_cv/tracking/data.py for minimal code changes. Creates model directory if model path does not exist. Allows cell to rerun without re-downloading if model already exists.

Checklist:

[x] I have followed the contribution guidelines and code style for this project.
[x] I have added tests covering my contributions.
[x] I have updated the documentation accordingly.
[x] This PR is being made to staging and not master
[x] I will squash merge this PR into staging

review-notebook-app[bot] commented 4 years ago

Check out this pull request on

Review Jupyter notebook visual diffs & provide feedback on notebooks.

Powered by ReviewNB

abuendia commented 4 years ago

Thanks a lot for this PR. Could you please run the notebook before checking it in, so that it has all the images, plots, etc.

@PatrickBue Fixed! Thanks.

PatrickBue commented 4 years ago

@abuendia Shall I merge this?

microsoft / computervision-recipes

Add automatic model download for 01 tracking notebook #596

Description

Checklist: