xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-25 23:44:22 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
539 Commits 1 Branch 2 Tags
Commit Graph

250 Commits

Author SHA1 Message Date
Laughing
620f3eb218
ultralytics 8.0.136 refactor and simplify package (#3748) 
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>
 2023-07-16 17:47:45 +02:00
Glenn Jocher
8ebe94d1e9
ultralytics 8.0.135 remove deprecated v5loader (#3744) 2023-07-15 01:38:32 +02:00
Glenn Jocher
135a10f1fa
Avoid CUDA round-trip for relevant export formats (#3727) 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2023-07-14 20:38:31 +02:00
Adrian Boguszewski
e2f9bba964
Update OpenVINO requirement to openvino>=2023.0 (#3722) 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2023-07-13 18:13:55 +02:00
Chaoning Zhang
201e69e4e4
ultralytics 8.0.134 add MobileSAM support (#3474) 
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>
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
Co-authored-by: Laughing-q <1185102784@qq.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2023-07-13 14:25:56 +02:00
Glenn Jocher
c55a98ab8e
ultralytics 8.0.133 add torchvision compatibility check (#3703) 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2023-07-13 01:40:50 +02:00
Yonghye Kwon
cfcae388f2
Add datetime to exported model metadata (#3694) 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2023-07-12 19:05:06 +02:00
triple Mu
5592a3abdf
Fix PNNX debug filename (#3687) 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2023-07-12 13:22:56 +02:00
Glenn Jocher
18ada51931
Add tuner.py to Docs (#3689) 2023-07-12 13:19:40 +02:00
triple Mu
44aa18c99a
Remove PNNX debug files (#3677) 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2023-07-12 13:03:37 +02:00
Glenn Jocher
391b7e67cf
Filter matplotlib>=3.7.2 tight layout warning (#3614) 2023-07-10 14:52:59 +02:00
Glenn Jocher
d9db6cd42d
ultralytics 8.0.131 NCNN, HUB and FastSAM fixes (#3587) 2023-07-08 02:04:33 +02:00
Colin Wong
23c7cd4c9f
Avoid Python 3.9 with_stem (#3589) 
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>
 2023-07-08 00:16:42 +02:00
Glenn Jocher
abb0939fe8
ultralytics 8.0.129 add YOLOv8 Tencent NCNN export (#3529) 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2023-07-07 15:17:38 +02:00
Glenn Jocher
91905b4b0b
Pass export=True to RTDETRDecoder (#3550) 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2023-07-05 18:25:27 +02:00
Glenn Jocher
ee6cf11378
Save Keypoints visible to text (#3547) 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2023-07-05 17:14:53 +02:00
Glenn Jocher
4c22273198
ultralytics 8.0.126 Ray Tune refactoring (#3511) 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2023-07-05 00:30:17 +02:00
Glenn Jocher
8a11eda4a9
ultralytics 8.0.125 NMS speed improvements (#3463) 
Co-authored-by: Compunet <117437050+dme-compunet@users.noreply.github.com>
 2023-07-03 16:34:13 +02:00
Neloy262
586c95b8a9
Add torch.Tensor results image saving (#3475) 
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>
 2023-07-03 16:08:02 +02:00
Glenn Jocher
8b50b2a897
Fix video writing (#3379) 2023-06-25 23:11:00 +02:00
Glenn Jocher
682c9ef70f
ultralytics 8.0.122 Fix torch.Tensor inference (#3363) 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: krzysztof.gonia <4281421+kgonia@users.noreply.github.com>
 2023-06-25 01:36:07 +02:00
Glenn Jocher
5381fc8a58
Warn save disabled for torch inference (#3361) 2023-06-24 15:57:17 +02:00
Jieun Park
2ebd808b69
Increase NCCL timeout from 1 hour to 3 hours (#3343) 
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>
 2023-06-24 15:56:18 +02:00
Glenn Jocher
35c02c8319
Fix Pose export from YAML (#3312) 2023-06-21 18:04:22 +02:00
Glenn Jocher
9c44560ec4
Return save_dir with prediction results (#3240) 2023-06-18 02:34:31 +02:00
Glenn Jocher
bd0f7ecf6f
Docstrings arguments cleanup (#3229) 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2023-06-17 16:17:07 +02:00
Ayush Chaurasia
a0ba8ef5f0
Add RTDETR Trainer (#2745) 
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>
Co-authored-by: Laughing-q <1185102784@qq.com>
Co-authored-by: Kayzwer <68285002+Kayzwer@users.noreply.github.com>
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
 2023-06-17 13:46:18 +02:00
Glenn Jocher
0dfafaf39b
Fix batch=-1 for exports (#3217) 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2023-06-17 01:07:24 +02:00
Glenn Jocher
d69a1e8046
Update save_dir on new predict args (#3215) 2023-06-16 19:59:46 +02:00
Bruno Arine
0d91d6df6e
Add world_size check before setting up DDP train (#3191) 2023-06-16 02:05:15 +02:00
Glenn Jocher
c340f84ce9
ultralytics 8.0.117 NAS export, classify and tasks banner URL fixes (#3145) 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2023-06-12 20:14:46 +02:00
Glenn Jocher
e88c89d269
Fix RayTune for < 8 CPUs (#3102) 2023-06-08 18:50:05 +02:00
Glenn Jocher
8ac8ff72ae
Fix code scanning alert - Incomplete URL substring sanitization (#3054) 2023-06-07 08:16:31 +02:00
Glenn Jocher
8940a27bdb
ultralytics 8.0.114 automatic optimizer selection (#3037) 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Burhan <62214284+Burhan-Q@users.noreply.github.com>
 2023-06-06 01:09:27 +02:00
Zlobin Vladimir
01273c5ca1
Add rt info for OpenVINO ModelAPI (#2891) 
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>
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
 2023-06-06 00:29:12 +02:00
Glenn Jocher
451cf8b647
Add Adamax, NAdam, RAdam optimizers (#2969) 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2023-06-04 22:35:50 +02:00
Laughing
bb5bbc2893
Fix keypoints.conf update Results docs (#2977) 2023-06-04 22:33:17 +02:00
crbrz
a9129fb40e
Fix workspace and verbose arguments in TensorRT export (#2954) 
Co-authored-by: crbrz <cristiab@gmail.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2023-06-02 16:06:01 +02:00
Ayush Chaurasia
67cf53b475
Move check_amp() to checks.py (#2948) 
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>
 2023-06-02 15:43:07 +02:00
Laughing
f4b34fc30b
Fix save_txt in track mode and add Keypoints and Probs (#2921) 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2023-06-01 00:40:10 +02:00
Laughing
305cde69d0
Update generate_ddp_file for improved overrides (#2909) 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2023-05-30 19:41:44 +02:00
Ayush Chaurasia
facb7861cf
Move loss calculation to head (#2874) 
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>
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
 2023-05-30 19:24:30 +02:00
Afif Al Mamun
75012e575e
Rename predictor postprocess orig_img to orig_imgs (#2877) 
Signed-off-by: Afif Al Mamun <afifanik@gmail.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2023-05-28 17:54:05 +02:00
John-Henry Lim
1a26e1e195
kwargs override cfg argument in model.train (#2839) (#2853) 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2023-05-28 16:36:47 +02:00
setwgx
0c62f85750
Fix KeyError:line_thickness (#2852) 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2023-05-28 16:23:10 +02:00
Ayush Chaurasia
527a97759b
Revert loss head PR (#2873) 
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>
 2023-05-28 16:15:41 +02:00
Glenn Jocher
61fa5efe6d
build_optimizer() assign all parameters (#2855) 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2023-05-26 21:03:03 +02:00
Ayush Chaurasia
d19c5b6ce8
Move loss to task heads (#2825) 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Laughing-q <1185102784@qq.com>
 2023-05-25 16:07:54 +05:30
Glenn Jocher
ffc0e8ccf7
ultralytics 8.0.108 add Meituan YOLOv6 models (#2811) 
Co-authored-by: Michael Currie <mcurrie@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Hicham Talaoubrid <98521878+HichTala@users.noreply.github.com>
Co-authored-by: Zlobin Vladimir <vladimir.zlobin@intel.com>
Co-authored-by: Szymon Mikler <sjmikler@gmail.com>
 2023-05-25 00:43:32 +02:00
Glenn Jocher
dada5b73c4
ultralytics 8.0.107 (#2778) 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Peter van Lunteren <contact@pvanlunteren.com>
 2023-05-23 13:50:24 +02:00