Glenn Jocher
|
8acb5734c7
|
Created using Colaboratory
|
3 лет назад |
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
|
3 лет назад |
Glenn Jocher
|
3764277f95
|
Created using Colaboratory
|
3 лет назад |
Glenn Jocher
|
6e4358f3f3
|
Add `export.py` ONNX inference suggestion (#4146)
|
3 лет назад |
Glenn Jocher
|
d17b45eaad
|
Update README.md (#4143)
|
3 лет назад |
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
|
3 лет назад |
Glenn Jocher
|
63dd65e7ed
|
Update train.py (#4136)
* Refactor train.py
* Update imports
* Update imports
* Update optimizer
* cleanup
|
3 лет назад |
Glenn Jocher
|
264be1a616
|
Rename `opset_version` to `opset` (#4135)
|
3 лет назад |
Glenn Jocher
|
2e538443b7
|
ONNX inference update (#4073)
|
3 лет назад |
Glenn Jocher
|
39ef6c7a80
|
Update README.md (#4134)
|
3 лет назад |
imyhxy
|
4495e00016
|
Fix indentation in `log_training_progress()` (#4126)
|
3 лет назад |
Glenn Jocher
|
4bad914761
|
Update resume.py (#4115)
|
3 лет назад |
Glenn Jocher
|
2c073cd207
|
Add train.py ``--img-size` floor (#4099)
|
3 лет назад |
Glenn Jocher
|
3bef77f5cb
|
Addition refactor `export.py` (#4089)
* Addition refactor `export.py`
* Update export.py
|
3 лет назад |
Glenn Jocher
|
442a7abdf2
|
Refactor `export.py` (#4080)
* Refactor `export.py`
* cleanup
* Update check_requirements()
* Update export.py
|
3 лет назад |
Ayush Chaurasia
|
0cc7c58787
|
W&B: fix refactor bugs (#4069)
|
3 лет назад |
Glenn Jocher
|
7fdcc77bf4
|
Albumentations >= 1.0.3 (#4068)
|
3 лет назад |
Glenn Jocher
|
ee76a68f1d
|
Created using Colaboratory
|
3 лет назад |
Glenn Jocher
|
c8a98cb7cb
|
Missing `nc` and `names` handling in check_dataset() (#4066)
|
3 лет назад |
Glenn Jocher
|
b1be685005
|
Module `super().__init__()` (#4065)
* Module `super().__init__()`
* remove NMS
|
3 лет назад |
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>
|
3 лет назад |
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
|
3 лет назад |
Glenn Jocher
|
dd62e2d05c
|
Update README.md (#4041)
* Update README.md
* Update README.md
* Update README.md
|
3 лет назад |
Glenn Jocher
|
0067d9578a
|
Update greetings.yml (#4037)
|
3 лет назад |
Glenn Jocher
|
951922c735
|
Add `--sync-bn` known issue (#4032)
* Add `--sync-bn` known issue
* Update train.py
|
3 лет назад |
Glenn Jocher
|
b7e985e397
|
Update greetings.yml (#4024)
* Update greetings.yml
* Update greetings.yml
|
3 лет назад |
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>
|
3 лет назад |
Glenn Jocher
|
720aaa65c8
|
Rename `test.py` to `val.py` (#4000)
|
3 лет назад |
Glenn Jocher
|
8ee9fd1505
|
Update README.md (#3996)
|
3 лет назад |
Glenn Jocher
|
d204a61834
|
Alert (no detections) (#3984)
* `Detections()` class `print()` overload
* Update common.py
|
3 лет назад |
Glenn Jocher
|
b3dabdcc38
|
Update `probability` to `p` (#3980)
|
3 лет назад |
Glenn Jocher
|
41fdf9fa53
|
Created using Colaboratory
|
3 лет назад |
KEN
|
647223a7a8
|
`Ensemble()` visualize fix (#3973)
* fix visualize error
* Revert "fix visualize error"
* add visualise profile
|
3 лет назад |
Glenn Jocher
|
a544d59f52
|
Update requirements.txt `albumentations>=1.0.2` (#3972)
|
3 лет назад |
Glenn Jocher
|
90e60b403d
|
Update `np.random.random()` to `random.random()` (#3967)
|
3 лет назад |
Glenn Jocher
|
8298ce5e88
|
Update `albumentations>=1.0.2` (#3966)
|
3 лет назад |
Glenn Jocher
|
80299a57e2
|
Numerical stability fix for Albumentations (#3958)
|
3 лет назад |
Glenn Jocher
|
443af8b25a
|
Cache v0.4 update (#3954)
|
3 лет назад |
Glenn Jocher
|
a26e7de2bf
|
Update augmentations.py (#3948)
|
3 лет назад |
Glenn Jocher
|
248504cf13
|
Feature visualization improvements 32 (#3947)
|
3 лет назад |
Glenn Jocher
|
dabad5793a
|
Created using Colaboratory
|
3 лет назад |
Eldar Kurtic
|
e7888af94c
|
Fix inconsistent NMS IoU value for COCO (#3934)
Evaluation of 'best' and 'last' models will use the same params as the evaluation during the training phase.
This PR fixes https://github.com/ultralytics/yolov5/issues/3907
|
3 лет назад |
Glenn Jocher
|
8c6f9e15bf
|
Update `dataset_stats()` for zipped datasets (#3926)
* Update `dataset_stats()` for zipped datasets
@KalenMike
* cleanup
|
3 лет назад |
Glenn Jocher
|
850970e081
|
`feature_visualization()` CUDA fix (#3925)
|
3 лет назад |
jmiranda-laplateforme
|
588094eb7a
|
Update `setattr()` default for Hub PIL images (#3923)
Fix inference from PIL source.
|
3 лет назад |
Glenn Jocher
|
411842e058
|
Fix `torch.hub.list('ultralytics/yolov5')` pathlib bug (#3921)
|
3 лет назад |
Glenn Jocher
|
87b094bcbc
|
Feature visualization update (#3920)
* Feature visualization update
* Save to jpg (faster)
* Save to png
|
3 лет назад |
johnohagan
|
61047a2b4f
|
Save PyTorch Hub models to `/root/hub/cache/dir` (#3904)
* Create hubconf.py
* Add save_dir variable
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
|
3 лет назад |
Glenn Jocher
|
33202b7f0b
|
YOLOv5 + Albumentations integration (#3882)
* Albumentations integration
* ToGray p=0.01
* print confirmation
* create instance in dataloader init method
* improved version handling
* transform not defined fix
* assert string update
* create check_version()
* add spaces
* update class comment
|
3 лет назад |
Glenn Jocher
|
6a3ee7cf03
|
Hub models `map_location=device` (#3894)
* Hub models `map_location=device`
* cleanup
|
3 лет назад |