Can Ristretto framework be used in quantizing yolo v2?

110 views
Skip to first unread message

jiangwei...@gmail.com

unread,
Mar 31, 2018, 12:24:36 PM3/31/18
to ristretto-users
There are several works on training yolov2 on caffe, so I wonder if there exist the possibility that I can use Ristretto for quantizing yolov2 by firstly train yolov2 on caffe then quantizing it on ristretto.

pmg...@ucdavis.edu

unread,
Apr 17, 2018, 2:34:10 AM4/17/18
to ristretto-users
Thanks for your question, sorry for the later response. Yes, you can quantize YOLO V2 with Ristretto. As you said, you first train the network in Caffe, then quantize it using Ristretto. The network has many convolution layers which will get quantized by Ristretto. I have never quantized this particular CNN myself though, so let me know about your results.

Best,
Philipp
Reply all
Reply to author
Forward
0 new messages