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

Use COCOMetric for mAP calculation #4

Open
wants to merge 759 commits into
base: master
Choose a base branch
from
Open

Use COCOMetric for mAP calculation #4

wants to merge 759 commits into from

Conversation

potipot
Copy link
Collaborator

@potipot potipot commented Sep 20, 2021

This PR replaces calculation of mAP done by ultarlytics implementation with more established and community accepted pycocotools COCOMetric.

glenn-jocher and others added 30 commits June 9, 2021 22:50
* Refactoring cleanup

* Update test.py
* Ignore Seaborn plot warnings

* Update plots.py

* Update metrics.py
* train.py GPU memory fix

* ema

* cuda

* cuda

* zeros input

* to device

* batch index 0
@kalenmike this is a PR to add image filenames and labels to our stats dictionary and to save the dictionary to JSON. Save location is next to the train labels.cache file. The single JSON contains all stats for entire dataset.

Usage example:
```python
from utils.datasets import *

dataset_stats('coco128.yaml', verbose=True)
```
* Update README.md

* added hosted images

* added new logo

* testing image hosting

* changed svgs to pngs

* removed old header

* Update README.md

* correct colab image source

* splash.jpg

* rocket and W&B fix

* added contributing template

* added social media to top section

* increased size of top social media

* cleanup and updates

* rearrange quickstarts

* API cleanup

* PyTorch Hub cleanup

* Add tutorials

* cleanup

* update CONTRIBUTING.md

* Update README.md

* update wandb link

* Update README.md

* remove tutorials header

* update environments and integrations

* Comment API image

* Update README.md

* Update README.md

* Update README.md

* Update README.md

* Update README.md

* double spaces after section

* Update README.md

* Update README.md

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* 'changes-in_dataset'

* Update datasets.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* supress wandb images size mismatch warning

* supress wandb images size mismatch warning

* PEP8 reformat and optimize imports

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update `check_file()`

* Update datasets.py
* premature end of JPEG images

* PEP8 reformat

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update CONTRIBUTING.md

* Update CONTRIBUTING.md

* Update CONTRIBUTING.md

* Update CONTRIBUTING.md
Be more user-friendly to new users
* Update `dataset_stats()` to list of dicts

@kalenmike

* Update datasets.py
* Remove `/weights` directory

* cleanup
glenn-jocher and others added 28 commits August 29, 2021 17:56
* Auto-download Arial.ttf on init

* Fix ROOT
…#4608)

Layer 21 includes the information of xsmall objects
* Close plots

* Replace fig.close() for plt.close()
* Remove assert

* debug0

* trace=not opt.sync

* sync to sync_bn fix

* Cleanup
Fix 3 for Arial.ttf redownloads with hub inference, follow-on to ultralytics#4628.
Copy link

@github-actions github-actions bot left a comment

Choose a reason for hiding this comment

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

Hello @potipot, thank you for submitting a PR! To allow your work to be integrated as seamlessly as possible, we advise you to:

  • Verify your PR is up-to-date with origin/master. If your PR is behind origin/master update by running the following, replacing 'feature' with the name of your local branch:
git remote add upstream https://github.com/ultralytics/yolov5.git
git fetch upstream
git checkout feature  # <----- replace 'feature' with local branch name
git rebase upstream/master
git push -u origin -f
  • Verify all Continuous Integration (CI) checks are passing.
  • Reduce changes to the absolute minimum required for your bug fix or feature addition. "It is not daily increase but daily decrease, hack away the unessential. The closer to the source, the less wastage there is." -Bruce Lee

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.