mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-02 22:29:06 +08:00
Code Issues Projects Releases Wiki Activity
2,141 Commits 10 Branches 10 Tags
Commit Graph

2141 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
Vardan Agarwal
fa569cdae5
Add support for different normalization layers (#7377) 
* Add support for different normalization layers.

* [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>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-04-11 14:04:22 +02:00
Glenn Jocher
bd2dda8e64
Update optimizer param group strategy (#7376) 
* Update optimizer param group strategy

Avoid empty lists on missing BathNorm2d models as in https://github.com/ultralytics/yolov5/issues/7375

* fix init
 2022-04-11 12:34:35 +02:00
dependabot[bot]
71685cbf91
Bump actions/stale from 4 to 5 (#7371) 
Bumps [actions/stale](https://github.com/actions/stale) from 4 to 5.
- [Release notes](https://github.com/actions/stale/releases)
- [Changelog](https://github.com/actions/stale/blob/main/CHANGELOG.md)
- [Commits](https://github.com/actions/stale/compare/v4...v5)

---
updated-dependencies:
- dependency-name: actions/stale
  dependency-type: direct:production
  update-type: version-update:semver-major
...

Signed-off-by: dependabot[bot] <support@github.com>

Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
 2022-04-11 10:26:13 +02:00
Glenn Jocher
8c420c4c1f
Update ci-testing.yml (#7365) 
Remove keras==2.6.0 patch
 2022-04-10 15:17:25 +02:00
Glenn Jocher
b8d4f2bf74
Replace Slack with Community Forum in issues (#7364) 2022-04-10 14:50:01 +02:00
Glenn Jocher
db36f13c7a
Delete FUNDING.yml (#7363) 
Deleting as redundant with FUNDING.yml present in organization repo at https://github.com/ultralytics/.github
 2022-04-10 14:40:33 +02:00
Glenn Jocher
1993efd59e
Swap unsafe_chunk() for chunk() (#7362) 
Eliminates all unsafe function in YOLOv5 out of an abundance of caution.
 2022-04-10 13:53:28 +02:00
Glenn Jocher
406ee528f0
Loss and IoU speed improvements (#7361) 
* Loss speed improvements

* bbox_iou speed improvements

* bbox_ioa speed improvements

* box_iou speed improvements

* box_iou speed improvements
 2022-04-10 13:46:07 +02:00
rglkt
aa542ce6a6
DetectMultiBackend() default stride=32 (#7342) 
* set common default stride as 32

* restore default stride, and set it on argument optional

* fix wrong use of opt

* fix missing parameter of stride

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

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

* fix format of parameters

* Update val.py

* 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-04-09 19:11:55 +02:00
Glenn Jocher
3bb233a7fb
Add ONNX export metadata (#7353) 2022-04-09 13:27:49 +02:00
Glenn Jocher
698a5d7f26
Add python benchmarks.py --test for export-only (#7350) 
* Test exports

* Fix precommit
 2022-04-09 01:32:16 +02:00
Glenn Jocher
446e6f563a
Rename 'MacOS' to 'macOS' (#7349) 2022-04-08 23:05:15 +02:00
Glenn Jocher
302b00b5f4
Update _make_grid() (#7346) 2022-04-08 12:55:16 +02:00
Glenn Jocher
5f941a84ef
Print dataset scan only if RANK in (-1, 0) (#7337) 
* Print dataset scan only `if RANK in (-1, 0)`

* [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-04-07 16:44:08 +02:00
Glenn Jocher
676e10cf1a
Simplify callbacks.py return (#7333) 
* Simplify callbacks.py return

* Indent args (pytorch convention)

* [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-04-07 16:15:01 +02:00
Glenn Jocher
5783de26fe
Objects365 dataset breakdown images vs zips (#7335) 2022-04-07 16:12:44 +02:00
Nrupatunga
b7faeda0f2
Fix Tf export for BottleneckCSP (#7330) 2022-04-07 13:52:44 +02:00
Glenn Jocher
0ca85ed65f
Update Objects365.yaml (#7323) 
Updated dataset size to 712GB (includes undeleted zips).
 2022-04-06 23:52:19 +02:00
Nick Martin
a88a81469a
Copy wandb param dict before training to avoid overwrites (#7317) 
* Copy wandb param dict before training to avoid overwrites.

Copy the hyperparameter dict retrieved from wandb configuration before passing it to `train()`. Training overwrites parameters in the dictionary (eg scaling obj/box/cls gains), which causes the values reported in wandb to not match the input values. This is confusing as it makes it hard to reproduce a run, and also throws off wandb's Bayesian sweep algorithm.

* Cleanup

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-04-06 18:35:33 +02:00
Glenn Jocher
245d6459a9
Add callbacks (#7315) 
* Add `on_train_start()` callback

* Update

* Update
 2022-04-06 17:23:34 +02:00
Glenn Jocher
32661f75ac
Add retry=3 to download() (#7313) 
* Add `retry=3` to `download()`

* Update general.py

* Update general.py

* Update general.py

* Update VOC.yaml

* Update VisDrone.yaml
 2022-04-06 13:12:41 +02:00
Glenn Jocher
f735458987
Use tqdm.auto (#7311) 2022-04-06 12:20:24 +02:00
Glenn Jocher
d257c75c84
Update export.py (#7301) 
* Update export.py

Simplify code.

* [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-04-05 21:21:57 +02:00
Max Strobel
741fac815e
fix: disable usage of root logger (#7296) 
* fix: disable usage of root logger

`logging.basicConfig` configures Python's root logger. This prohibits
fine control of logging, overwrites logging configuration done outside
the package, and is not best practice. Instead, the used logger is now
configured directly, and the root logger is untouched.

Example:
    If yolov5 is used as part of another project with some sophisticated
    logging, the internal `logging.basicConfig` call overwrites all the
    external configuration.

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

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

* Update general.py

* Update general.py

* Comment kaggle

* Uncomment kaggle

Co-authored-by: Maximilian Strobel <Maximilian.Strobel@infineon.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-04-05 17:38:13 +02:00
Glenn Jocher
c759bbdf19
Add check_requirements(('pycocotools>=2.0',)) (#7295) 
Add  `check_requirements(('pycocotools>=2.0',))`
 2022-04-05 15:55:16 +02:00
Glenn Jocher
b1300f3e0b
Add dataset sizes (zipped) (#7293) 2022-04-05 15:14:54 +02:00
Glenn Jocher
d2e7ba2a3a
val.py --weights and --data compatibility check (#7292) 
Improved error messages for understanding of user error with val.py. May help https://github.com/ultralytics/yolov5/issues/7291
 2022-04-05 14:23:15 +02:00
Glenn Jocher
5f97001ed4
Context manager open(file) as f fixes (#7289) 
* Flask context manager `open()` fix

* Additional read context manager fixes
 2022-04-05 12:54:25 +02:00
Glenn Jocher
2181ef371e
Update cv2.imread() patch with flags argument (#7287) 2022-04-05 11:49:32 +02:00
leeflix
8d0291f3af
Enable TensorFlow ops for --nms and --agnostic-nms (#7281) 
* enable TensorFlow ops if flag --nms or --agnostic-nms is used

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

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

* Update export.py

* Update export.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>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-04-05 11:33:08 +02:00
Glenn Jocher
2da68664b5
Update Dockerfile (#7282) 2022-04-05 00:52:37 +02:00
pre-commit-ci[bot]
7882950577
[pre-commit.ci] pre-commit suggestions (#7279) 
* [pre-commit.ci] pre-commit suggestions

updates:
- [github.com/asottile/pyupgrade: v2.31.0 → v2.31.1](https://github.com/asottile/pyupgrade/compare/v2.31.0...v2.31.1)
- [github.com/pre-commit/mirrors-yapf: v0.31.0 → v0.32.0](https://github.com/pre-commit/mirrors-yapf/compare/v0.31.0...v0.32.0)

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

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

* Update yolo.py

* Update activations.py

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

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

* Update activations.py

* Update tf.py

* Update tf.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>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2022-04-04 22:47:00 +02:00
Glenn Jocher
ea72b84f5e
Integrate offset into grid (#7262) 
Eliminate 1 op during training and inference.
 2022-04-03 23:40:23 +02:00
Glenn Jocher
8bc839ed8e
TorchScript single-output fix (#7261) 2022-04-03 22:51:11 +02:00
Glenn Jocher
05cf0d1a44
Export single output only (#7259) 
* Update

* Update

* [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-04-03 21:29:20 +02:00
Glenn Jocher
779efbb9ca Update 2022-04-03 21:21:55 +02:00
Glenn Jocher
ad0e4d5d19
torch.split() replace slicing on out-of-place inference (#7258) 2022-04-03 20:05:50 +02:00
Glenn Jocher
4f839b7970
Refactor out-of-place Detect() for reduced ops (#7257) 2022-04-03 19:26:23 +02:00
Glenn Jocher
ffcbd8ca97
Export with official nn.SiLU() (#7256) 
* Update

* Update time_limit
 2022-04-03 18:45:05 +02:00
Zengyf-CVer
dda669a12c
Fix Flask REST API (#7210) 
* Update restapi.py

* Update restapi.py

* [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

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-04-03 13:19:26 +02:00
Glenn Jocher
035b5548e4
Update tutorial.ipynb (#7255) 2022-04-03 12:18:24 +02:00
Glenn Jocher
3d3483cf0c
Update tutorial.ipynb (#7254) 2022-04-03 12:14:12 +02:00
Glenn Jocher
6f4eb95af7
Update setup.cfg to description_file field (#7248) 
Resolve `UserWarning: Usage of dash-separated 'description-file' will not be supported in future versions. Please use the underscore name 'description_file' instead`
 2022-04-03 00:18:18 +02:00
Glenn Jocher
a19406b39d
Update minimum Python>=3.7.0 (#7247) 2022-04-02 15:05:00 +02:00
Glenn Jocher
37675e110f
Fix www.youtube.com hostname (#7242) 
* Fix `www.youtube.com` hostname

* Update datasets.py
 2022-04-01 21:38:49 +02:00
Glenn Jocher
71621df875
Create CODE_OF_CONDUCT.md (#7233) 2022-04-01 00:24:37 +02:00
Glenn Jocher
734ab033fd
SavedModel TF Serve Fix (#7228) 
* SavedModel TF Serve Fix

Fix for https://github.com/ultralytics/yolov5/issues/7205 proposed by @tylertroy

* Update export.py
 2022-04-01 00:07:23 +02:00
Glenn Jocher
4d157f578a
Update .pre-commit-config.yaml (#7230) 2022-03-31 17:26:34 +02:00
Glenn Jocher
2c3221844b
CLI fire prep updates (#7229) 
* CLI fire prep updates

* revert unintentional TF export change
 2022-03-31 17:11:43 +02:00
Jirka Borovec
c3d5ac151e
precommit: yapf (#5494) 
* precommit: yapf

* align isort

* fix

# Conflicts:
#	utils/plots.py

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

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

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

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

* Update setup.cfg

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

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

* Update setup.cfg

* Update setup.cfg

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

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

* Update wandb_utils.py

* Update augmentations.py

* Update setup.cfg

* Update yolo.py

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

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

* Update val.py

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

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

* simplify colorstr

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

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

* val run fix

* export.py last comma

* Update export.py

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

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

* Update hubconf.py

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

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

* PyTorch Hub tuple fix

* PyTorch Hub tuple fix2

* PyTorch Hub tuple fix3

* Update setup

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-03-31 16:52:34 +02:00