squeezeDet icon indicating copy to clipboard operation
squeezeDet copied to clipboard

Caffe model

Open ShervinAr opened this issue 7 years ago • 7 comments

Hello,

Is there any chance you could provide your model's structure in Caffe?

many thanks

ShervinAr avatar May 11 '17 08:05 ShervinAr

Check out this git repository:

https://github.com/DeepScale/SqueezeNet

Squeezedet builds upon this network by adding fire 10, fire 11 and conv12 and structure for the loss function. So that might be a good starting point.

Timen avatar May 12 '17 11:05 Timen

@Timen many thanks for your response. could you please provide the conv12 structure in Caffe?

ShervinAr avatar May 12 '17 13:05 ShervinAr

I do not have experience with Caffe, so I can't help you with that.

Timen avatar May 12 '17 13:05 Timen

@ShervinAr Thanks for your question.

Currently, I don't have a plan to re-implement SqueezeDet in Caffe or other deep learning tools. But if you only need inference or deployment, it should be straightforward to convert SqueezeDet's network structure to protobuf and convert tensorflow checkpoint file to caffemodel file. If anyone plan to implement this conversion, please let me know and I'm happy to add a pointer on the front page to your work.

BichenWuUCB avatar May 13 '17 01:05 BichenWuUCB

@BichenWuUCB Many thanks for your response. Yes, actually I only need inference (deployment). Could you please provide some hints on how to do such a conversion (i.e. convert SqueezeDet's structure to prototxt and convert the tensorflow check point file to a caffemodel file) as I am not familiar with Tensorflow?

ShervinAr avatar May 13 '17 05:05 ShervinAr

@BichenWuUCB I would also be very thankful if you could let me know the dimensionalities of the outputs of fire10 and fire11 modules in the SqueezDet . Are these both 384?

ShervinAr avatar May 13 '17 05:05 ShervinAr

You can find the model definition from here

BichenWuUCB avatar May 13 '17 05:05 BichenWuUCB