SCUT-AILab / DCP

Code for “Discrimination-aware-Channel-Pruning-for-Deep-Neural-Networks”
BSD 3-Clause "New" or "Revised" License
184 stars 27 forks source link

channel pruning on detection task ? #14

Open hanfanggithub opened 5 years ago

hanfanggithub commented 5 years ago

I think the additional discrimination-aware loss in this project is suitable for classification task but not detection task ,the classification task focus on the whole feature map , therefore averagepooling+softmax works , but the detection task focus on feature in receptive field , if I want to use this method on yolo-v3 , how can I modify the discrimination-aware loss ?