Update to v5 2 #5

robin-maillot · 2021-09-21T12:38:46Z

Code changes to incorporate mojo metrics as well as new yolov5 format.

* train.py GPU memory fix * ema * cuda * cuda * zeros input * to device * batch index 0

@kalenmike

@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

@kalenmike

* Update `dataset_stats()` to list of dicts @kalenmike * Update datasets.py

* Remove `/weights` directory * cleanup

…3668)

* Cache v0.3: improved corrupt image/label reporting Fix for ultralytics#3656 (comment) * cleanup

* 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.

…used correctly during training

# Conflicts: # README.md

…arch more generic.

into update-to-v5-2

glenn-jocher and others added 30 commits June 11, 2021 11:46

ConfusionMatrix normalize=True fix (ultralytics#3587)

e8c5237

train.py GPU memory fix (ultralytics#3590)

4984cf5

* train.py GPU memory fix * ema * cuda * cuda * zeros input * to device * batch index 0

W&B: Allow changed in config variable ultralytics#3588

53d4fc2

Delete __init__.py (ultralytics#3596)

88b1945

merge develop

f8adee1

Update datasets.py (ultralytics#3591)

6062319

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

Download COCO and VOC by default (ultralytics#3608)

239a11c

Suppress wandb images size mismatch warning (ultralytics#3611)

daab682

* 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>

Fix incorrect end epoch comment (ultralytics#3612)

4c5d9bf

Update check_file() (ultralytics#3622)

7d3686a

* Update `check_file()` * Update datasets.py

Update README.md (ultralytics#3624)

30e4c4f

FROM nvcr.io/nvidia/pytorch:21.05-py3 (ultralytics#3633)

de56813

Add **/*.torchscript.pt (ultralytics#3634)

65f81bf

Update verify_image_label() (ultralytics#3635)

6c0e1d9

RUN pip install --no-cache -U torch torchvision (ultralytics#3637)

bb79e13

Assert non-premature end of JPEG images (ultralytics#3638)

d808855

* premature end of JPEG images * PEP8 reformat Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Update CONTRIBUTING.md (ultralytics#3645)

3ce0db8

* Update CONTRIBUTING.md * Update CONTRIBUTING.md * Update CONTRIBUTING.md * Update CONTRIBUTING.md

Update CONTRIBUTING.md (ultralytics#3647)

6187edc

is_coco list fix (ultralytics#3646)

fa29125

Update README.md (ultralytics#3650)

df7706d

Be more user-friendly to new users

Update dataset_stats() to list of dicts (ultralytics#3657)

9b6dba6

* Update `dataset_stats()` to list of dicts @kalenmike * Update datasets.py

Remove /weights directory (ultralytics#3659)

2754ada

* Remove `/weights` directory * cleanup

Update download_weights.sh comment (ultralytics#3662)

ac34834

Update train.py (ultralytics#3667)

6d6e2ca

Update train(hyp, *args) to accept hyp file or dict (ultralytics#…

fa201f9

…3668)

Update TensorBoard (ultralytics#3669)

045d5d8

Update WORLD_SIZE and RANK retrieval (ultralytics#3670)

2296f15

Cache v0.3: improved corrupt image/label reporting (ultralytics#3676)

f527704

* Cache v0.3: improved corrupt image/label reporting Fix for ultralytics#3656 (comment) * cleanup

glenn-jocher and others added 28 commits August 30, 2021 17:05

Update check_git_status() warning (ultralytics#4610)

3a72d4a

W&B: Don't log models in evolve operation (ultralytics#4611)

11e7c7b

Close matplotlib plots after opening (ultralytics#4612)

bb5ebc2

* Close plots * Replace fig.close() for plt.close()

DDP torch.jit.trace() --sync-bn fix (ultralytics#4615)

50a9828

* Remove assert * debug0 * trace=not opt.sync * sync to sync_bn fix * Cleanup

Fix for Arial.ttf redownloads with hub inference (ultralytics#4627)

ba0f808

Fix 2 for Arial.ttf redownloads with hub inference (ultralytics#4628)

a4e8f78

Fix 3 for Arial.ttf redownloads with hub inference (ultralytics#4629)

de534e9

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

Checkpoint code.

fda85a5

Fix for plot_evolve() string argument (ultralytics#4639)

234e8ae

Fix is_coco on missing data['val'] key (ultralytics#4642)

fad57c2

Fix workers to 1 for windows and fix issue with image_size not being …

c610bc4

…used correctly during training

Merge branch 'master' into modify_wandb

730970b

Merge branch 'modify_wandb' into update-to-v5-2

c0ee52e

# Conflicts: # README.md

Remove mojo files.

aa2432f

Add mojo_test.py and update gitignore.

b20b128

Move entity and project to variables.

d16ac8f

Update installation of dependencies to only if needed and make whl se…

4e0abe6

…arch more generic.

Fix missing parameter in _find_module_wheel_path.

87fc7ff

Remove extra prints.

0195cbc

Fix weights download bug and pretraining always using yolov5s weights.

c379794

Update code to work with Ultralytics YOLOv5:4 env.

4e5279e

Add confidence threshold plot

8b16998

Minor cleanup of azure_wrapper.

c16dad8

Merge branch 'update-to-v5-2' of https://github.com/robin-maillot/yolov5

a114edc

into update-to-v5-2

Fix click/typer incompatibility before 4.0.0

1b9d689

Merge branch 'update-to-v5-2' of https://github.com/robin-maillot/yolov5

066de6e

into update-to-v5-2

Restore gitignore and remove wrong error import print in Azure wrapper.

2c35fc5

Fix wrong typer version in requirements.

de2e75c

robin-maillot changed the base branch from master to develop September 22, 2021 08:26

robin-maillot merged commit 6c7846a into develop Sep 22, 2021

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Update to v5 2 #5

Update to v5 2 #5

robin-maillot commented Sep 21, 2021

Update to v5 2 #5

Update to v5 2 #5

Conversation

robin-maillot commented Sep 21, 2021