Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity
2,215 Commits 1 Branch 0 Tags
3456fe65c652fca855687d876bdc873c8f85f798
Commit Graph
884 Commits
Author SHA1 Message Date
Victor Sonckpre-commit-ci[bot]Glenn Jocher
 		3456fe65c6 Fix torchvision dependency for ClearML logging (#8993) 
* Replace torchvision annotator with native local Annotator

* Add colored bounding boxes and rewrite for efficiency

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Fix pep8 issue

* Update clearml_utils.py

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update clearml_utils.py

* Reduced conf_threshold to 0.25 to match defaults

The 0.25 default conf threshold value is used a default throughout YOLOv5 (i.e. in detect.py) and also in other tools like CoreML

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-17 11:51:07 +02:00
Glenn Jocher 464c2c6713 Colab and Kaggle loggers removeHandler (#8985) 
Resolve duplicate outputs in colab. Also optimizes Colab and Kaggle environment checks.

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-16 16:49:32 +02:00
Michel Promonetpre-commit-ci[bot]Glenn Jocher
 		4d05472d2b Allow multiple-model serving from Flask REST API (#8973) 
* allow to serve multiple models using flask restapi

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Cleanup

* Update restapi.py

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-15 23:42:25 +02:00
Glenn Jocher d7bc5d7773 Add .tar support for datasets (#8963) 2022-08-14 23:06:42 +02:00
Glenn Jocher 7639e4c7ef Fix windows LOGGER with emojis output (#8958) 
* Fix duplicate LOGGER output

* Update general.py
 2022-08-14 00:12:09 +02:00
Glenn Jocher dc38cd03f4 New smart_inference_mode() conditional decorator (#8957) 
New smart_inference_mode()
 2022-08-13 20:38:51 +02:00
Glenn Jocher 6aed0a7c00 GFLOPs computation fix for classification models (#8954) 
* GFLOPs computation fix for classification models

Improved robustness in reading input channel count

* Update torch_utils.py

* Update torch_utils.py
 2022-08-13 16:38:11 +02:00
Kalen Michael 11a76bb4f5 Allow train to use zipped content (#8940) 
* fix: added transparent image and empty alt to social bar

* fix: remove underlines

* fix: hide underlines for all social links

* fix: always unzip into folder
 2022-08-13 03:39:09 +02:00
Glenn Jocher 7ab1f50633 Scope torchvision.utils.draw_bounding_boxes import (#8915) 
* Scope `torchvision.utils.draw_bounding_boxes` import

Quick fix for https://github.com/ultralytics/yolov5/issues/8898#issuecomment-1208851874 until a more permanent fix is implemented

* Update clearml_utils.py
 2022-08-09 20:29:25 +02:00
Code-Bat b551098d59 About "os.R_OK" in general.py (#8909) 
The notes says "Return True if directory has write permissions", however, the code below is "os.R_OK", I think "os.W_OK" is preferred.
 2022-08-09 16:59:36 +02:00
Glenn Jocher f5335f22bb Capitalize YouTube (#8903) 2022-08-08 20:36:18 +02:00
Glenn Jocher 8f28ea4945 emoji-safe default logging (#8888) 
* emoji-safe default logging

* implement everywhere
 2022-08-06 12:30:08 +02:00
Glenn Jocher 20f1b7ea08 smart_optimizer() improved reporting (#8887) 
Update smart_optimizer() weight_decay reporting
 2022-08-06 11:30:54 +02:00
Victor Sonckpre-commit-ci[bot]Glenn Jocher
 		378bde4bba ClearML experiment tracking integration (#8620) 
* Add titles to matplotlib plots

* Add ClearML Experiment Tracking integration.

* Add ClearML Data Version Management automatic download when requested

* Add ClearML Hyperparameter Optimization

* ClearML save period integration

* Fix wandb breaking when used with ClearML dataset

* Fix wandb breaking when used with ClearML resume and dataset

* Add ClearML documentation

* fixed small bug in clearml integration that misreports epoch number

* Final ClearMl additions before refactor

* Add correct epoch reporting

* Add remote execution and autoscaling docs for ClearML integration

* Added images to clearml integration docs

* fixed logo alignment bug and added hpo screenshot clearml

* Fixed small epoch number bug in clearml integration

* Remove saved model flush clearml

* Cleanup clearml readme section

* Cleaned up clearml logger docstring

* Remove resume readme section clearml

* Clearml integration cleanup

* Updated ClearML documentation

* Added dark vs light icons ClearML Readme

* Clearml Readme styling

* Add better gifs

* Fixed gif file size

* Add better images in tutorial notebook

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Addressed comments in PR #8620

* Fixed circular import

* Fixed circular import

* Update tutorial.ipynb

* Update tutorial.ipynb

* Inline comment

* Restructured tutorial notebook

* Add correct ClearML link to README

* Update tutorial.ipynb

* Update general.py

* Update __init__.py

* Update __init__.py

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update __init__.py

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update __init__.py

* Update README.md

* Update __init__.py

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* spelling

* Update tutorial.ipynb

* notebook cutt.ly links

* Update README.md

* Update README.md

* cutt.ly links in tutorial

* Removed labels as they show up on last subplot only

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-08-05 20:50:49 +02:00
Glenn Jocher bc9fcb1767 HUBDatasetStats() preview images to 50 quality (#8880) 
@kalenmike should represent a 30% filesize reduction vs 75 quality
 2022-08-05 14:45:41 +02:00
Glenn Jocher 84e7748564 Update dataloaders.py remove float64 shapes (#8865) 
May help https://github.com/ultralytics/yolov5/issues/8862
 2022-08-04 17:06:08 +02:00
Glenn Jocherandpre-commit-ci[bot] a75a1105a1 Self-contained checkpoint --resume (#8839) 
* Single checkpoint resume

* Update train.py

* Add hyp

* Add hyp

* Add hyp

* FIX

* avoid resume on url data

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* avoid resume on url data

* avoid resume on url data

* Update

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-08-03 21:28:22 +02:00
Glenn Jocher 08c8c3e00a New smart_resume() (#8838) 
* New `smart_resume()`

* Update torch_utils.py

* Update torch_utils.py

* Update torch_utils.py

* fix
 2022-08-02 15:13:58 +02:00
Glenn Jocher ba140e5685 RUN git clone instead of COPY to /usr/src/app (#8827) 
Update
 2022-08-01 21:45:31 +02:00
Glenn Jocher 39ce8ca19a Remove assert not is_docker() from GitHub checks (#8813) 
* Update

* Update
 2022-08-01 03:01:44 +02:00
Amir PourmandandGlenn Jocher afa403a809 Check git status on upstream ultralytics or origin dynamically (#8694) 
* Add remote ultralytics and check git status with that

* Simplify

* Update general.py

* Update general.py

* s fix

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-07-31 18:17:38 +02:00
Glenn Jocher 555976b346 FROM nvcr.io/nvidia/pytorch:22.07-py3 (#8808) 2022-07-31 15:23:57 +02:00
jbutle55andGlenn Jocher 567397d67a Fix confusion matrix update when no predictions are made (#8748) 
* Fix confusion matrix update when no predictions are made

* Update metrics.py

* Simply confusion matrix changes

* Simply confusion matrix fix

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-07-29 14:06:23 +02:00
Max StrobelMaximilian StrobelGlenn Jocher
 		0b5ac224ae fix: broken `is_docker` check (#8711) 
Checking if ``/workspace`` exists is not a reliable method to check if
the process runs in a docker container.

Reusing the logic from the npm "is-docker" package to check if the
process runs in a container.
References: https://github.com/sindresorhus/is-docker/blob/main/index.js

Fixes #8710.

Co-authored-by: Maximilian Strobel <Maximilian.Strobel@infineon.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-07-26 19:02:44 +02:00
Glenn Jocher c775a296a7 Revert "Fix BGR->RGB Bug in albumentations #8641" (#8727) 
Revert "Fix BGR->RGB Bug in albumentations #8641 (#8695)"

This reverts commit 2e1291fdce.
 2022-07-26 19:00:48 +02:00
UnglvKitDepre-commit-ci[bot]Glenn Jocher
 		2e1291fdce Fix BGR->RGB Bug in albumentations #8641 (#8695) 
* Fix BGR->RGB Bug in albumentations https://github.com/ultralytics/yolov5/issues/8641

* Change transform methode from cv2 to numpy

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Simplify

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update augmentations.py

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-07-26 13:52:56 +02:00
Glenn Jocherandpre-commit-ci[bot] b367860196 New HUBDatasetStats() class (#8716) 
* New `HUBDatasetStats()` class

Usage examples:
```
from utils.dataloaders import *

stats = HUBDatasetStats('coco128.yaml', autodownload=True)  # method 1
stats = HUBDatasetStats('path/to/coco128_with_yaml.zip')  # method 1

stats.get_json(save=False)
stats.process_images()
```

@kalenmike

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update dataloaders.py

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update dataloaders.py

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update dataloaders.py

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update dataloaders.py

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-07-25 18:20:01 +02:00
Glenn Jocher a6f197ae79 Update dataloaders.py (#8714) 
* Update dataloaders.py

* Update dataloaders.py
 2022-07-25 16:52:28 +02:00
Glenn Jocher 0ab303b044 Remove else: from load_image() (#8692) 2022-07-25 00:02:09 +02:00
Glenn Jocher 916bdb1d61 Faster crop saving (#8696) 
Faster crops

Following https://github.com/ultralytics/yolov5/issues/8641#issuecomment-1193190325
 2022-07-23 23:30:30 +02:00
Glenn Jocher b510957650 Move .dockerignore to root (#8690) 2022-07-23 18:50:19 +02:00
Glenn Jocher 7f7bd6fbcd Set torch.cuda.manual_seed_all() for DDP (#8688) 
* Set `torch.cuda.manual_seed_all()` for DDP

* Update general.py

* Update general.py
 2022-07-23 18:24:14 +02:00
UnglvKitDepre-commit-ci[bot]Glenn Jocher
 		1c5e92aba1 Add generator and worker seed (#8602) 
* Add generator and worker seed

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update dataloaders.py

* Update dataloaders.py

* Update dataloaders.py

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-07-23 01:25:17 +02:00
Glenn Jocher b17629e54f Update warning emojis (#8678) 2022-07-22 15:23:22 +02:00
Rüdiger Busche 602d7ffb0e Check TensorBoard logger before adding graph (#8664) 
Otherwise, an error is thrown if the tensorboard logger is not included.
 2022-07-21 17:40:53 +02:00
Glenn Jocher fbe67e4653 Fix OMP_NUM_THREADS=1 for macOS (#8624) 
Resolves https://github.com/ultralytics/yolov5/issues/8623
 2022-07-18 17:53:30 +02:00
Glenn Jocherandpre-commit-ci[bot] 9cf5fd5ac3 assert torch!=1.12.0 for DDP training (#8621) 
* assert torch!=1.12.0 for DDP training

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-07-18 15:05:58 +02:00
Glenn Jocherandpre-commit-ci[bot] 51fb467b63 Refactor optimizer initialization (#8607) 
* Refactor optimizer initialization

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update train.py

* Update train.py

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-07-17 11:43:52 +02:00
Glenn Jocher 24305787ae FROM nvcr.io/nvidia/pytorch:22.06-py3 (#8600) 2022-07-17 00:55:55 +02:00
Glenn Jocher 6e86af3de8 Add pip install wheel to avoid legacy setup.py install (#8597) 
* Update ci-testing with `pip install wheel`

* Update ci-testing.yml

* Update dockerfiles
 2022-07-16 16:41:43 +02:00
Glenn Jocher 04146371b9 Update metrics.py with IoU protected divides (#8550) 2022-07-12 01:58:25 +02:00
Glenn Jocher a84cd02387 CIoU protected divides (#8546) 
Protected divides in IOU function to resolve https://github.com/ultralytics/yolov5/issues/8539
 2022-07-11 16:07:11 +02:00
Colin WongColin WongGlenn Jocher
 		526e650553 Fix LoadImages() with dataset YAML lists (#8517) 
* Fix LoadImages with dataset yaml lists

* Update dataloaders.py

* Update dataloaders.py

* Simplify/refactor PR

* Update dataloaders.py

Co-authored-by: Colin Wong <noreply@brains4drones.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-07-08 14:32:40 +02:00
Glenn Jocher f17444abcd Simplify benchmarks.py assertions (#8515) 2022-07-07 23:46:55 +02:00
Glenn Jocherandpre-commit-ci[bot] 3e54651fca Add --hard-fail argument to benchmarks for CI errors (#8513) 
* Add `--hard-fail` list argument to benchmarks for CI

Will cause CI to fail on a benchmark failure for given indices.

* Update ci-testing.yml

* Attempt Failure (CI should fail)

* Update benchmarks.py

* Update export.py

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update benchmarks.py

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update ci-testing.yml

* Update benchmarks.py

* Update benchmarks.py

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-07-07 23:41:34 +02:00
Glenn Jocher dd28df98c2 Avoid FP64 ops for MPS support in train.py (#8511) 
Avoid FP64 ops for MPS support

Resolves https://github.com/ultralytics/yolov5/pull/7878#issuecomment-1177952614
 2022-07-07 20:36:23 +02:00
Ayush Chaurasiapre-commit-ci[bot]Glenn Jocher
 		27d831b6e4 Training reproducibility improvements (#8213) 
* attempt at reproducibility

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* use deterministic algs

* fix everything :)

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* revert dataloader changes

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* process_batch as np

* remove newline

* Remove dataloader init fcn

* Update val.py

* Update train.py

* revert additional changes

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update train.py

* Add --seed arg

* Update general.py

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update train.py

* Update train.py

* Update val.py

* Update train.py

* Update general.py

* Update general.py

* Add deterministic argument to init_seeds()

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-07-07 14:39:29 +02:00
Glenn Jocher 29d79a6360 Do not prefer Apple MPS (#8446) 
Require explicit request for MPS, i.e.
```bash
python detect.py --device mps
```

Reverts https://github.com/ultralytics/yolov5/pull/8210 for preferring MPS if available. 

Note that torch MPS is experiencing ongoing compatibility issues in https://github.com/pytorch/pytorch/issues/77886
 2022-07-02 18:35:45 +02:00
Glenn Jocher 898332433a Update cache comments (#8414) 
* Update cache comments

For better readability

* Update dataloaders.py
 2022-06-30 16:19:22 +02:00
Glenn Jocherandpre-commit-ci[bot] 34df5032a7 Add File Size (MB) column to benchmarks (#8359) 
* Add filesize to benchmarks.py

* Add filesize to benchmarks.py

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-06-27 17:46:49 +02:00