Object-Detection-Knowledge-Distillation icon indicating copy to clipboard operation
Object-Detection-Knowledge-Distillation copied to clipboard

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

Results 14 Object-Detection-Knowledge-Distillation issues
Sort by recently updated
recently updated
newest added

Hello, Can you please explain how these values were computed? MEANS = (127, 127, 127) I tried to compute them by myself on the VOC2007+2012 dataset, and I obtained these...

can I use different dataset instead of VOC,please ?

``` size mismatch for loc.0.weight: copying a param with shape torch.Size([16, 512, 3, 3]) from checkpoint, the shape in current model is torch.Size([16, 1024, 3, 3]). size mismatch for loc.1.weight:...

Hello, Firstly, thank you very much for this great repository. I am really interested to run your code on a video, where the teacher transfers his knowledge to the student....

Hi, I always got the run time error after I ran python eval.py mbv2 --trained_model=checkpoints/student_mbv2_500_3934.pth. Does anyone know how to solve this error? Thanks.

Thanks for the awesome implementation. I had a look at the losses and for the yolov5 implementation: 1. this implementation does not have the teacher bounded regression loss. 2. Also,...

https://github.com/SsisyphusTao/SSD-Knowledge-Distillation/blob/0597fbee635afcf0b8710ba3a9e40ab9f010aea5/nets/multibox_loss.py#L17-L23 F.mse_loss has reduction='mean' by default, but we should use reduction='sum' as for `loss_l` https://github.com/SsisyphusTao/SSD-Knowledge-Distillation/blob/0597fbee635afcf0b8710ba3a9e40ab9f010aea5/nets/multibox_loss.py#L120 I think the next code corresponds to the formula from the article. ``` def bounded_regression_loss(Rs,...

https://github.com/SsisyphusTao/SSD-Knowledge-Distillation/blob/0597fbee635afcf0b8710ba3a9e40ab9f010aea5/nets/multibox_loss.py#L158

https://github.com/SsisyphusTao/SSD-Knowledge-Distillation/blob/0597fbee635afcf0b8710ba3a9e40ab9f010aea5/nets/multibox_loss.py#L8