mirrors
/
yolov5
mirror of https://github.com/ultralytics/yolov5.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
1,351 Commits
10 Branches
10 Tags
584 MiB
Commit Graph

1351 Commits (b60b62e874e7cf0581c51936e39287c6906a419f)
 

Author SHA1 Message Date
Glenn Jocher b60b62e874
PyCharm reformat (#4209) 
* PyCharm reformat

* YAML reformat

* Markdown reformat
 2021-07-28 23:35:14 +02:00
Ayush Chaurasia 750465edae
W&B: More improvements and refactoring (#4205) 
* Improve docstrings and run names

* default wandb login prompt with timeout

* return key

* Update api_key check logic

* Properly support zipped dataset feature

* update docstring

* Revert tuorial change

* extend changes to log_dataset

* add run name

* bug fix

* bug fix

* Update comment

* fix import check

* remove unused import

* Hardcore .yaml file extension

* reduce code

* Reformat using pycharm

* Remove redundant try catch

* More refactoring and bug fixes

* retry

* Reformat using pycharm

* respect LOGGERS include list

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-07-28 23:25:15 +02:00
Glenn Jocher e016b15555
Update yolov5-bifpn.yaml (#4208) 2021-07-28 21:25:20 +02:00
Ayush Chaurasia e88e8f7a98
W&B: Restructure code to support the new dataset_check() feature (#4197) 
* Improve docstrings and run names

* default wandb login prompt with timeout

* return key

* Update api_key check logic

* Properly support zipped dataset feature

* update docstring

* Revert tuorial change

* extend changes to log_dataset

* add run name

* bug fix

* bug fix

* Update comment

* fix import check

* remove unused import

* Hardcore .yaml file extension

* reduce code

* Reformat using pycharm

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-07-28 17:40:08 +02:00
Glenn Jocher 2683b18079
Update Hub Path inputs (#4200) 2021-07-28 16:55:39 +02:00
Glenn Jocher 1f31b7c503
Create yolov5-bifpn.yaml (#4195) 2021-07-28 15:04:30 +02:00
Glenn Jocher 5d66e48723
Train from `--data path/to/dataset.zip` feature (#4185) 
* Train from `--data path/to/dataset.zip` feature

* Update dataset_stats()

* cleanup

* cleanup2
 2021-07-28 02:04:10 +02:00
Glenn Jocher 3fef11706c
Update comments header (#4184) 2021-07-27 23:23:41 +02:00
Ayush Chaurasia 63a1971e31
Improve docstrings and run names (#4174) 2021-07-27 15:13:32 +02:00
Glenn Jocher 0ad6301c96
Update script headers (#4163) 
* Update download script headers

* cleanup

* bug fix attempt

* bug fix attempt2

* bug fix attempt3

* cleanup
 2021-07-26 15:23:33 +02:00
Glenn Jocher f8e11483df
Update dataset headers (#4162) 2021-07-26 14:23:43 +02:00
Glenn Jocher 8acb5734c7 Created using Colaboratory 2021-07-26 00:57:09 +02:00
Glenn Jocher 96e36a7c91
New CSV Logger (#4148) 
* New CSV Logger

* cleanup

* move batch plots into Logger

* rename comment

* Remove total loss from progress bar

* mloss :-1 bug fix

* Update plot_results()

* Update plot_results()

* plot_results bug fix
 2021-07-25 19:06:37 +02:00
Glenn Jocher 3764277f95 Created using Colaboratory 2021-07-25 18:59:18 +02:00
Glenn Jocher 6e4358f3f3
Add `export.py` ONNX inference suggestion (#4146) 2021-07-25 13:46:16 +02:00
Glenn Jocher d17b45eaad
Update README.md (#4143) 2021-07-25 01:22:00 +02:00
Glenn Jocher efe60b5681
Refactor train.py and val.py `loggers` (#4137) 
* Update loggers

* Config

* Update val.py

* cleanup

* fix1

* fix2

* fix3 and reformat

* format sweep.py

* Logger() class

* cleanup

* cleanup2

* wandb package import fix

* wandb package import fix2

* txt fix

* fix4

* fix5

* fix6

* drop wandb into utils/loggers

* fix 7

* rename loggers/wandb_logging to loggers/wandb

* Update message

* Update message

* Update message

* cleanup

* Fix x axis bug

* fix rank 0 issue

* cleanup
 2021-07-25 01:18:39 +02:00
Glenn Jocher 63dd65e7ed
Update train.py (#4136) 
* Refactor train.py

* Update imports

* Update imports

* Update optimizer

* cleanup
 2021-07-24 16:11:39 +02:00
Glenn Jocher 264be1a616
Rename `opset_version` to `opset` (#4135) 2021-07-24 13:19:12 +02:00
Glenn Jocher 2e538443b7
ONNX inference update (#4073) 2021-07-24 13:08:51 +02:00
Glenn Jocher 39ef6c7a80
Update README.md (#4134) 2021-07-24 12:36:07 +02:00
imyhxy 4495e00016
Fix indentation in `log_training_progress()` (#4126) 2021-07-23 14:55:00 +02:00
Glenn Jocher 4bad914761
Update resume.py (#4115) 2021-07-22 17:22:11 +02:00
Glenn Jocher 2c073cd207
Add train.py ``--img-size` floor (#4099) 2021-07-21 16:50:47 +02:00
Glenn Jocher 3bef77f5cb
Addition refactor `export.py` (#4089) 
* Addition refactor `export.py`

* Update export.py
 2021-07-20 18:42:27 +02:00
Glenn Jocher 442a7abdf2
Refactor `export.py` (#4080) 
* Refactor `export.py`

* cleanup

* Update check_requirements()

* Update export.py
 2021-07-20 13:21:52 +02:00
Ayush Chaurasia 0cc7c58787
W&B: fix refactor bugs (#4069) 2021-07-19 13:57:13 +02:00
Glenn Jocher 7fdcc77bf4
Albumentations >= 1.0.3 (#4068) 2021-07-19 13:23:19 +02:00
Glenn Jocher ee76a68f1d Created using Colaboratory 2021-07-19 13:14:09 +02:00
Glenn Jocher c8a98cb7cb
Missing `nc` and `names` handling in check_dataset() (#4066) 2021-07-19 13:10:21 +02:00
Glenn Jocher b1be685005
Module `super().__init__()` (#4065) 
* Module `super().__init__()`

* remove NMS
 2021-07-19 12:41:15 +02:00
Glenn Jocher f7d8562060
`val.py` refactor (#4053) 
* val.py refactor

* cleanup

* cleanup

* cleanup

* cleanup

* save after eval

* opt.imgsz bug fix

* wandb refactor

* dataloader to train_loader

* capitalize global variables

* runs/hub/exp to runs/detect/exp

* refactor wandb logging

* Refactor wandb operations (#4061)

Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
 2021-07-19 10:43:01 +02:00
Glenn Jocher 9dd33fd20f
AutoShape PosixPath support (#4047) 
* AutoShape PosixPath support

Usage example:

````python
from pathlib import Path

model = ...
file = Path('data/images/zidane.jpg')

results = model(file)
```

* Update common.py
 2021-07-18 15:25:37 +02:00
Glenn Jocher dd62e2d05c
Update README.md (#4041) 
* Update README.md

* Update README.md

* Update README.md
 2021-07-17 23:12:27 +02:00
Glenn Jocher 0067d9578a
Update greetings.yml (#4037) 2021-07-17 15:47:05 +02:00
Glenn Jocher 951922c735
Add `--sync-bn` known issue (#4032) 
* Add `--sync-bn` known issue

* Update train.py
 2021-07-17 13:07:19 +02:00
Glenn Jocher b7e985e397
Update greetings.yml (#4024) 
* Update greetings.yml

* Update greetings.yml
 2021-07-16 14:38:59 +02:00
Ayush Chaurasia 62409eea08
W&B sweeps support (#3938) 
* Add support for W&B Sweeps

* Update and reformat

* Update search space

* reformat

* reformat sweep.py

* Update sweep.py

* Move sweeps files to wandb dir

* Remove print

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-07-14 19:13:02 +02:00
Glenn Jocher 720aaa65c8
Rename `test.py` to `val.py` (#4000) 2021-07-14 15:43:54 +02:00
Glenn Jocher 8ee9fd1505
Update README.md (#3996) 2021-07-13 23:07:09 +02:00
Glenn Jocher d204a61834
Alert (no detections) (#3984) 
* `Detections()` class `print()` overload

* Update common.py
 2021-07-12 19:48:47 +02:00
Glenn Jocher b3dabdcc38
Update `probability` to `p` (#3980) 2021-07-12 15:54:43 +02:00
Glenn Jocher 41fdf9fa53 Created using Colaboratory 2021-07-12 12:43:26 +02:00
KEN 647223a7a8
`Ensemble()` visualize fix (#3973) 
* fix visualize error

* Revert "fix visualize error"

* add visualise profile
 2021-07-11 19:47:08 +02:00
Glenn Jocher a544d59f52
Update requirements.txt `albumentations>=1.0.2` (#3972) 2021-07-11 19:07:42 +02:00
Glenn Jocher 90e60b403d
Update `np.random.random()` to `random.random()` (#3967) 2021-07-11 16:56:36 +02:00
Glenn Jocher 8298ce5e88
Update `albumentations>=1.0.2` (#3966) 2021-07-11 15:49:30 +02:00
Glenn Jocher 80299a57e2
Numerical stability fix for Albumentations (#3958) 2021-07-10 19:50:53 +02:00
Glenn Jocher 443af8b25a
Cache v0.4 update (#3954) 2021-07-10 14:18:46 +02:00
Glenn Jocher a26e7de2bf
Update augmentations.py (#3948) 2021-07-09 16:45:04 +02:00