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,104 Commits 1 Branch 2 Tags
Commit Graph

1104 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
nielseni6
1ef6cbcec5 added pgt_coeff to argparser 2024-11-01 11:19:22 -04:00
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