computervision-recipes
computervision-recipes copied to clipboard
Automatic FairMOT baseline model download and MOT17 download patch
Description
This PR adds:
- Automatic FairMOT model download in 01 and 02 tracking notebooks. The baseline dla model is uploaded to the public cvbp blob container and gets pulled down into both notebooks. Removes the need for the user to manually download the model from a Google Drive link.
- Automatic download of MOT17 into 02 notebook. The relevant train/test subsets of MOT17 (2 GB) are uploaded to the public cvbp blob container and get pulled into the 02 notebook. I extract only the subdirectories needed for 02 to avoid a 5 GB download.
- Minor changes to 01 and 02 markdown, i.e. typo/grammar fixes.
Reuses unzip_url util. Downloads if does not exist, and avoids re-downloading if exists. Adds relevant download URLs to data.py.
Important: When running the current version of 02 in staging, I encountered a bug that indicates the MOT17 zip from the MOT Challenge website was recently changed and corrupted. I have uploaded an earlier working version from @karineip to blob store.
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
stagingand notmaster - [x] I will squash merge this PR into
staging
Check out this pull request on ![]()
Review Jupyter notebook visual diffs & provide feedback on notebooks.
Powered by ReviewNB