21 Commits (441126a77e3d65409e2a2b78d742a5f30a41acb2)

Author SHA1 Message Date
Glenn Jocher 6f0ba81427
Add best.pt val and COCO pycocotools val (#98)
2 years ago
Glenn Jocher 48cffa176e
Shorten module paths with new 'nn' dir (#96)
2 years ago
Glenn Jocher 249dfbdc05
Fix ClearML Mosaic callback to 'on_train_epoch_end' (#92)
2 years ago
Glenn Jocher 98815d560f
YOLOv5 updates (#90)
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)
2 years ago
Ayush Chaurasia e6737f1207
Predictor support (#65)
2 years ago
Ayush Chaurasia 5a52e7663a
standalone val (#56)
2 years ago
Laughing 3a241e4cea
update segment training (#57)
2 years ago
ayush chaurasia 793dde365d Revert "update"
2 years ago
ayush chaurasia 486bc49414 update
2 years ago
Ayush Chaurasia 298287298d
Add clearml logging (#51)
2 years ago
Ayush Chaurasia 4291b9c31c
Add EMA and model checkpointing (#49)
2 years ago
Glenn Jocher 27d6545117
General console printout updates (#48)
2 years ago
Laughing 47f1cb3ef4
Fix some cuda training issues of segmentation (#46)
2 years ago
Ayush Chaurasia d143ac666f
[rename] - preprocess-batch -> preprocess, preprocess_preds -> postprocess (#42)
2 years ago
Glenn Jocher 4c68b9dcf6
Add is_colab() and is_kaggle() (#41)
2 years ago
Ayush Chaurasia f56c9bcc26
Segmentation support & other enchancements (#40)
2 years ago
Ayush Chaurasia 7b560f7861
Model builder (#29)
2 years ago
Ayush Chaurasia c5cb76b356
Metrics and loss structure (#28)
2 years ago
Ayush Chaurasia 419f6d7167
Add yolo package structure (#15)
2 years ago