Glenn Jocher
|
d76d7af566
|
Simplify usage, remove YOLO.new() and YOLO.load()
|
2 years ago |
Ayush Chaurasia
|
2bc9a5c87e
|
HUB setup (#108)
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>
|
2 years ago |
|
Ayush Chaurasia
|
c6eb6720de
|
Add pred, export and val callbacks (#126)
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>
|
2 years ago |
|
Glenn Jocher
|
598f17a472
|
Add global `settings.yaml` in `USER_CONFIG_DIR` (#125)
|
2 years ago |
|
Glenn Jocher
|
df4fc14c10
|
Docstring additions (#122)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
Glenn Jocher
|
c9f3e469cb
|
Add CoreML iOS updates (#121)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
|
2 years ago |
|
Glenn Jocher
|
076d73cfaa
|
Create Exporter() Class (#117)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
Glenn Jocher
|
92dad1c1b5
|
Start export implementation (#110)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
Ayush Chaurasia
|
100e21ee80
|
fix multiple dirs in DDP mode (#113)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Laughing-q <1185102784@qq.com>
|
2 years ago |
|
Ayush Chaurasia
|
34829a6b29
|
Make YOLO a module (#111)
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>
|
2 years ago |
Laughing
|
8406b49b49
|
Integration of v8 segmentation (#107)
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>
|
2 years ago |
|
Ayush Chaurasia
|
384f0ef1c6
|
Model interface enhancement (#106)
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>
|
2 years ago |
|
Glenn Jocher
|
6f0ba81427
|
Add best.pt val and COCO pycocotools val (#98)
Co-authored-by: ayush chaurasia <ayush.chaurarsia@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
Ayush Chaurasia
|
a1808eeda4
|
logger updates (#97)
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>
|
2 years ago |
|
Glenn Jocher
|
48cffa176e
|
Shorten module paths with new 'nn' dir (#96)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
Ayush Chaurasia
|
4fb04be20b
|
DDP and new dataloader Fix (#95)
Co-authored-by: Laughing-q <1185102784@qq.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>
|
2 years ago |
|
Ayush Chaurasia
|
16e3c08883
|
Import YOLOv5 dataloader (#94)
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>
|
2 years ago |
|
Glenn Jocher
|
ae05d44877
|
Simplify augmentations (#93)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
Glenn Jocher
|
249dfbdc05
|
Fix ClearML Mosaic callback to 'on_train_epoch_end' (#92)
|
2 years ago |
|
Ayush Chaurasia
|
0298821467
|
ClearML: Mosaic Logging (#91)
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>
|
2 years ago |
|
Glenn Jocher
|
98815d560f
|
YOLOv5 updates (#90)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
Glenn Jocher
|
ebd3cfb2fd
|
YOLOv8 architecture updates from R&D branch (#88)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
Glenn Jocher
|
cb4f20f3cf
|
Add TensorBoard support (#87)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
Glenn Jocher
|
248d54ca03
|
Update metrics names (#85)
|
2 years ago |
|
Glenn Jocher
|
6432afc5f9
|
Rename `img_size` to `imgsz` (#86)
|
2 years ago |
|
Ayush Chaurasia
|
ae2443c210
|
Add flops, num_params, inference speed logging and best.pt logging (#84)
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>
|
2 years ago |
|
Ayush Chaurasia
|
40f7e0806e
|
Add v8 modules (#81)
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>
|
2 years ago |
|
Ayush Chaurasia
|
681cfc1c35
|
Make config overrides user friendly (#80)
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>
|
2 years ago |
|
Laughing
|
7690cae2fc
|
add a naive DDP for model interface (#78)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
|
2 years ago |
|
Ayush Chaurasia
|
eb5adf4e0b
|
Model enhancement (#75)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
Laughing
|
e629335f6d
|
Revert augment_hyps (#70)
|
2 years ago |
|
Laughing
|
d63ee112d4
|
General cleanup (#69)
Co-authored-by: ayush chaurasia <ayush.chaurarsia@gmail.com>
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>
|
2 years ago |
|
Ayush Chaurasia
|
e6737f1207
|
Predictor support (#65)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Laughing-q <1185102784@qq.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
|
2 years ago |
|
Laughing
|
fbeeb5d1e1
|
add resuming (#63)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
Ayush Chaurasia
|
7ec7cf3aef
|
Detection support (#60)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Laughing-q <1185102784@qq.com>
|
2 years ago |
|
Ayush Chaurasia
|
5a52e7663a
|
standalone val (#56)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
Laughing
|
3a241e4cea
|
update segment training (#57)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: ayush chaurasia <ayush.chaurarsia@gmail.com>
|
2 years ago |
|
Ayush Chaurasia
|
d0b0fe2592
|
CLI updates (#58)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
Ayush Chaurasia
|
c5f5b80c04
|
Deterministic training (#53)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
ayush chaurasia
|
793dde365d
|
Revert "update"
This reverts commit 486bc49414.
|
2 years ago |
|
ayush chaurasia
|
486bc49414
|
update
|
2 years ago |
|
Ayush Chaurasia
|
d7df1770fa
|
Add warmup and accumulation (#52)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
Ayush Chaurasia
|
298287298d
|
Add clearml logging (#51)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
Ayush Chaurasia
|
512a225ce8
|
Cli support (#50)
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>
|
2 years ago |
|
Ayush Chaurasia
|
4291b9c31c
|
Add EMA and model checkpointing (#49)
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>
|
2 years ago |
|
Glenn Jocher
|
27d6545117
|
General console printout updates (#48)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
Glenn Jocher
|
8530e3fae0
|
Logging fix from YOLOv5 (#47)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
Ayush Chaurasia
|
832ea56eb4
|
update model initialization design, supports custom data/num_classes (#44)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
Ayush Chaurasia
|
1f3aad86c1
|
new check_dataset functions (#43)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
|
Glenn Jocher
|
4c68b9dcf6
|
Add is_colab() and is_kaggle() (#41)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2 years ago |
