Skip to content

zjkhahah/yolov10_distillation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

YOLOV10的知识蒸馏 支持mgd和cwd

  1. 使用


    model_s = YOLOv10('yolov10s.pt')#教师模型


    model_l = YOLOv10("yolov10l.pt")#学生模型



    model_s.train(data="coco128.yaml", distillation=model_l, Distillation_loss="cwd",amp=False )




2.自定义更改通道

image-20241019142400314

自定义更改通道

About

yolov10的知识蒸馏

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published