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

325 Commits

Author SHA1 Message Date
Glenn Jocher
ec4b6dd2a3
Update export format docstrings (#6151) 
* Update export documentation

* Cleanup

* Update export.py

* Update README.md

* Update README.md

* Update README.md

* Update README.md

* Update README.md

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

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

* Update README.md

* Update README.md

* Update README.md

* Update train.py

* Update train.py

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2022-01-02 16:09:45 -08:00
bilzard
e1dc894364
Enable AdamW optimizer (#6152) 2022-01-02 13:10:19 -08:00
Ayush Chaurasia
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
Glenn Jocher
dc54ed5763
--freeze fix (#6044) 
Fix for https://github.com/ultralytics/yolov5/issues/6038
 2021-12-20 18:24:07 +01:00
Felix You
361705d9be
Multi-layer capable --freeze argument (#6019) 
* support specfiy multiple frozen layers

* fix bug

* Cleanup Freeze section

* Cleanup argument

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-12-17 15:42:26 +01:00
Glenn Jocher
7c6bae0ae6
Remove NCOLS from tqdm (#5804) 
* Remove NCOLS from tqdm

* [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-11-26 13:37:28 +01:00
Ayush Chaurasia
f2ca30a407
W&B: refactor W&B tables (#5737) 
* update

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

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

* reformat

* Single-line argparser argument

* Update README.md

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

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

* Update README.md

* [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>
 2021-11-25 17:19:21 +01:00
Glenn Jocher
4ca4aec46f
Do not save hyp.yaml and opt.yaml on evolve (#5775) 
* Do not save hyp.yaml and opt.yaml on evolve

* Update general.py
 2021-11-24 23:25:51 +01:00
Glenn Jocher
47fac9ff73
Update train.py comment to 'Model attributes' (#5670) 2021-11-16 13:58:15 +01:00
Werner Duvaud
09d170381c
Default DataLoader shuffle=True for training (#5623) 
* Fix shuffle DataLoader argument

* Add shuffle argument

* Disable shuffle when rect

* Cleanup, add rect warning

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

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

* Cleanup2

* Cleanup3

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>
 2021-11-13 13:07:32 +01:00
Glenn Jocher
7473f0f95d
DDP WORLD_SIZE-safe dataloader workers (#5631) 
* WORLD_SIZE-safe workers

* Update with DDP comment
 2021-11-12 14:48:55 +01:00
Glenn Jocher
61c50199a2
Update train, val tqdm to fixed width (#5367) 
* Update tqdm for fixed width

* Update val.py

* Update val.py

* Try ncols= in train.py

* NCOLS

* NCOLS

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

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

* bar_format

* position 0 leave true

* exp0

* auto

* auto

* Cleanup

* Cleanup

* Cleanup

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2021-11-10 15:47:38 +01:00
Glenn Jocher
79bca2bf64
LOGGER consolidation (#5569) 
* Logger consolidation

* [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-11-08 16:32:15 +01:00
Glenn Jocher
e189fa15ea
intersect_dicts() in hubconf.py fix (#5542) 2021-11-06 15:41:17 +01:00
Glenn Jocher
bcc085d83f
Common is_coco logic betwen train.py and val.py (#5521) 2021-11-05 13:32:46 +01:00
Yonghye Kwon
99a45bad81
Write date in checkpoint file (#5514) 
* write date in checkpoint file

write date in checkpoint file

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

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

* isoformat

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>
 2021-11-05 11:48:10 +01:00
Jirka Borovec
0155548384
precommit: isort (#5493) 
* precommit: isort

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

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

* Update isort config

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

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

* Update name

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>
 2021-11-04 17:24:25 +01:00
Glenn Jocher
5866646cc8
Fix float zeros format (#5491) 
* Fix float zeros format

* 255 to integer
 2021-11-03 23:36:53 +01:00
Glenn Jocher
7b1f7aec46
Update get_loggers() (#4854) 
* Update `set_logging()`

* Update export.py

* pre-commit fixes

* Update LoadImages

* Update LoadStreams

* Update print_args

* Single LOGGER definition

* yolo.py fix

Co-authored-by: pre-commit <pre-commit@example.com>
 2021-11-01 18:22:13 +01:00
Glenn Jocher
a4fece8c14
DDP nl fix (#5332) 2021-10-25 16:03:39 +02:00
Glenn Jocher
ca19df5f7f
Add autobatch feature for best batch-size estimation (#5092) 
* Autobatch

* fix mem

* fix mem2

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update train.py

* print result

* Cleanup print result

* swap fix in call

* to 64

* use total

* fix

* fix

* fix

* fix

* fix

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Cleanup printing

* Update final printout

* Update autobatch.py

* Update autobatch.py

* Update autobatch.py
 2021-10-25 13:56:13 +02:00
Glenn Jocher
752d97df2c
Add on_fit_epoch_end callback (#5232) 
* Add `on_fit_epoch_end` callback

* Add results to train

* Update __init__.py
 2021-10-18 14:28:22 +02:00
Malte Lorbach
153873e9e4
Fix ROOT as relative path (#5129) 
* use os.path.relpath instead of relative_to

* use os.path.relpath instead of relative_to

* Remove os.path from val.py

* Remove os.path from train.py

* Update detect.py import to os

* Update export.py import to os

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-10-11 09:47:24 -07:00
Nan
4a6dfffdaa
Pass LOCAL_RANK to torch_distributed_zero_first() (#5114) 
Co-authored-by: qiningonline <qiningonline@gmail.com>
 2021-10-09 18:41:50 -07:00
Glenn Jocher
070af88108
Fix yaml.safe_load() ignore emoji errors (#5060) 2021-10-05 13:41:52 -07:00
Glenn Jocher
5afc9c25ef
Implement --save-period locally (#5047) 
This PR adds a new training argument `--save-period` to save training checkpoints every `x` epochs. To save training every 50 epochs for example:
```
python train.py --save-period 50  # saves epoch50.pt, epoch100.pt, epoch150.pt, ... etc.
```

This saved checkpoints in addition to existing last.pt and best.pt checkpoints and does not affect their behavior. Default value is -1, i.e. disabled.
 2021-10-04 18:48:46 -07:00
Zhiqiang Wang
1922ddeac0
Fix pylint: do not use bare 'except' (#5025) 
* Fix E722, do not use bare 'except'

* Remove used codes

* Add FileNotFoundError in LoadImagesAndLabels

* Remove AssertionError

* Ignore LoadImagesAndLabels

* Ignore downloads.py

* Ignore torch_utils.py

* Ignore train.py

* Ignore datasets.py

* Enable utils/download.py

* Fixing exception in thop

* Remove unused code

* Fixing exception in LoadImagesAndLabels

* Fixing exception in exif_size

* Fixing exception in parse_model

* Ignore exceptions in requests

* Revert the exception as suggested

* Revert the exception as suggested
 2021-10-03 17:54:40 -07:00
Glenn Jocher
89a6e9ffa7
Compute loss on final val (#5017) 2021-09-30 15:16:11 -07:00
Glenn Jocher
452e84703c
Update train.py (#5014) 2021-09-30 13:57:53 -07:00
Glenn Jocher
4e04cb0dc8
Allow YOLOv5 execution from arbitrary cwd (#4954) 
* Allow YOLOv5 execution from arbitrary `cwd`

* Fix str bugs
 2021-09-27 17:40:20 -07:00
Glenn Jocher
d856c48298
Validate best.pt on train end (#4889) 
* Validate best.pt on train end

* 0.7 iou for COCO only

* pass callbacks

* active model.float() if not half

* print Validating best.pt...

* add newline
 2021-09-26 11:54:53 -07:00
Glenn Jocher
2b3109eeb0
Fix DDP destruction LOGGER.info() (#4863) 2021-09-20 13:12:48 +02:00
Glenn Jocher
4c839eeb10
Simplify check_requirements() usage (#4855) 
* Simplify `check_requirements()` usage

* remove assert, print()
 2021-09-18 18:34:30 +02:00
Glenn Jocher
4d1a2ac87e
Update sys.path.append(str(ROOT)) (#4852) 
* Update `sys.path.append(str(ROOT))`

* Cleanup
 2021-09-18 15:02:08 +02:00
Glenn Jocher
3732f9ac8a
Refactor argparser printing to print_args() (#4850) 
* Refactor argparser printing to `print_args()`

* Cleanup
 2021-09-18 14:16:19 +02:00
Glenn Jocher
621b6d5ba8
Evolution --resume fix (#4802) 
Also disable `/weights` dir creation when evolving as no weights are saved and empty folder causes user expectations of weights.
 2021-09-15 13:13:33 +02:00
Glenn Jocher
c47be26f34
Replace path.absolute() with path.resolve() (#4763) 2021-09-11 22:46:33 +02:00
Glenn Jocher
c5360f6e70
Fix --data from_HUB.zip (#4732) 
@KalenMike
 2021-09-09 18:01:59 +02:00
Glenn Jocher
a2b3c71636
Add suffix checks (#4711) 
* Add suffix checks

* Cleanup

* Cleanup2

* Cleanup3
 2021-09-08 14:36:12 +02:00
Kalen Michael
2317f86ca4
Optimised Callback Class to Reduce Code and Fix Errors (#4688) 
* added callbacks

* added back callback to main

* added save_dir to callback output

* reduced code count

* updated callbacks

* added default callback class to main, added missing parameters to on_model_save

* Glenn updates

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-09-07 18:32:15 +02:00
Glenn Jocher
548745181a
EarlyStopper updates (#4679) 2021-09-05 19:09:53 +02:00
imyhxy
f64fab5825
Fixed 'meta' and 'hyp' may out of order when using evolve (#4657) 
* Fixed 'meta' and 'hyp' may out of order when using evolve

* Update gitignore
 2021-09-05 17:43:09 +02:00
Glenn Jocher
50a9828679
DDP torch.jit.trace() --sync-bn fix (#4615) 
* Remove assert

* debug0

* trace=not opt.sync

* sync to sync_bn fix

* Cleanup
 2021-08-30 18:35:07 +02:00
Glenn Jocher
de44376d1b
Create Annotator() class (#4591) 
* Add Annotator() class

* Download Arial

* 2x for loop

* Cleanup

* tuple 2 list

* max_size=1920

* bold logging results to

* tolist()

* im = annotator.im

* PIL save in detect.py

* Smart asarray in detect.py

* revert to cv2.imwrite

* Cleanup

* Return result asarray

* Add `Profile()` profiler

* CamelCase Timeout

* Resize after mosaic

* pillow>=8.0.0

* daemon imwrite

* Add cv2 support

* Remove plot_wh_methods and plot_one_box

* pil=False for hubconf.py annotations

* im.shape bug fix

* colorstr common.py

* join daemons

* Update t.daemon

* Removed daemon saving
 2021-08-29 16:46:13 +02:00
Glenn Jocher
d7aa3f153d
Remove image_weights DDP code (#4579) 
* Initial commit

* Update
 2021-08-28 19:17:21 +02:00
Glenn Jocher
93cc015748
Add EarlyStopping feature (#4576) 
* Add EarlyStopping feature

* Add comment

* Cleanup

* Cleanup2

* debug

* debug2

* debug3

* debug3

* debug4

* debug5

* debug6

* debug7

* debug8

* debug9

* debug10

* debug11

* debug12

* Cleanup

* Add TODO for known DDP issue
 2021-08-28 19:03:52 +02:00
Glenn Jocher
19d03a955c
Remove DDP process group timeout (#4422) 2021-08-15 18:32:41 +02:00
Glenn Jocher
24bea5e4b7
Standardize headers and docstrings (#4417) 
* Implement new headers

* Reformat 1

* Reformat 2

* Reformat 3 - math

* Reformat 4 - yaml
 2021-08-14 21:17:51 +02:00
Glenn Jocher
ce7deec440
int(mlc) (#4385) 2021-08-11 17:32:13 +02:00
Glenn Jocher
86c7150cfd
Update newline (#4308) 2021-08-04 17:41:38 +02:00