Commit Graph

250 Commits (e6d18cc944e0ca13ebb1aad2be0de35f2830c549)

Author SHA1 Message Date
Laughing 620f3eb218
`ultralytics 8.0.136` refactor and simplify package () 2 years ago
Glenn Jocher 8ebe94d1e9
`ultralytics 8.0.135` remove deprecated `v5loader` () 2 years ago
Glenn Jocher 135a10f1fa
Avoid CUDA round-trip for relevant export formats () 2 years ago
Adrian Boguszewski e2f9bba964
Update OpenVINO requirement to `openvino>=2023.0` () 2 years ago
Chaoning Zhang 201e69e4e4
`ultralytics 8.0.134` add MobileSAM support () 2 years ago
Glenn Jocher c55a98ab8e
`ultralytics 8.0.133` add `torchvision` compatibility check () 2 years ago
Yonghye Kwon cfcae388f2
Add `datetime` to exported model metadata () 2 years ago
triple Mu 5592a3abdf
Fix PNNX debug filename () 2 years ago
Glenn Jocher 18ada51931
Add tuner.py to Docs () 2 years ago
triple Mu 44aa18c99a
Remove PNNX debug files () 2 years ago
Glenn Jocher 391b7e67cf
Filter `matplotlib>=3.7.2` tight layout warning () 2 years ago
Glenn Jocher d9db6cd42d
`ultralytics 8.0.131` NCNN, HUB and FastSAM fixes () 2 years ago
Colin Wong 23c7cd4c9f
Avoid Python 3.9 with_stem () 2 years ago
Glenn Jocher abb0939fe8
`ultralytics 8.0.129` add YOLOv8 Tencent NCNN export () 2 years ago
Glenn Jocher 91905b4b0b
Pass export=True to RTDETRDecoder () 2 years ago
Glenn Jocher ee6cf11378
Save Keypoints visible to text () 2 years ago
Glenn Jocher 4c22273198
`ultralytics 8.0.126` Ray Tune refactoring () 2 years ago
Glenn Jocher 8a11eda4a9
`ultralytics 8.0.125` NMS speed improvements () 2 years ago
Neloy262 586c95b8a9
Add torch.Tensor results image saving () 2 years ago
Glenn Jocher 8b50b2a897
Fix video writing () 2 years ago
Glenn Jocher 682c9ef70f
`ultralytics 8.0.122` Fix `torch.Tensor` inference () 2 years ago
Glenn Jocher 5381fc8a58
Warn `save` disabled for torch inference () 2 years ago
Jieun Park 2ebd808b69
Increase NCCL timeout from 1 hour to 3 hours () 2 years ago
Glenn Jocher 35c02c8319
Fix Pose export from YAML () 2 years ago
Glenn Jocher 9c44560ec4
Return `save_dir` with prediction results () 2 years ago
Glenn Jocher bd0f7ecf6f
Docstrings arguments cleanup () 2 years ago
Ayush Chaurasia a0ba8ef5f0
Add RTDETR Trainer () 2 years ago
Glenn Jocher 0dfafaf39b
Fix `batch=-1` for exports () 2 years ago
Glenn Jocher d69a1e8046
Update save_dir on new predict args () 2 years ago
Bruno Arine 0d91d6df6e
Add world_size check before setting up DDP train () 2 years ago
Glenn Jocher c340f84ce9
`ultralytics 8.0.117` NAS export, classify and tasks banner URL fixes () 2 years ago
Glenn Jocher e88c89d269
Fix RayTune for < 8 CPUs () 2 years ago
Glenn Jocher 8ac8ff72ae
Fix code scanning alert - Incomplete URL substring sanitization () 2 years ago
Glenn Jocher 8940a27bdb
`ultralytics 8.0.114` automatic optimizer selection () 2 years ago
Zlobin Vladimir 01273c5ca1
Add rt info for OpenVINO ModelAPI () 2 years ago
Glenn Jocher 451cf8b647
Add Adamax, NAdam, RAdam optimizers () 2 years ago
Laughing bb5bbc2893
Fix keypoints.conf update Results docs () 2 years ago
crbrz a9129fb40e
Fix `workspace` and `verbose` arguments in TensorRT export () 2 years ago
Ayush Chaurasia 67cf53b475
Move `check_amp()` to checks.py () 2 years ago
Laughing f4b34fc30b
Fix `save_txt` in track mode and add Keypoints and Probs () 2 years ago
Laughing 305cde69d0
Update `generate_ddp_file` for improved `overrides` () 2 years ago
Ayush Chaurasia facb7861cf
Move loss calculation to head () 2 years ago
Afif Al Mamun 75012e575e
Rename predictor postprocess `orig_img` to `orig_imgs` () 2 years ago
John-Henry Lim 1a26e1e195
kwargs override `cfg` argument in model.train () () 2 years ago
setwgx 0c62f85750
Fix KeyError:line_thickness () 2 years ago
Ayush Chaurasia 527a97759b
Revert loss head PR () 2 years ago
Glenn Jocher 61fa5efe6d
`build_optimizer()` assign all parameters () 2 years ago
Ayush Chaurasia d19c5b6ce8
Move loss to task heads () 2 years ago
Glenn Jocher ffc0e8ccf7
`ultralytics 8.0.108` add Meituan YOLOv6 models () 2 years ago
Glenn Jocher dada5b73c4
`ultralytics 8.0.107` () 2 years ago