TensorRT-SSD icon indicating copy to clipboard operation
TensorRT-SSD copied to clipboard

Channel pruning SSD

Open GraphicsHunter opened this issue 7 years ago • 13 comments

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.

GraphicsHunter avatar Mar 09 '18 23:03 GraphicsHunter

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.

chenzhi1992 avatar Mar 12 '18 00:03 chenzhi1992

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?

GraphicsHunter avatar Mar 20 '18 21:03 GraphicsHunter

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

chenzhi1992 avatar Mar 22 '18 03:03 chenzhi1992

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?

GraphicsHunter avatar Mar 22 '18 23:03 GraphicsHunter

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

chenzhi1992 avatar Mar 25 '18 11:03 chenzhi1992

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

GraphicsHunter avatar Mar 25 '18 22:03 GraphicsHunter

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

GraphicsHunter avatar Apr 01 '18 19:04 GraphicsHunter

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

GraphicsHunter avatar Apr 15 '18 23:04 GraphicsHunter

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

chenzhi1992 avatar Apr 16 '18 00:04 chenzhi1992

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 avatar Apr 16 '18 09:04 GraphicsHunter

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 avatar Apr 21 '18 09:04 GraphicsHunter

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

GraphicsHunter avatar Apr 29 '18 14:04 GraphicsHunter

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

GraphicsHunter avatar May 06 '18 11:05 GraphicsHunter