Data

Tools

Indexes

Applications

Experiments

Open Source Science

yolov8_distillation

https://github.com/huangzongmou/yolov8_distillation

Science Score: 44.0%

This score indicates how likely this project is to be science-related based on various indicators:

  • ✓
    CITATION.cff file
    Found CITATION.cff file
  • ✓
    codemeta.json file
    Found codemeta.json file
  • ✓
    .zenodo.json file
    Found .zenodo.json file
  • ○
    DOI references
  • ○
    Academic publication links
  • ○
    Academic email domains
  • ○
    Institutional organization owner
  • ○
    JOSS paper metadata
  • ○
    Scientific vocabulary similarity
    Low similarity (0.3%) to scientific vocabulary
Last synced: 7 months ago · JSON representation ·

Repository

Basic Info
  • Host: GitHub
  • Owner: huangzongmou
  • License: agpl-3.0
  • Language: Python
  • Default Branch: master
  • Size: 1.53 MB
Statistics
  • Stars: 93
  • Watchers: 1
  • Forks: 15
  • Open Issues: 12
  • Releases: 0
Created almost 3 years ago · Last pushed over 2 years ago
Metadata Files
Readme Contributing License Citation Security

README.md

知识蒸馏

训练教师模型

model_t = YOLO('yolov8l.pt')
model_t.train(data=data, epochs=100, imgsz=640, Distillation = None)

学生模型

model_s = YOLO('yolov8s.pt')
model_s.train(data=data, epochs=100, imgsz=640, Distillation = model_t.model)

提示

选择蒸馏方法施主自行到/ultralytics/yolo/engine/trainer.py：176行更改通道数以及CWDLoss 或者 MGDLoss。

效果

炼丹需要看各位施主的缘分。一般提高0.2-0.5map。

心得

大学教授不一定教好小学生。

Owner

  • Login: huangzongmou
  • Kind: user

Citation (CITATION.cff) 

cff-version: 1.2.0
preferred-citation:
  type: software
  message: If you use this software, please cite it as below.
  authors:
  - family-names: Jocher
    given-names: Glenn
    orcid: "https://orcid.org/0000-0001-5950-6979"
  - family-names: Chaurasia
    given-names: Ayush
    orcid: "https://orcid.org/0000-0002-7603-6750"
  - family-names: Qiu
    given-names: Jing
    orcid: "https://orcid.org/0000-0003-3783-7069"
  title: "YOLO by Ultralytics"
  version: 8.0.0
  # doi: 10.5281/zenodo.3908559  # TODO
  date-released: 2023-1-10
  license: AGPL-3.0
  url: "https://github.com/ultralytics/ultralytics"

GitHub Events

Total
  • Watch event: 20
  • Fork event: 3
Last Year
  • Watch event: 20
  • Fork event: 3

Dependencies

docker/Dockerfile docker
  • pytorch/pytorch 2.0.0-cuda11.7-cudnn8-runtime build
requirements.txt pypi
  • Pillow >=7.1.2
  • PyYAML >=5.3.1
  • matplotlib >=3.2.2
  • numpy >=1.21.6
  • opencv-python >=4.6.0
  • pandas >=1.1.4
  • psutil *
  • requests >=2.23.0
  • scipy >=1.4.1
  • seaborn >=0.11.0
  • thop >=0.1.1
  • torch >=1.7.0
  • torchvision >=0.8.1
  • tqdm >=4.64.0
setup.py pypi