search

pjreddie / darknet

Convolutional Neural Networks
http://pjreddie.com/darknet/
Other
25.7k stars 21.33k forks source link

Yolo parameter quantization #1476

Open wu-y opened 5 years ago

wu-y commented 5 years ago

Hello, I want to quantify the yolov2 parameter, how can I extract the settlement result to each layer in the middle?

AlexeyAB commented 5 years ago

@wu-y Hi,

You can try to use this repository, as described here: https://github.com/AlexeyAB/yolo2_light#yolo2_light

wu-y commented 5 years ago

Thank you very much for your help. The code you provided seems to only quantify weights. I want to save the calculation results of each layer of yolo and quantify it. Can you help?

AlexeyAB commented 5 years ago

@wu-y

What do you mean?

It quantify both weights & inputs, so forward inference executed by using only INT8 or BIT1 for convolution, so the speed will be increased.