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

1103 Commits

This Branch
This Branch
All Branches
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
44c912647e added working validation 2024-10-23 20:33:28 -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
943a333fae Merge branch 'PGT_and_attribution' of https://github.com/nielseni6/yolov10 into PGT_and_attribution 2024-10-22 23:41:53 -04:00
nielseni6
0efbfe7f4d updated grad to create_graph during training, enabling better pgt_loss optimization 2024-10-22 23:40:46 -04:00
nielseni6
2ccec65edc fixed val and saving weights 2024-10-21 13:56:29 -04:00
nielseni6
411157c18a fixed yaml default 2024-10-21 13:05:25 -04:00
nielseni6
5953d3c9c6 changed save path to not override previous training 2024-10-21 13:04:57 -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
2a95a652bd fixed run_val 2024-10-15 11:53:27 -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
aad320dd80 prediction augment (TTA) fix (#288) 2024-06-23 08:21:40 +00:00
leonnil
2c36ab0f10 ddp resume checkpoint fix (#184) 2024-06-15 16:22:32 +00:00
Wang Ao
36efe34fe1
update README 2024-06-12 13:28:44 +08:00
shiraha721
3a53d91ec8
fix cv2.error when use PIL.Image.Image (#241) 2024-06-11 21:55:42 +08:00
wa22
ea93d4f379 update huggingface hub card template 2024-06-07 16:22:42 +08:00
wa22
e161d70ff7 update huggingface hub card template 2024-06-07 16:19:04 +08:00
Pbatch
d8777c1449
Make postprocessing compatible with nobuco (#204) 2024-06-06 07:57:59 +08:00
Ankan Ghosh
13f6ab770d
Update README.md (#198) 2024-06-05 21:33:23 +08:00
kangsanha
e6d80f3fa6
Update __init__.py (#191) 
/Desktop/ultralytics/runs/detect/ -> /Desktop/yolov10/runs/detect/
 2024-06-04 19:19:53 +08:00
kangsanha
15bdea757a
fix onnx model val.py AssertionError (#190) 2024-06-04 19:18:04 +08:00
leonnil
ba0e6cbc1a fix onnx validation (#179) 2024-06-04 10:29:21 +00:00
wa22
e5f2f03c55 use yolov10 by default in cli 2024-06-04 11:06:48 +08:00
wa22
dfed201384 add support for HuggingFace Hub for cli 2024-06-03 23:04:27 +08:00
Ekaterina Aidova
060af8b8c1
Small fix in detection head for enabling ov export from scratch (#183) 2024-06-03 17:59:44 +08:00
Sencer Yücel
1cfe7a4e13
Added video output support for gradio (#113) 
added video output support for gradio

---------

Co-authored-by: Sencer Yücel <sencer.yucel@turkai.com>
Co-authored-by: wa22 <wa22@mails.tsinghua.edu.cn>
 2024-06-03 17:15:31 +08:00
wa22
1539b5a678 add calculation for flops 2024-06-03 11:21:49 +08:00
wa22
587981fc27 update app.py 2024-06-03 10:50:08 +08:00
NielsRogge
a6e064d331
Track downloads (#172) 
Track downloads from the HuggingFace hub
 2024-06-03 10:23:37 +08:00
wa22
7a0cce17b6 add names when pushing to huggingface hub 2024-06-03 10:21:43 +08:00
Erlangga Yudi Pradana
0efdd2f0c4
docs: update README.md (#174) 2024-06-01 22:48:36 +08:00
wa22
22f49e1249 update requirements.txt 2024-06-01 21:32:44 +08:00
wa22
16676546f1 update README 2024-06-01 21:32:27 +08:00
wa22
8d0e567e58 update README 2024-06-01 20:18:30 +08:00
wa22
0f87aa3173 Merge branch 'main' of github.com:jameslahm/yolov10 into main 2024-06-01 19:12:41 +08:00
wa22
43c4a76043 update requirements.txt 2024-06-01 19:11:00 +08:00
NielsRogge
989cdb812d
Add mixin (#168) 2024-06-01 19:05:46 +08:00
wa22
b70277e940 update README 2024-06-01 18:57:08 +08:00
wa22
22aa8412a7 update README 2024-06-01 10:53:02 +08:00
wa22
3594353249 update README 2024-05-31 18:33:16 +08:00
yuyoujiang
393091b63e
update README (#157) 2024-05-31 17:02:14 +08:00
mohamedsamirx
c829350969
Update README.md (#152) 2024-05-31 12:57:49 +08:00
wa22
6d402a68e1 Merge branch 'main' of github.com:jameslahm/yolov10 into main 2024-05-31 10:55:17 +08:00
wa22
1788e9d4e8 update README 2024-05-31 10:55:07 +08:00
Veeja Liu
634ceee537
Change the format of README.md (#137) 
Co-authored-by: weijialiu <weijialiu@hireez.com>
 2024-05-31 10:27:26 +08:00
wa22
27a889edbd update README 2024-05-31 09:26:52 +08:00
wa22
bbcad68d77 update README 2024-05-31 09:25:17 +08:00