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

201 Commits

Author SHA1 Message Date
nielseni6
21e660fde9 Fixed issues with pgt training, adding the pgt loss to the YOLO loss function. PGT loss can now be tracked during training. 2024-11-01 11:04:12 -04:00
nielseni6
9524af3bfe Model loading using YOLOv10PGT added, and pgt_coeff is now a cfg parameter 2024-10-23 20:28:51 -04:00
nielseni6
3a449d5a6c Fixed PGT by including it in the loss function 2024-10-21 12:20:24 -04:00
nielseni6
38fa59edf2 PGT Training now functioning. Run run_pgt_train.py to train the model. 2024-10-16 19:26:41 -04:00
nielseni6
b02bf58de6 added attribution visualization with run_attribution.py 2024-10-15 11:51:37 -04:00
inisis
cd2f79c702
fix onnxslim version (#380) 2024-07-30 15:16:55 +08:00
leonnil
0e51495ce6 replace onnxsim with onnxslim 2024-07-30 04:34:28 +00:00
leonnil
2c36ab0f10 ddp resume checkpoint fix (#184) 2024-06-15 16:22:32 +00:00
shiraha721
3a53d91ec8
fix cv2.error when use PIL.Image.Image (#241) 2024-06-11 21:55:42 +08:00
NielsRogge
a6e064d331
Track downloads (#172) 
Track downloads from the HuggingFace hub
 2024-06-03 10:23:37 +08:00
NielsRogge
989cdb812d
Add mixin (#168) 2024-06-01 19:05:46 +08:00
wa22
c617472b55 update README 2024-05-26 23:58:44 +08:00
wa22
7a8f9643d2 fix print val map 2024-05-26 03:49:16 +00:00
Thomas Friedel
2f6208c356
fix for KeyError: 'epoch' (#48) 
* fix for KeyError: 'epoch'
 2024-05-26 10:32:05 +08:00
wa22
66bca50c47 fix non-coco val 2024-05-24 07:38:21 +00:00
wa22
b714fa8bae update 2024-05-24 05:40:00 +00:00
wa22
8d9f45a955 add yolov10 2024-05-23 14:12:00 +00:00
Glenn Jocher
5be2ffbd13
ultralytics 8.1.34 Inference API robust imgsz checks (#9274) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2024-03-25 02:00:38 +01:00
Kalen Michael
ec1d110689
ultralytics 8.1.33 fix HUB model checks (#9153) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2024-03-24 06:14:15 +01:00
Laughing
33607ca278
Add bgr hyperparameter (#9139) 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2024-03-23 23:17:00 +01:00
Glenn Jocher
3c179f87cb
ultralytics 8.1.31 NCNN and CLIP updates (#9235) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2024-03-22 23:51:50 +01:00
Kalen Michael
8617fcf32d
ultralytics 8.1.30 add advanced HUB train arguments (#9110) 
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2024-03-20 15:53:00 -07:00
UltralyticsAssistant
bc34ae560b Auto-format by https://ultralytics.com/actions 2024-03-15 00:12:44 +00:00
Lakshantha Dissanayake
368abb4cc8
Optimize TFJS export on ARM64 (#8946) 
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2024-03-15 01:12:22 +01:00
Glenn Jocher
d24ead447e
More robust PNNX downloads (#8866) 2024-03-11 20:13:49 +01:00
Glenn Jocher
617ab21523
Faster batch same_shapes (#8851) 2024-03-11 19:53:35 +01:00
Laughing
2ea6b2b889
ultralytics 8.1.27 batched tracking fixes (#8842) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2024-03-11 19:29:41 +01:00
Glenn Jocher
7638c5ce4d
Integrate OpenVINO CUMULATIVE_THROUGHPUT mode batched inference (#8834) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2024-03-11 18:50:29 +01:00
Burhan
ff8ae0d2e2
Batch inference fixes (#8854) 2024-03-11 17:14:03 +01:00
Glenn Jocher
7451ca1f54
ultralytics 8.1.26 LoadImagesAndVideos batched inference (#8817) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2024-03-10 19:15:44 +01:00
Glenn Jocher
2bc605f32a
ultralytics 8.1.25 fix **kwargs: (dict) warnings (#8815) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2024-03-09 18:51:38 +01:00
Finlay Morrison
10bb39d828
Modify tuner best.pt logic to train first (#8792) 2024-03-08 15:57:12 +01:00
Johnny
b53a3989ab
OpenVino 2024.0.0 (#8726) 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2024-03-06 21:55:23 +01:00
Glenn Jocher
90943946a7
ultralytics 8.1.25 OpenVINO LATENCY and THROUGHPUT modes (#8058) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Adrian Boguszewski <adekboguszewski@gmail.com>
 2024-03-06 11:38:05 +01:00
Laughing
ddc94a6981
OBB: Fix when training on zip data (#8680) 
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2024-03-05 18:37:33 +01:00
Adrian Boguszewski
a7cfd83c5f
ultralytics 8.1.24 new OpenVINO 2023.3 export updates (#8417) 
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2024-03-05 15:49:02 +01:00
Laughing
2071776a36
ultralytics 8.1.23 add YOLOv9-C and E models (#8571) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2024-03-04 14:27:23 +01:00
Laughing
946e18f79c
ultralytics 8.1.21 Add YOLOv8-World-v2 models (#8580) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
 2024-03-03 22:56:57 +01:00
Glenn Jocher
906b8d31dc
Add distance calculation feature in vision-eye (#8616) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Muhammad Rizwan Munawar <chr043416@gmail.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
 2024-03-03 19:36:33 +01:00
Glenn Jocher
36408c974c
Update TFLite Docs images (#8605) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2024-03-03 01:59:43 +01:00
Glenn Jocher
1146bb0582
Update IoU capitalization (#8604) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Dean Mark <2552482+deanmark@users.noreply.github.com>
 2024-03-02 22:12:23 +01:00
Andrew
e0b8b36967
Replace onnx2tf usage from CLI to Python (#8429) 
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2024-03-02 21:29:53 +01:00
Lakshantha Dissanayake
2100d28638
Fix cmake requirements for ARM64 (#8589) 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2024-03-02 16:03:21 +01:00
Glenn Jocher
6bdf8dfaa2
Fix undefined ‘im_array’ bug in predict.md (#8565) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: fang_chenfang <1217690899@qq.com>
 2024-03-01 00:53:17 +01:00
Abirami Vina
1db8e912a3
Add TorchScript Docs Integrations Page (#8501) 
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
 2024-03-01 00:25:31 +01:00
Glenn Jocher
f8e681c2be
ultralytics 8.1.19 PNNX aarch64 linux fix (#8480) 
Co-authored-by: Burhan <62214284+Burhan-Q@users.noreply.github.com>
Co-authored-by: Kayzwer <68285002+Kayzwer@users.noreply.github.com>
 2024-02-27 00:02:29 +01:00
Lakshantha Dissanayake
0572b29445
ultralytics 8.1.18 add cmake for building onnxsim on aarch64 (#8395) 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
 2024-02-22 23:43:09 +01:00
Alexander Suslov
5171d6679d
Added check of parameter type before setting required_grad=True for frozen layers (#8318) 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2024-02-20 17:32:04 +01:00
Lakshantha Dissanayake
9e24743861
Fix an issue with flatbuffer version for Raspberry Pi (#8311) 
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2024-02-19 22:16:40 +01:00
Glenn Jocher
7f0d5ef9d7
Model typehints Docker fix (#8306) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2024-02-19 20:29:46 +01:00