search

chenzhi1992 / TensorRT-SSD

Use TensorRT API to implement Caffe-SSD， SSD（channel pruning）， Mobilenet-SSD
251 stars 84 forks source link

Channel pruning SSD #13

Open GraphicsHunter opened 6 years ago

GraphicsHunter commented 6 years ago

hi chenzhi. I am trying to understand how to channel prune ssd, and found by googling this link: https://github.com/yihui-he/channel-pruning. I do not understand how to channel prune ssd as you have said you did in the README. Could you make a short guide on what you did to channel prune SSD? I am not sure what prototxt and caffemodel I should pass in to the script, and so on... Thanks.

chenzhi1992 commented 6 years ago

There is the vgg16-pruning prototxt in https://github.com/yihui-he/channel-pruning, and what you need to do is to combine it with SSD.

GraphicsHunter commented 6 years ago

hi @chenzhi1992 , how do I combine the network with SSD? Do you have the time to make a short guide with the commands you run to do this?

chenzhi1992 commented 6 years ago

It mainly reduces some of the channels and does not add some layers. If your precision is not good enough, you can choose to add some bn and scale layers. @tianfanzhu

GraphicsHunter commented 6 years ago

Hi again @chenzhi1992. I think you misunderstood my question. I was wondering how to perform the "combining" of VGG16-caffemodel and SSD-caffemodel. Do I need to combine both the caffemodel and prototxt, or only the prototxt? And how do I do it?

chenzhi1992 commented 6 years ago

Okay, I see what you mean. I will organize a tutorial.

GraphicsHunter commented 6 years ago

Thank you so much. I will look forward to the tutorial

GraphicsHunter commented 6 years ago

Hi @chenzhi1992, hope you had a great easter. When can a possible tutorial for channel pruning SSD be expected?

GraphicsHunter commented 6 years ago

@chenzhi1992 sorry for asking much, when can we expect an tutorial on how to perform channel pruning on SSD?

chenzhi1992 commented 6 years ago

This tutorial is not very well written. I perform channel pruning on ssd by referring to the SSD+VGG format .@tianfanzhu

GraphicsHunter commented 6 years ago

Hi again @chenzhi1992 , I just want to know which commands to run to do the channel pruning on SSD as i do not understand what scripts to run to do the pruning on the VGG part of SSD. Are you able to provide which files(i.e. prototxt and caffemodel files) and which scripts(from channel pruning repo) you ran to perform the channel pruning on SSD?

GraphicsHunter commented 6 years ago

Hello @chenzhi1992. I have still not figured out how to do channel pruning on SSD, what commands did and file did you use to do it?

GraphicsHunter commented 6 years ago

@chenzhi1992 hi, are you able to look into it?

GraphicsHunter commented 6 years ago

@chenzhi1992 Hello. Hope you are doing well. Will the tutorial be coming out anytime soon?