1088 Commits

Author SHA1 Message Date
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>
2022-12-28 19:38:37 +01:00
Ayush Chaurasia
0303ced8ab
Model enhancements 2 (#109)
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-12-28 21:01:51 +05:30
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>
2022-12-28 16:01:38 +01:00
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>
2022-12-28 13:35:01 +01:00
Glenn Jocher
38d6df55cb
Fix save_json(predn, batch) (#105)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-27 18:42:36 +01:00
Ayush Chaurasia
8b6466f731
Move tests outside package (#104) 2022-12-27 22:16:07 +05:30
Ayush Chaurasia
8028e2b1b8
Make optimizer static method inside trainer (#103)
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-12-27 17:02:03 +01:00
Laughing
5c6d11bdb2
Allocated updated pycocotools metrics fix (#101)
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-12-27 15:49:13 +01:00
Glenn Jocher
441126a77e
Add l/x models (#100) 2022-12-27 05:06:15 +01:00
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>
2022-12-27 04:56:24 +01:00
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>
2022-12-26 18:12:17 +01:00
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>
2022-12-26 17:34:28 +01:00
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>
2022-12-26 15:35:49 +01:00
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>
2022-12-26 01:30:01 +01:00
Glenn Jocher
ae05d44877
Simplify augmentations (#93)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-25 21:21:26 +01:00
Glenn Jocher
249dfbdc05
Fix ClearML Mosaic callback to 'on_train_epoch_end' (#92) 2022-12-25 17:04:24 +01:00
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>
2022-12-25 14:36:41 +01:00
Glenn Jocher
98815d560f
YOLOv5 updates (#90)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-25 14:33:18 +01:00
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>
2022-12-24 18:10:44 +01:00
Ayush Chaurasia
5fbea25f0b
Use integrations only in master process (#89)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-12-24 17:39:30 +01:00
Glenn Jocher
cb4f20f3cf
Add TensorBoard support (#87)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-24 14:37:46 +01:00
Glenn Jocher
248d54ca03
Update metrics names (#85) 2022-12-24 02:32:24 +01:00
Glenn Jocher
6432afc5f9
Rename img_size to imgsz (#86) 2022-12-24 00:39:09 +01:00
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>
2022-12-23 20:14:21 +01:00
Glenn Jocher
f0fff8c13e
Simplify cli.py and fix Detect train Usage (#83)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-22 15:22:16 +01:00
Glenn Jocher
74feef30c4
Update cache_version = 1.0 (#82) 2022-12-22 12:44:28 +01:00
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>
2022-12-21 00:29:58 +01:00
Ayush Chaurasia
d198f3f9d1
Update docs (#73)
Co-authored-by: Laughing-q <1185102784@qq.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-20 09:53:55 +05:30
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>
2022-12-20 09:50:08 +05:30
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>
2022-12-19 13:24:39 +08:00
Ayush Chaurasia
48c95ba083
Change class depending on dataset in model interface (#77)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-16 13:56:25 +05:30
Ayush Chaurasia
24a7c068ad
update docs (#76) 2022-12-15 11:48:03 +05:30
Ayush Chaurasia
eb5adf4e0b
Model enhancement (#75)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-14 14:33:31 +05:30
Ayush Chaurasia
d85b44f259
Update docs (#71)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-12 09:21:00 +05:30
Laughing
e629335f6d
Revert augment_hyps (#70) 2022-12-09 15:49:53 +05:30
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>
2022-12-08 19:58:13 +05:30
Ayush Chaurasia
7ae45c6cc4
[WIP] Model interface (#68)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Laughing-q <1185102784@qq.com>
2022-12-08 13:22:56 +05:30
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>
2022-12-07 10:33:10 +05:30
Glenn Jocher
479992093c
Flake8 updates (#66) 2022-12-06 15:09:53 -08:00
Laughing
fbeeb5d1e1
add resuming (#63)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-06 08:26:41 +05:30
Glenn Jocher
de3e6ca54d
Fix CITATION.cff typos (#64) 2022-12-05 12:17:25 -08:00
Glenn Jocher
6903ccdf6b
Create CITATION.cff (#62) 2022-12-05 12:14:59 -08:00
Ayush Chaurasia
fe75a9ce67
docs setup (#61)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-05 06:04:57 +05:30
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>
2022-12-03 21:01:23 +05:30
Ayush Chaurasia
5a52e7663a
standalone val (#56)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-30 15:04:44 +05:30
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>
2022-11-29 17:00:08 +05:30
Ayush Chaurasia
d0b0fe2592
CLI updates (#58)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-29 15:41:37 +05:30
Ayush Chaurasia
c5f5b80c04
Deterministic training (#53)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-25 02:06:14 +05:30
ayush chaurasia
793dde365d Revert "update"
This reverts commit 486bc494142744e9d90f0483b611a785381401a7.
2022-11-25 01:54:50 +05:30
ayush chaurasia
486bc49414 update 2022-11-25 01:11:55 +05:30