Glenn Jocher
c1fcfe8cd9
YouTube Usage example comments ( #10106 )
...
* YouTube Usage example comments
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update dataloaders.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-09 17:20:26 +01:00
janus-zheng
892c4cd4a5
AutoShape integer image-size fix ( #10090 )
...
Update common.py
We have a division at line 694, and then a multiplication at line 695, so it makes `y*g` not an integer. And since `shape1` will be used at line 697 to ensure the size is divisible by the `stride`, this may lead to different image size.
In my experiment, my image is [640, 640], it's divisible by the default stride 32, but I found that the result is changed to [672, 672] after line 697. So the final detection result is slightly different from that directly using the `detect.py` script, which does not call the AutoShape methods.
Signed-off-by: janus-zheng <106574221+janus-zheng@users.noreply.github.com>
Signed-off-by: janus-zheng <106574221+janus-zheng@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-09 15:21:43 +01:00
Glenn Jocher
86decb3c49
Add `ultralytics` pip package ( #10103 )
2022-11-09 15:05:19 +01:00
Glenn Jocher
0039870b89
Update Dockerfile keep default torch installation ( #10071 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-08 00:58:00 +01:00
Glenn Jocher
e00d02d78b
Use MNIST160 ( #10069 )
...
New 160-image MNIST subset composed of first 8 examples of each class. Suitable for fast CI.
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-07 23:07:10 +01:00
pre-commit-ci[bot]
58b3d07854
[pre-commit.ci] pre-commit suggestions ( #10068 )
...
updates:
- [github.com/asottile/pyupgrade: v2.38.2 → v3.2.0](https://github.com/asottile/pyupgrade/compare/v2.38.2...v3.2.0 )
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-07 22:56:23 +01:00
Glenn Jocher
6ae3dff7d4
Update get_coco.sh ( #10057 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-06 18:43:07 +01:00
Glenn Jocher
78ed31c95a
IoU `eps` adjustment ( #10051 )
...
IoU eps adjustment
Unify h1 and h2 with eps values
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-06 15:06:18 +01:00
Glenn Jocher
fde7758468
YOLOv5 AutoCache Update ( #10027 )
...
* AutoCache
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* AutoCache
* AutoCache
* AutoCache
* AutoCache
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-04 15:30:12 +01:00
Glenn Jocher
02b8a4c21b
Update default `--epochs 100` ( #10024 )
...
* Update default `--epochs 100`
@AyushExel @kalenmike updating default Detection and Segmentation trainings to 100 epochs
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update train.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update README.md
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update README_cn.md
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-03 17:58:15 +01:00
Glenn Jocher
82a558557a
Fix `min_items` ( #10001 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-01 16:41:37 +01:00
Glenn Jocher
067ad9a2d1
Improved `check_online()` robustness ( #10000 )
...
* Improved check_online() robustness
YOLOv5-wide improvement, not just in check_requirements()
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update general.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-01 15:55:18 +01:00
Glenn Jocher
c55e2cd73b
Add `min_items` filter option ( #9997 )
...
* Add `min_items` filter option
@AyushExel @Laughing-q dataset filter
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update dataloaders.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* fix
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-01 14:53:14 +01:00
Glenn Jocher
cf99788823
Check online twice on AutoUpdate ( #9999 )
...
Increased robustness to network failures
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-01 14:27:36 +01:00
Glenn Jocher
49156eb1d1
Fix `is_colab()` ( #9994 )
...
@AyushExel @kalenmike
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-01 12:02:05 +01:00
Paul Guerrie
a83d2a5013
Segmentation Tutorial ( #9521 )
...
* Added a tutorial notebook for segmentation.
* Updated header for segmentation tutorial and included other YOLOv5 sponsor sections.
* Updated segmentation tutorial to match main object detection tutorial.
* Update tutorial.ipynb
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-31 15:25:11 +01:00
Glenn Jocher
e704970f7f
Update README.md ( #9970 )
...
* Update README.md
@taliabender updated spacing per our convo
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update README.md
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-10-31 12:43:17 +01:00
Glenn Jocher
a625f29967
Full-size proto code (optional) ( #9980 )
...
* Update tf.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update tf.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update tf.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-10-30 22:09:36 +01:00
Glenn Jocher
6e544d5f7c
FROM nvcr.io/nvidia/pytorch:22.10-py3 ( #9966 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-29 13:31:01 +02:00
Kalen Michael
575055ce70
Switch from suffix checks to archive checks ( #9963 )
...
* fix: switched from suffix checks to archive checks
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Cleanup
seems like both functions accept Path type input so removing str()
* import always
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-10-28 21:16:03 +02:00
Glenn Jocher
6db0fac66c
Update README.md ( #9961 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-28 18:25:33 +02:00
Glenn Jocher
be9ef3871e
Update README.md ( #9958 )
...
* Update README.md
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update README_cn.md
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update tutorial.ipynb
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-10-28 17:48:41 +02:00
Glenn Jocher
38e5aae9a2
Update README.md ( #9957 )
...
* Update README.md
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update README.md
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update README.md
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update README_cn.md
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-28 17:37:25 +02:00
Glenn Jocher
32a9218573
Update Comet Integrations table text ( #9937 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-26 23:51:40 +02:00
Glenn Jocher
f9bb984e81
Add ClearML minimum version requirement ( #9933 )
...
* Add ClearML minimum version requirement
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update README.md
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update README.md
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-26 14:06:05 +02:00
Glenn Jocher
a5d875adca
Add `gnupg` to Dockerfile-cpu ( #9932 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-26 13:42:52 +02:00
Glenn Jocher
8236d8818b
Created using Colaboratory
2022-10-25 18:13:48 +02:00
paradigm
54f49fa581
Add TFLite Metadata to TFLite and Edge TPU models ( #9903 )
...
* added embedded meta data to tflite models
* added try block for inference
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* refactored tfite meta data into separate function
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Creat tmp file in /tmp
* Update export.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update export.py
* Update export.py
* Update export.py
* Update export.py
* Update common.py
* Update export.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update common.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-10-25 17:53:22 +02:00
SSTato
fba61e5583
Windows Python 3.7 .isfile() fix ( #9879 )
...
* Update dataloaders.py
Signed-off-by: SSTato <1210546396@qq.com>
* Update general.py
Signed-off-by: SSTato <1210546396@qq.com>
* Update general.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update dataloaders.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update dataloaders.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update general.py
Signed-off-by: SSTato <1210546396@qq.com>
* Update ci-testing.yml
Signed-off-by: SSTato <1210546396@qq.com>
* Update ci-testing.yml
Signed-off-by: SSTato <1210546396@qq.com>
* Update ci-testing.yml
Signed-off-by: SSTato <1210546396@qq.com>
* Update general.py
Signed-off-by: SSTato <1210546396@qq.com>
* Update general.py
Signed-off-by: SSTato <1210546396@qq.com>
* Update dataloaders.py
Signed-off-by: SSTato <1210546396@qq.com>
Signed-off-by: SSTato <1210546396@qq.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-24 16:20:47 +02:00
Glenn Jocher
eef90572bf
ClearML Dockerfile fix ( #9876 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-20 20:16:58 +02:00
Glenn Jocher
3b1a9d22a4
Fix OpenVINO Usage example ( #9874 )
...
* Fix OpenVINO Usage example
* Fix OpenVINO Usage example
2022-10-20 19:54:07 +02:00
Glenn Jocher
6371de8879
Webcam show fix ( #9847 )
...
* Webcam show fix
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update common.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-10-18 22:26:53 +02:00
Glenn Jocher
c4710012d8
Update README.md ( #9846 )
...
@pderrenger
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-18 21:22:37 +02:00
Glenn Jocher
d0df6c8403
Update ZipFile to context manager ( #9843 )
...
* Update zipFile to context manager
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Cleanup
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Cleanup
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-10-18 16:35:04 +02:00
Glenn Jocher
010cd0db7d
Update HUBDatasetStats() usage examples ( #9842 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-18 15:25:21 +02:00
Glenn Jocher
f1482b0667
Remove ipython from `check_requirements` exclude list ( #9841 )
...
May resolve e3ff780676 (commitcomment-87136818)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-18 13:54:33 +02:00
Glenn Jocher
acff977af3
Logger Cleanup ( #9828 )
2022-10-17 15:30:42 +02:00
Glenn Jocher
e3ff780676
Allow PyTorch Hub results to display in notebooks ( #9825 )
...
* Allow PyTorch Hub results to display in notebooks
* fix CI
* fix CI
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* fix CI
* fix CI
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* fix CI
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* fix CI
* fix CI
* fix CI
* fix CI
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* fix CI
* fix CI
* fix CI
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-10-17 14:34:33 +02:00
Glenn Jocher
e42c89d4ef
`smart_optimizer()` revert to weight with decay ( #9817 )
...
If a parameter does not fall into any other category
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-16 20:51:32 +02:00
Glenn Jocher
df80e7c723
Argoverse HUB fix ( #9809 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-14 18:18:58 +02:00
Glenn Jocher
2a19d070d8
Fix xView dataloaders import ( #9807 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-14 12:28:52 +02:00
Glenn Jocher
15b75659dd
Update hubconf.py local repo Usage example ( #9803 )
...
* Update hubconf.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update hubconf.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-13 15:27:16 +02:00
Glenn Jocher
16f87bb38e
HUB VOC fix ( #9792 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-13 02:32:06 +02:00
Vladislav Veklenko
85ae985b6a
update mask2segments and saving results ( #9785 )
...
* update mask2segments and saving results
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update general.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update predict.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
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-10-13 01:44:01 +02:00
Glenn Jocher
7a69035eb8
Update Loggers ( #9760 )
...
* Update
* Update tutorial.ipynb
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update tutorial.ipynb
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update tutorial.ipynb
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update requirements.txt
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update
* Update README.md
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-12 12:53:12 +02:00
Glenn Jocher
2f1eb21ad6
Remove #YOLOVISION22 notice ( #9751 )
...
Update README.md
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-10 14:54:21 +02:00
Glenn Jocher
209be932de
Fix segment evolution keys ( #9742 )
...
* Update
* Cleanup
2022-10-09 23:51:29 +02:00
Paul Guerrie
5ef69ef3e6
Error in utils/segment/general `masks2segments()` ( #9724 )
...
When running segmentation predict on gpu, the conversion from tensor to numpy fails. Calling `.cpu()` solves this problem.
Signed-off-by: Paul Guerrie <97041392+paulguerrie@users.noreply.github.com>
Signed-off-by: Paul Guerrie <97041392+paulguerrie@users.noreply.github.com>
2022-10-06 22:55:15 +02:00
Glenn Jocher
7f097ddb6c
FROM nvcr.io/nvidia/pytorch:22.09-py3 ( #9711 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-05 22:29:46 +02:00
Glenn Jocher
e4398cf179
TensorRT `--dynamic` fix ( #9691 )
...
* Update export.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update export.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-04 16:32:19 +02:00