Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity
1,811 Commits 1 Branch 0 Tags
dbbb57cf0b8004dddb7e3bc4506b68b44b60896a
Commit Graph
712 Commits
Author SHA1 Message Date
Glenn Jocher c43f135571 Update nw to max(nd, 1) (#6714) 2022-02-20 21:47:46 +01:00
Glenn Jocher 4de8b24881 Suppress torch AMP-CPU warnings (#6706) 
This is a torch bug, but they seem unable or unwilling to fix it so I'm creating a suppression in YOLOv5. 

Resolves https://github.com/ultralytics/yolov5/issues/6692
 2022-02-19 16:48:33 +01:00
Samuel Yvon 0365379016 Fix floating point in number of workers nw (#6701) 
Integer division by a float yields a (rounded) float. This causes
the dataloader to crash when creating a range.
 2022-02-18 21:06:21 +01:00
Glenn Jocher 2e5c67e537 Robust scipy.cluster.vq.kmeans too few points (#6668) 
* Handle `scipy.cluster.vq.kmeans` too few points

Resolves #6664

* Update autoanchor.py

* Cleanup
 2022-02-17 12:55:03 +01:00
Glenn Jocher 1ff43702a8 Validate with 2x --workers single-GPU/CPU fix (#6659) 
Fix for #6658 for single-GPU and CPU training use cases
 2022-02-16 14:50:37 +01:00
Glenn Jocher 6769021901 Validate with 2x --workers (#6658) 2022-02-16 14:44:44 +01:00
imyhxy ca0a00784d Fixed wandb logger KeyError (#6637) 2022-02-14 17:08:36 +01:00
Glenn Jocher 538930023c Fix ConfusionMatrix scale vmin=0.0 (#6638) 
Fix attempt for https://github.com/ultralytics/yolov5/issues/6626
 2022-02-14 14:54:55 +01:00
Glenn Jocherandpre-commit-ci[bot] a45e472358 YOLOv5 Export Benchmarks (#6613) 
* Add benchmarks.py

* Update

* Add requirements

* Updates

* Updates

* Updates

* Updates

* Updates

* Updates

* dataset autodownload from root

* Update

* Redirect to /dev/null

* sudo --help

* Cleanup

* Add exports pd df

* Updates

* Updates

* Updates

* Cleanup

* dir handling fix

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

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

* Cleanup

* Cleanup2

* Cleanup3

* Cleanup model_type

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-02-12 16:05:43 +01:00
Ayush Chaurasia 96d8f86085 W&B: don't log media in evolve (#6617) 2022-02-12 13:04:50 +01:00
Ayush ChaurasiaandGlenn Jocher 51f7750aad W&B: Improve resume stability (#6611) 
* log best.pt metrics at train end

* update

* Update __init__.py

* flush callbacks when using evolve

* remember batch size on resuming

* Update train.py

* improve stability of resume

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-02-12 13:03:09 +01:00
Glenn Jocher b40bdd6b2d YouTube dependency fix youtube_dl==2020.12.2 (#6612) 
Per https://github.com/ultralytics/yolov5/issues/5860#issuecomment-1035320018 by @hdnh2006
 2022-02-11 13:46:20 +01:00
or-toledano cb2ad9f685 Fix ROOT / data when running W&B log_dataset() (#6606) 
* Fix missing data folder when running log_dataset

* Use ROOT/'data'

* PEP8 whitespace
 2022-02-10 16:17:47 +01:00
Glenn Jocher c21da596f3 Fix hyp_evolve.yaml indexing bug (#6604) 
* Fix `hyp_evolve.yaml` indexing bug

Bug caused hyp_evolve.yaml to display latest generation result rather than best generation result.

* Update plots.py

* Update general.py

* Update general.py

* Update general.py
 2022-02-10 14:51:29 +01:00
Glenn Jocher 364b7c5111 Fix YouTube dislike button bug in pafy package (#6603) 
Per https://github.com/ultralytics/yolov5/issues/6583#issuecomment-1034421945 by @alicera
 2022-02-10 13:13:44 +01:00
Glenn Jocher 9c513ca629 Add DATASETS_DIR global in general.py (#6578) 2022-02-08 22:20:44 +01:00
Glenn Jocher f627bc53f8 Fix plot_labels() colored histogram bug (#6574) 
* Fix `plot_labels()` colored histogram bug

* Cleanup
 2022-02-08 14:03:50 +01:00
Glenn Jocher b57abb17f2 Move trainloader functions to class methods (#6559) 
* Move trainloader functions to class methods

* results = ThreadPool(NUM_THREADS).imap(self.load_image, range(n))

* Cleanup
 2022-02-07 18:59:26 +01:00
Jirka BorovecandGlenn Jocher cba4303d32 Fix 6 Flake8 issues (#6541) 
* F541

* F821

* F841

* E741

* E302

* E722

* Apply suggestions from code review

* Update general.py

* Update datasets.py

* Update export.py

* Update plots.py

* Update plots.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-02-05 15:22:59 +01:00
Glenn Jocherandpre-commit-ci[bot] c3e599cfda Update val_batch*.jpg for Chinese fonts (#6526) 
* Update plots for Chinese fonts

* make is_chinese() non-str safe

* Add global FONT

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

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

* Update general.py

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-02-03 19:09:24 +01:00
Glenn Jocher 6445a8137e Resolve dataset paths (#6489) 2022-02-01 15:54:51 +01:00
Glenn Jocher d8b5beb0b0 Fix2 select_device() for Multi-GPU (#6461) 
* Fix2 select_device() for Multi-GPU

* Cleanup

* Cleanup

* Simplify error message

* Improve assert

* Update torch_utils.py
 2022-01-28 08:18:01 +01:00
Glenn Jocher 856d4e5733 Fix select_device() for Multi-GPU (#6434) 
* Fix `select_device()` for Multi-GPU

Possible fix for https://github.com/ultralytics/yolov5/issues/6431

* Update torch_utils.py

* Update torch_utils.py

* Update torch_utils.py

* Update torch_utils.py

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update
 2022-01-26 23:47:00 +01:00
Glenn Jocher fe7de6a82d Revert "Remove dataset_stats() autodownload capability (#6303)" (#6442) 
This reverts commit 3119b2f27c.
 2022-01-26 19:10:59 +01:00
toschi23 3b7ac28ed1 Add *.asf video support (#6436) 2022-01-26 14:26:02 +01:00
Glenn Jocher d5966c93f1 Namespace VERBOSE env variable to YOLOv5_VERBOSE (#6428) 
* Verbose updates

* Verbose updates
 2022-01-25 17:57:27 -05:00
Jonathan SamelsonandGlenn Jocher ed9bac8392 Rename logger from 'utils.logger' to 'yolov5' (#6421) 
* Gave a more explicit name to the logger

* Cleanup

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-01-25 07:21:06 -08:00
Glenn Jocher 482af479c0 Add detect.py GIF video inference (#6410) 
* Add detect.py GIF video inference

* Cleanup
 2022-01-24 13:11:11 -08:00
MatthiasandGlenn Jocher 8efe97719c Add stop_training=False flag to callbacks (#6365) 
* New flag 'stop_training' in util.callbacks.Callbacks class to prematurely stop training from callback handler

* Removed most of the new  checks, leaving only the one after calling 'on_train_batch_end'

* Cleanup

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-01-22 16:37:21 -10:00
Ayush ChaurasiaandGlenn Jocher 9bcc32a5bf Flush callbacks when on --evolve (#6374) 
* log best.pt metrics at train end

* update

* Update __init__.py

* flush callbacks when using evolve

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-01-20 13:22:47 -10:00
sitecao e1893c894a DEVICE_COUNT instead of WORLD_SIZE to calculate nw (#6324) 2022-01-20 13:06:26 -10:00
imyhxy 4e841b9b16 Reuse de_parallel() rather than is_parallel() (#6354) 2022-01-20 10:50:17 -10:00
johnk2hawaii 9708cf56ea New environment variable VERBOSE (#6353) 
New environment variable `VERBOSE`
 2022-01-19 14:32:19 -10:00
Glenn Jocher fd55271c04 Console corrupted -> corrupt (#6338) 
* Console corrupted -> corrupt 

Minor style changes.

* Update export.py
 2022-01-18 10:49:26 -10:00
Glenn Jocher 3119b2f27c Remove dataset_stats() autodownload capability (#6303) 
* Remove `dataset_stats()` autodownload capability

@kalenmike security update per Slack convo

* Update datasets.py
 2022-01-18 10:04:49 -10:00
Glenn Jocher 436ffc417a select_device() cleanup (#6302) 
* `select_device()` cleanup

* Update torch_utils.py

* Update torch_utils.py

* Update torch_utils.py

* Update torch_utils.py

* Update torch_utils.py
 2022-01-14 15:48:15 -10:00
Henry a1a9c6884c Fixing bug multi-gpu training (#6299) 
* Fixing bug multi-gpu training

This solves this issue: https://github.com/ultralytics/yolov5/issues/6297#issue-1103853348

* Update torch_utils.py for pep8
 2022-01-14 10:11:06 -10:00
Glenn Jocher e7bf38277f Fix device count check (#6290) 
* Fix device count check()

* Update torch_utils.py

* Update torch_utils.py

* Update hubconf.py
 2022-01-13 21:23:03 -10:00
Glenn Jocher af001349e4 Add is_kaggle() function (#6285) 
* Add `is_kaggle()` function

Return True if environment is Kaggle Notebook.

* Remove root loggers only if is_kaggle() == True

* Update general.py
 2022-01-13 12:39:42 -10:00
Glenn Jocher 9e9219fe17 Fix nan-robust stream FPS (#6198) 
Fix for Webcam stop working suddenly (Issue #6197)
 2022-01-04 19:32:42 -08:00
Glenn Jocher b4a29b5a8d Update NMS max_wh=7680 for 8k images (#6178) 2022-01-03 10:54:52 -08:00
Chen GenCmosGlenn Jocher
 		affa284352 Refactor/reduce G/C/D/IoU if: else statements (#6087) 
* Refactor the code to reduece else

* Update metrics.py

* Cleanup

Co-authored-by: Cmos <gen.chen@ubisoft.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-12-30 13:59:29 -08:00
Awsaf 7b6938d5b5 Log best results (#6085) 
* log best result in summary

* comment added

* add space for `flake8`

* log `best/epoch`

* fix `dimension` for epoch

ValueError: all the input arrays must have same number of dimensions

* log `best/` in `utils.logger.__init__`

* fix pre-commit

1. missing whitespace around operator
2.  over-indented
 2021-12-30 13:47:53 -08:00
Ayush ChaurasiaandGlenn Jocher 9155eb8641 W&B: Log best results after training ends (#6120) 
* log best.pt metrics at train end

* update

* Update __init__.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-12-30 11:43:19 -08:00
Ayush ChaurasiaandGlenn Jocher db6ec66a60 W&B: track batch size after autobatch (#6039) 
* track batch size after autobatch

* remove redundant import

* Update __init__.py

* Update __init__.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-12-23 14:23:50 +01:00
JieLiand李杰 afa5cfb0f8 Reduce G/D/CIoU logic operations (#6074) 
Consider that the default value is CIOU，adjust the order of judgment could reduce the number of judgments.
And “elif CIoU:” didn't need 'if'.

Co-authored-by: 李杰 <360751194@qq.comqq.com>
 2021-12-23 11:53:00 +01:00
Glenn Jocher b8a4babd60 Simplify set_logging() indexing (#6042) 2021-12-20 17:42:52 +01:00
Glenn Jocher 0db9d5b6a2 Kaggle LOGGER fix (#6041) 2021-12-20 17:30:46 +01:00
Glenn Jocherandpre-commit-ci[bot] 407a905747 Check TensorRT>=8.0.0 version (#6021) 
* Check TensorRT>=8.0.0 version

* [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>
 2021-12-17 14:59:46 +01:00
Glenn Jocher da9a1b719b Allow --weights URL (#5991) 2021-12-15 15:27:08 +01:00