Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity
1,846 Commits 1 Branch 0 Tags
c13d4ce7ef30acc78e3dbdd9aa4f17e01ed34521
Commit Graph
100 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
Glenn Jocher 701e1177ac Tensor initialization on device improvements (#6959) 
* Update common.py speed improvements

Eliminate .to() ops where possible for reduced data transfer overhead. Primarily affects warmup and PyTorch Hub inference.

* Updates

* Updates

* Update detect.py

* Update val.py
 2022-03-12 14:00:48 +01:00
Glenn Jocher 52c1399fdc DetectMultiBackend() return device update (#6958) 
Fixes ONNX validation that returns outputs on CPU.
 2022-03-12 13:16:29 +01:00
Glenn Jocher c84dd27d62 New val.py cuda variable (#6957) 
* New val.py `cuda` variable

Fix for ONNX GPU val.

* Update val.py
 2022-03-12 12:57:08 +01:00
Glenn Jocher c6b4f84fd1 Update Dockerfile torch==1.11.0+cu113 (#6954) 2022-03-12 00:45:07 +01:00
Glenn Jocher b94b59e199 DetectMultiBackend() --half handling (#6945) 
* DetectMultiBackend() `--half` handling

* CI fixes

* rename .half to .fp16 to avoid conflict

* warmup fix

* val update

* engine update

* engine update
 2022-03-11 16:31:52 +01:00
Glenn Jocher 84efa62b2d Fix PyTorch Hub export inference shapes (#6949) 
May resolve https://github.com/ultralytics/yolov5/issues/6947
 2022-03-11 16:18:40 +01:00
Glenn Jocher caf7ad0500 Allow 3-point segments (#6938) 
May resolve https://github.com/ultralytics/yolov5/issues/6931
 2022-03-10 18:41:47 +01:00
Glenn Jocher 055e72af5b Optimize PyTorch 1.11.0 compatibility update (#6933) 2022-03-10 12:58:41 +01:00
Glenn Jocher d3d9cbce22 PyTorch 1.11.0 compatibility updates (#6932) 
Resolves `AttributeError: 'Upsample' object has no attribute 'recompute_scale_factor'` first raised in https://github.com/ultralytics/yolov5/issues/5499
 2022-03-10 12:41:06 +01:00
Glenn Jocher 6dd82c0252 Move git_describe() to general.py (#6918) 
* Move `git_describe()` to general.py

* Move `git_describe()` to general.py
 2022-03-09 18:22:53 +01:00
Glenn Jocher e6e36aac10 Update bytes to GB with bitshift (#6886) 2022-03-07 19:26:37 +01:00
Glenn Jocher acc58c1dcf Fix TRT max_workspace_size deprecation notice (#6856) 
* Fix TRT `max_workspace_size` deprecation notice

* Update export.py

* Update export.py
 2022-03-07 13:52:53 +01:00
Glenn Jocher 4728840745 Update --cache disk deprecate *_npy/ dirs (#6876) 
* Updates

* Updates

* Updates

* Updates

* Updates

* Updates

* Updates

* Updates

* Updates

* Updates

* Cleanup

* Cleanup
 2022-03-06 16:16:17 +01:00
Glenn Jocher 8a66ebad44 Add *.ts to VID_FORMATS (#6859) 2022-03-04 14:10:13 +01:00
Glenn Jocherandpre-commit-ci[bot] 601dbb83f0 AutoAnchor improved initialization robustness (#6854) 
* Update AutoAnchor

* Update AutoAnchor

* [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-03-04 10:32:18 +01:00
Glenn Jocher 63ddb6f0d0 Update autoanchor.py (#6794) 
* Update autoanchor.py

* Update autoanchor.py
 2022-02-26 19:15:12 +01:00
Glenn Jocher c2403eb04c Update PULL_REQUEST_TEMPLATE.md (#6783) 2022-02-25 14:01:58 +01:00
Glenn Jocher 9ec51a6396 YOLOv5s6 params FLOPs fix (#6782) 2022-02-25 13:59:21 +01:00
Glenn Jocher 0f819919ad Update EMA decay tau (#6769) 
* Update EMA

* Update EMA

* ratio invert

* fix ratio invert

* fix2 ratio invert

* warmup iterations to 100

* ema_k

* implement tau

* implement tau
 2022-02-25 12:33:09 +01:00
Glenn Jocher cea994b3f6 Update hyp.VOC.yaml (#6772) 2022-02-25 00:14:17 +01:00
Glenn Jocher 741cd0eb23 Update tutorial.ipynb (#6771) 2022-02-25 00:12:27 +01:00
Glenn Jocher c161557563 Default OMP_NUM_THREADS=8 (#6770) 2022-02-25 00:09:14 +01:00
Glenn Jocher 47f265dde2 Update min warmup iterations from 1k to 100 (#6768) 2022-02-24 23:37:25 +01:00
Glenn Jocher 4bab56526b Update tutorial.ipynb (2 CPUs, 12.7 GB RAM, 42.2/166.8 GB disk) (#6767) 2022-02-24 18:29:59 +01:00
Glenn Jocher 66aaf5184d Pre-commit table fix (#6744) 2022-02-22 19:35:31 +01:00
Glenn Jocher 3752807c0b YOLOv5 v6.1 release (#6739) 2022-02-22 12:35:24 +01:00
Glenn Jocher 2692e67c5f Updated VOC hyperparameters (#6732) 
* Update hyps

* Update hyp.VOC.yaml

* Update pathlib

* Update hyps

* Update hyps

* Update hyps

* Update hyps
 2022-02-22 12:11:18 +01:00
Glenn Jocher a936f5f219 Switch default LR scheduler from cos to linear (#6729) 
* Switch default LR scheduler from cos to linear

Based on empirical results of training both ways on all YOLOv5 models.

* linear bug fix
 2022-02-21 14:23:27 +01:00
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
Glenn Jocher de9c25b35e Use export_formats() in export.py (#6705) 
* Use `export_formats()` in export.py

* list fix
 2022-02-19 16:08:33 +01:00
Glenn Jocherandpre-commit-ci[bot] 56697c43ff Fix CoreML P6 inference (#6700) 
* Fix CoreML P6 inference

* [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-02-18 20:38:33 +01:00
Glenn Jocher 3670d91b05 Fix --evolve --bucket gs://... (#6698) 2022-02-18 19:15:37 +01:00
Glenn Jocher 07221f1591 FROM nvcr.io/nvidia/pytorch:21.10-py3 (#6677) 
Revert to 21.10 on autobuild fail
 2022-02-17 16:05:30 +01:00
Glenn Jocher 18b5081166 FROM nvcr.io/nvidia/pytorch:21.10-py3 (#6673) 
Reordered installation may help reduce resource usage in autobuild
 2022-02-17 15:42:47 +01:00
Glenn Jocher 25456284b4 Update Dockerfile reorder installs (#6672) 
Also `nvidia-tensorboard-plugin-dlprof`, `nvidia-tensorboard` are no longer installed in NVCR base.
 2022-02-17 15:10:19 +01:00
Glenn Jocher 09d263b376 FROM nvcr.io/nvidia/pytorch:21.10-py3 (#6671) 
22.10 returns 'no space left on device' error message.

Seems like a bug at docker. Raised issue in https://github.com/docker/hub-feedback/issues/2209
 2022-02-17 14:51:53 +01:00
Glenn Jocher bb054d5f55 FROM nvcr.io/nvidia/pytorch:22.01-py3 (#6670) 2022-02-17 14:20:42 +01:00
Glenn Jocher 9a8ebe671c Update Dockerfile torch==1.10.2+cu113 (#6669) 2022-02-17 13:24:44 +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 7b80545e8e Add --cache val (#6663) 
New `--cache val` argument will cache validation set only into RAM. Should help multi-GPU training speeds without consuming as much RAM as full `--cache ram`.
 2022-02-17 00:47:42 +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
Glenn Jocher ee6c70ae43 Fix yolov3.yaml remove list (#6655) 
Per https://github.com/ultralytics/yolov3/issues/1887#issuecomment-1041135181
 2022-02-16 10:13:05 +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
Glenn Jocher 3eef80457f Add YOLOv5n to Reproduce section (#6619) 2022-02-11 21:48:01 +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
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 a5c9057dcc Update train.py 2022-02-10 14:15:18 +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 22da63e30f Fix zero-export handling with if any(f): (#6569) 
* Fix zero-export handling with `if any(f):`

Partial fix for https://github.com/ultralytics/yolov5/issues/6563

* Cleanup
 2022-02-08 12:20:39 +01:00
Glenn Jocher 006eb40ab7 Improved AutoBatch DDP error message (#6568) 
* Improved AutoBatch DDP error message

* Cleanup
 2022-02-08 11:45:30 +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
Glenn Jocher dc7e093071 Edge TPU TF imports fix (#6542) 
* Edge TPU TF imports fix

Fix for https://github.com/ultralytics/yolov5/issues/6535#issuecomment-1030631526

* Update common.py
 2022-02-05 15:45:44 +01:00
Glenn Jocherandpre-commit-ci[bot] 079b36d72b Edge TPU tf.lite.experimental.load_delegate fix (#6536) 
* Edge TPU `tf.lite.experimental.load_delegate` fix

Fix attempt for #6535

* [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-02-04 09:30:01 +01:00
Glenn Jocher 9bc72a3ac2 Edge TPU export 'list index out of range' fix (#6533) 2022-02-04 00:04:06 +01:00
Glenn Jocher 8d05716b66 Edge TPU compiler sudo fix (#6531) 
* Edge TPU compiler sudo fix

Allows for auto-install of Edge TPU compiler on non-sudo systems like the YOLOv5 Docker image.

@kalenmike

* Update export.py

* Update export.py

* Update export.py
 2022-02-03 20:40:09 +01:00
Glenn Jocherandpre-commit-ci[bot] a82292ec53 Social icons after text (#6473) 
* Social icons after text

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

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

* Update README.md

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-02-03 19:55:19 +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 cb40c9afda TODO issues exempt from stale action (#6530) 2022-02-03 18:11:28 +01:00
Glenn Jocher 19e0208fc9 Update hyp.scratch-high.yaml (#6525) 
Update `lrf: 0.1`, tested on YOLOv5x6 to 55.0 mAP@0.5:0.95, slightly higher than current.
 2022-02-03 12:15:13 +01:00
Glenn Jocher 4c40933266 Suppress export.run() TracerWarnings (#6499) 
Suppresses warnings when calling export.run() directly, not just CLI python export.py.

Also adds Requirements examples for CPU and GPU backends
 2022-02-01 23:52:50 +01:00
Glenn Jocher 842d049e1b Suppress torch.jit.TracerWarning on export (#6498) 
* Suppress torch.jit.TracerWarning

TracerWarnings can be safely ignored.

* Cleanup
 2022-02-01 22:59:26 +01:00
Glenn Jocher 77977e0791 CoreML inference fix list() -> sorted() (#6496) 2022-02-01 22:34:15 +01:00
Glenn Jocher 5e4ff195b2 Improved export.py usage examples (#6495) 
* Improved `export.py` usage examples

* Cleanup
 2022-02-01 22:06:29 +01:00
Glenn Jocher b884ea36c4 Simplify TF normalized to pixels (#6494) 2022-02-01 21:17:56 +01:00
Glenn Jocher 6445a8137e Resolve dataset paths (#6489) 2022-02-01 15:54:51 +01:00
Glenn Jocher 7539cd75c3 Add Product Hunt social media icon (#6464) 
* Social media icons update

* fix URL

* Update README.md
 2022-01-28 20:23:17 +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
Glenn Jocher d5966c93f1 Namespace VERBOSE env variable to YOLOv5_VERBOSE (#6428) 
* Verbose updates

* Verbose updates
 2022-01-25 17:57:27 -05:00
Glenn Jocher ff8646cdea Update workflows (#6427) 
* Workflow updates

* quotes fix

* best to weights fix
 2022-01-25 17:33:22 -05:00
Glenn Jocher cfecd903a3 Update greetings.yaml email address (#6412) 
* Update `greetings.yaml` email address

* Update greetings.yml
 2022-01-24 15:28:52 -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
Glenn Jocher c43439aa31 Add albumentations to Dockerfile (#6392) 2022-01-21 21:06:02 -10:00
Glenn Jocher bd815d48df FROM nvcr.io/nvidia/pytorch:21.10-py3 (#6379) 
21.12 generates dockerhub errors so rolling back to 21.10 with latest pytorch install. Not sure if this torch install will work on non-GPU dockerhub autobuild so this is an experiment.
 2022-01-20 19:15:04 -10:00
Glenn Jocher 1b41a1d059 FROM nvcr.io/nvidia/pytorch:21.12-py3 (#6377) 2022-01-20 17:59:41 -10:00
Glenn Jocherandpre-commit-ci[bot] 750c42e43e export.py automatic forward_export (#6352) 
* `export.py` automatic `forward_export`

* [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-01-19 12:24:40 -10:00
Glenn Jocher e5219099cd Created using Colaboratory 2022-01-19 10:18:29 -10:00
Glenn Jocher 0cf932bf63 export.py return exported files/dirs (#6343) 
* `export.py` return exported files/dirs

* Path to str
 2022-01-18 15:18:23 -10:00
Glenn Jocher e2e95b2d8e TensorRT assert im.device.type != 'cpu' on export (#6340) 
* TensorRT `assert im.device.type != 'cpu'` on export

* Update export.py
 2022-01-18 13:52:25 -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
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 80473a6551 Update export.py with Detect, Validate usages (#6280) 2022-01-12 15:48:40 -10:00
Glenn Jocherandpre-commit-ci[bot] f3085accd3 Enable ONNX --half FP16 inference (#6268) 
* Enable ONNX ``--half` FP16 inference

* [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-01-11 10:13:17 -10:00
Glenn Jocher b3eaf5008b TensorRT pip install 2022-01-10 16:49:10 -10:00
Glenn Jocher 6865d19a92 TensorRT 7 export fix (#6235) 2022-01-07 09:31:17 -08:00
Glenn Jocher 33a67b4918 Update P2-P7 models/hub variants (#6230) 
* Update p2-p7 `models/hub` variants

* Update common.py

* AutoAnchor camelcase corrections
 2022-01-06 11:08:09 -08:00
Glenn Jocher ad565e31d2 Update README speed reproduction command (#6228) 2022-01-06 09:55:31 -08:00
Glenn Jocher f80c463010 Attempt edgetpu-compiler autoinstall (#6223) 
* Attempt `edgetpu-compiler` autoinstall

Attempt to install edgetpu-compiler dependency if missing on Linux.

* Update export.py

* Update export.py
 2022-01-05 20:57:20 -08:00
Glenn Jocher b4ac3df6ff Add edgetpu_compiler checks (#6218) 
* Add `edgetpu_compiler` checks

* Update export.py

* Update export.py

* Update export.py

* Update export.py

* Update export.py

* Update export.py
 2022-01-05 14:55:04 -08:00
Glenn Jocher 00d7b97869 TFLite --int8 'flatbuffers==1.12' fix 2 (#6217) 
* TFLite `--int8` 'flatbuffers==1.12' fix 2

Reorganizes #6216 fix to update before `tensorflow` import so no restart required.

* Update export.py
 2022-01-05 13:34:36 -08:00
Glenn Jocher 8125ec5d42 TFLite --int8 'flatbuffers==1.12' fix (#6216) 
* TFLite `--int8` 'flatbuffers==1.12' fix

Temporary workaround for TFLite INT8 export.

* Update export.py

* Update export.py
 2022-01-05 13:01:21 -08:00