Skip to content

An Object Detection Knowledge Distillation framework powered by pytorch, now having SSD and yolov5.

License

Notifications You must be signed in to change notification settings

bgilby59/Object-Detection-Knowledge-Distillation

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Object Detection Knowledge Distillation(ODKD)

version coverage

The function of this branch is not complete. For ssd and yolov5 distillation, checking other branches.

Release edition is coming Soon...

Update

  1. The first edition is the refactor of branch mbv2-lite, which is an implementation of Chen, G. et al. (2017) ‘Learning efficient object detection models with knowledge distillation’ with SSD-lite structure.

  2. Replace part of code with pytorch api which has same functionality.

  3. Very friendly beginner guidance.

  4. System Architecture

odkd

Useage

$ python setup.py install --user

$ odkd-train ./training_config.yml -t

$ odkd-train training_config.yml
or
$ python -m torch.distributed.launch --nproc_per_node=2 `which odkd-train` training_config.yml

$ odkd-eval ${CHECKPOINTS_PATH}/${RUN_INDEX}/config.yml

TODO

  • Evaluation Module

  • LOG Module

  • Coco dataset support

  • Yolov5 distillation

About

An Object Detection Knowledge Distillation framework powered by pytorch, now having SSD and yolov5.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%