Is it possible to use a model with YOLO framework?
matt-sharp opened this issue · 4 comments
Is it possible to use a model with YOLO framework?
Since this is written in C I'm not sure it will work with Pytorch.
Hi @matt-sharp ,
Technically, yes. There are some PyTorch implementations of YOLO models e.g., DeNA/PyTorch_YOLOv3, and probably you can introduce torchdistill to the framework.
If you can provide more details such as what teacher and student models you are considering, from which layers you'd like to extract input/output for distillation, etc, I could make more concrete suggestions.
@matt-sharp Closing this issue as I haven't seen any follow-up for a while.
Open a new Discussion (not Issue) if you still have questions
Dear anthor：
Thank you for your contribution, do you have any idea of adding distillation to YOLO in the near future, after all, there are too many people in YOLO all over the world, I think that is a very great work！
Hello @ppogg
Do you have any specific knowledge distillation method in mind? If so, feel free to create a Discussion (not here) and leave a comment.
Otherwise, I do not have any plan to add YOLO example because knowledge distillation in object detection is more challenging than other tasks as mentioned here