250 Commits (e71c0e3748c942b719e3dc9daa19cc2aaa5f4391)

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