Glenn Jocher
7a565f130a
Update `dataset_stats()` ( #3593 )
@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)
```
3 years ago
Glenn Jocher
4984cf54be
train.py GPU memory fix ( #3590 )
* train.py GPU memory fix
* ema
* cuda
* cuda
* zeros input
* to device
* batch index 0
3 years ago
Glenn Jocher
e8c5237403
ConfusionMatrix `normalize=True` fix ( #3587 )
3 years ago
Glenn Jocher
ec2da4a82c
Add ConfusionMatrix `normalize=True` flag ( #3586 )
3 years ago
Glenn Jocher
46e1fdfbc6
Update stale.yml ( #3585 )
3 years ago
Glenn Jocher
5c32bd3080
Created using Colaboratory
3 years ago
Glenn Jocher
53ed872c28
Update export.py, yolo.py `sys.path.append()` ( #3579 )
3 years ago
Glenn Jocher
095197bd4a
Ignore Seaborn plot warnings ( #3576 )
* Ignore Seaborn plot warnings
* Update plots.py
* Update metrics.py
3 years ago
Glenn Jocher
4695ca8314
Refactoring cleanup ( #3565 )
* Refactoring cleanup
* Update test.py
3 years ago
Glenn Jocher
0e5cfdbea7
Refactor models/export.py arguments ( #3564 )
* Refactor models/export.py arguments
* cleanup
* cleanup
3 years ago
Glenn Jocher
66cf5c28c1
Refactor detect.py arguments ( #3559 )
* Refactor detect.py arguments
@SkalskiP @KalenMike
* unused ok
* comment arguments
3 years ago
Glenn Jocher
a9553c04a7
Refactor test.py arguments ( #3558 )
* remove opt from test()
* pass kwargs
* update comments
* revert accidental default change
* multiple --img options
* add comments
3 years ago
dependabot[bot]
c6deb73a89
Bump pip from 18.1 to 19.2 in /utils/google_app_engine ( #3561 )
Bumps [pip](https://github.com/pypa/pip ) from 18.1 to 19.2.
- [Release notes](https://github.com/pypa/pip/releases )
- [Changelog](https://github.com/pypa/pip/blob/main/NEWS.rst )
- [Commits](https://github.com/pypa/pip/compare/18.1...19.2 )
---
updated-dependencies:
- dependency-name: pip
dependency-type: direct:production
...
Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
3 years ago
Glenn Jocher
4ef9261870
Remove redundant speed/study `half` argument ( #3557 )
3 years ago
Glenn Jocher
5948f20a3d
Update test.py profiling ( #3555 )
* Update test.py profiling
* half_precision to half
* inplace
3 years ago
Glenn Jocher
8b5086c21b
Multi-GPU default to single device 0 ( #3554 )
* Multi-GPU default to single device 0
* Multi-GPU default to single device 0
* add space
3 years ago
Glenn Jocher
63157d214d
Remove `is_coco` argument from `test()` ( #3553 )
3 years ago
Glenn Jocher
958ab92dc1
Remove `opt` from `create_dataloader()`` ( #3552 )
3 years ago
Glenn Jocher
0cfc5b2c18
Update README.md ( #3550 )
Add permanent splash URL and update hyperlink from iOS landing page to Ultralytics YOLOv5 landing page at https://ultralytics.com/yolov5
3 years ago
Glenn Jocher
f8ec71e1c2
Update `check_python(minimum=3.6.2)` ( #3548 )
3 years ago
Glenn Jocher
ef0b5c9d29
On-demand `pycocotools` pip install ( #3547 )
3 years ago
Glenn Jocher
1b5edb6f8e
Update `dataset_stats()` for HUB ( #3536 )
* Update `dataset_stats()` for HUB
Cleanup of b6fdd2e
* autodownload flag
* Update general.py
* cleanup
3 years ago
Glenn Jocher
b6fdd2e5e5
Create `dataset_stats()` for HUB
3 years ago
Glenn Jocher
ac8691e208
Created using Colaboratory
3 years ago
Glenn Jocher
78cf488556
Created using Colaboratory
3 years ago
Glenn Jocher
c6b51f4189
Update FP16 `--half` argument for test.py and detect.py ( #3532 )
* Update FP16 `--half` argument for test.py and detect.py
* Update detect.py
3 years ago
Glenn Jocher
8d52c1c5c5
Update datasets.py ( #3531 )
Minor updates to https://github.com/ultralytics/yolov5/pull/3505 , inplace accumulation.
3 years ago
Glenn Jocher
ed651873ff
Merge branch 'develop'
3 years ago
Dean Mark
28bff22df8
Use multi-threading in cache_labels ( #3505 )
* Use multi threading in cache_labels
* PEP8 reformat
* Add num_threads
* changed ThreadPool.imap_unordered to Pool.imap_unordered
* Remove inplace additions
* Update datasets.py
refactor initial desc
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Glenn Jocher
f3c3d2ce5d
Merge `develop` branch into `master` ( #3518 )
* update ci-testing.yml (#3322 )
* update ci-testing.yml
* update greetings.yml
* bring back os matrix
* update ci-testing.yml (#3322 )
* update ci-testing.yml
* update greetings.yml
* bring back os matrix
* Enable direct `--weights URL` definition (#3373 )
* Enable direct `--weights URL` definition
@KalenMike this PR will enable direct --weights URL definition. Example use case:
```
python train.py --weights https://storage.googleapis.com/bucket/dir/model.pt
```
* cleanup
* bug fixes
* weights = attempt_download(weights)
* Update experimental.py
* Update hubconf.py
* return bug fix
* comment mirror
* min_bytes
* Update tutorial.ipynb (#3368 )
add Open in Kaggle badge
* `cv2.imread(img, -1)` for IMREAD_UNCHANGED (#3379 )
* Update datasets.py
* comment
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* COCO evolution fix (#3388 )
* COCO evolution fix
* cleanup
* update print
* print fix
* Create `is_pip()` function (#3391 )
Returns `True` if file is part of pip package. Useful for contextual behavior modification.
```python
def is_pip():
# Is file in a pip package?
return 'site-packages' in Path(__file__).absolute().parts
```
* Revert "`cv2.imread(img, -1)` for IMREAD_UNCHANGED (#3379 )" (#3395 )
This reverts commit 21a9607e00
.
* Update FLOPs description (#3422 )
* Update README.md
* Changing FLOPS to FLOPs.
Co-authored-by: BuildTools <unconfigured@null.spigotmc.org>
* Parse URL authentication (#3424 )
* Parse URL authentication
* urllib.parse.unquote()
* improved error handling
* improved error handling
* remove %3F
* update check_file()
* Add FLOPs title to table (#3453 )
* Suppress jit trace warning + graph once (#3454 )
* Suppress jit trace warning + graph once
Suppress harmless jit trace warning on TensorBoard add_graph call. Also fix multiple add_graph() calls bug, now only on batch 0.
* Update train.py
* Update MixUp augmentation `alpha=beta=32.0` (#3455 )
Per VOC empirical results https://github.com/ultralytics/yolov5/issues/3380#issuecomment-853001307 by @developer0hye
* Add `timeout()` class (#3460 )
* Add `timeout()` class
* rearrange order
* Faster HSV augmentation (#3462 )
remove datatype conversion process that can be skipped
* Add `check_git_status()` 5 second timeout (#3464 )
* Add check_git_status() 5 second timeout
This should prevent the SSH Git bug that we were discussing @KalenMike
* cleanup
* replace timeout with check_output built-in timeout
* Improved `check_requirements()` offline-handling (#3466 )
Improve robustness of `check_requirements()` function to offline environments (do not attempt pip installs when offline).
* Add `output_names` argument for ONNX export with dynamic axes (#3456 )
* Add output names & dynamic axes for onnx export
Add output_names and dynamic_axes names for all outputs in torch.onnx.export. The first four outputs of the model will have names output0, output1, output2, output3
* use first output only + cleanup
Co-authored-by: Samridha Shrestha <samridha.shrestha@g42.ai>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Revert FP16 `test.py` and `detect.py` inference to FP32 default (#3423 )
* fixed inference bug ,while use half precision
* replace --use-half with --half
* replace space and PEP8 in detect.py
* PEP8 detect.py
* update --half help comment
* Update test.py
* revert space
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Add additional links/resources to stale.yml message (#3467 )
* Update stale.yml
* cleanup
* Update stale.yml
* reformat
* Update stale.yml HUB URL (#3468 )
* Stale `github.actor` bug fix (#3483 )
* Explicit `model.eval()` call `if opt.train=False` (#3475 )
* call model.eval() when opt.train is False
call model.eval() when opt.train is False
* single-line if statement
* cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* check_requirements() exclude `opencv-python` (#3495 )
Fix for 3rd party or contrib versions of installed OpenCV as in https://github.com/ultralytics/yolov5/issues/3494 .
* Earlier `assert` for cpu and half option (#3508 )
* early assert for cpu and half option
early assert for cpu and half option
* Modified comment
Modified comment
* Update tutorial.ipynb (#3510 )
* Reduce test.py results spacing (#3511 )
* Update README.md (#3512 )
* Update README.md
Minor modifications
* 850 width
* Update greetings.yml
revert greeting change as PRs will now merge to master.
Co-authored-by: Piotr Skalski <SkalskiP@users.noreply.github.com>
Co-authored-by: SkalskiP <piotr.skalski92@gmail.com>
Co-authored-by: Peretz Cohen <pizzaz93@users.noreply.github.com>
Co-authored-by: tudoulei <34886368+tudoulei@users.noreply.github.com>
Co-authored-by: chocosaj <chocosaj@users.noreply.github.com>
Co-authored-by: BuildTools <unconfigured@null.spigotmc.org>
Co-authored-by: Yonghye Kwon <developer.0hye@gmail.com>
Co-authored-by: Sam_S <SamSamhuns@users.noreply.github.com>
Co-authored-by: Samridha Shrestha <samridha.shrestha@g42.ai>
Co-authored-by: edificewang <609552430@qq.com>
3 years ago
Glenn Jocher
c058a61e3b
Update greetings.yml
revert greeting change as PRs will now merge to master.
3 years ago
Glenn Jocher
abb2a96e91
Update README.md ( #3512 )
* Update README.md
Minor modifications
* 850 width
3 years ago
Glenn Jocher
d986145b9a
Reduce test.py results spacing ( #3511 )
3 years ago
Glenn Jocher
eede7dc48c
Update tutorial.ipynb ( #3510 )
3 years ago
Yonghye Kwon
8d1ddc93c7
Earlier `assert` for cpu and half option ( #3508 )
* early assert for cpu and half option
early assert for cpu and half option
* Modified comment
Modified comment
3 years ago
Glenn Jocher
3f03acb3db
check_requirements() exclude `opencv-python` ( #3507 )
Duplicate of #3495 merged to `develop`. This PR will be merged to `master`.
Fixes https://github.com/ultralytics/yolov5/issues/3494 .
3 years ago
Glenn Jocher
90b7895d65
check_requirements() exclude `opencv-python` ( #3495 )
Fix for 3rd party or contrib versions of installed OpenCV as in https://github.com/ultralytics/yolov5/issues/3494 .
3 years ago
Yonghye Kwon
a1c3572bc9
Explicit `model.eval()` call `if opt.train=False` ( #3475 )
* call model.eval() when opt.train is False
call model.eval() when opt.train is False
* single-line if statement
* cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Glenn Jocher
cf4f95bc5f
Stale `github.actor` bug fix ( #3483 )
3 years ago
Glenn Jocher
3597d280ee
Update stale.yml HUB URL ( #3468 )
3 years ago
Glenn Jocher
739451da5a
Add additional links/resources to stale.yml message ( #3467 )
* Update stale.yml
* cleanup
* Update stale.yml
* reformat
3 years ago
edificewang
b31229ae89
Revert FP16 `test.py` and `detect.py` inference to FP32 default ( #3423 )
* fixed inference bug ,while use half precision
* replace --use-half with --half
* replace space and PEP8 in detect.py
* PEP8 detect.py
* update --half help comment
* Update test.py
* revert space
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Sam_S
044daafd9d
Add `output_names` argument for ONNX export with dynamic axes ( #3456 )
* Add output names & dynamic axes for onnx export
Add output_names and dynamic_axes names for all outputs in torch.onnx.export. The first four outputs of the model will have names output0, output1, output2, output3
* use first output only + cleanup
Co-authored-by: Samridha Shrestha <samridha.shrestha@g42.ai>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Glenn Jocher
317f2ccc9d
Improved `check_requirements()` offline-handling ( #3466 )
Improve robustness of `check_requirements()` function to offline environments (do not attempt pip installs when offline).
3 years ago
Glenn Jocher
563ea9475a
Add `check_git_status()` 5 second timeout ( #3464 )
* Add check_git_status() 5 second timeout
This should prevent the SSH Git bug that we were discussing @KalenMike
* cleanup
* replace timeout with check_output built-in timeout
3 years ago
Yonghye Kwon
c37f072ba7
Faster HSV augmentation ( #3462 )
remove datatype conversion process that can be skipped
3 years ago
Glenn Jocher
d40481acc5
Add `timeout()` class ( #3460 )
* Add `timeout()` class
* rearrange order
3 years ago
Glenn Jocher
8e3b4a0bf3
Update MixUp augmentation `alpha=beta=32.0` ( #3455 )
Per VOC empirical results https://github.com/ultralytics/yolov5/issues/3380#issuecomment-853001307 by @developer0hye
3 years ago
Glenn Jocher
4aa2959101
Suppress jit trace warning + graph once ( #3454 )
* Suppress jit trace warning + graph once
Suppress harmless jit trace warning on TensorBoard add_graph call. Also fix multiple add_graph() calls bug, now only on batch 0.
* Update train.py
3 years ago
Glenn Jocher
af2bc3a1c3
Add FLOPs title to table ( #3453 )
3 years ago