search

chenzhi1992 / TensorRT-SSD

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

template <int OutC>. #22

Open quocbh opened 6 years ago

quocbh commented 6 years ago

Is OutC number class? How to pass num class to OutC

chenzhi1992 commented 6 years ago

template class Reshape : public IPlugin ==> Reshape<21>

quocbh commented 6 years ago

Thanks. i see output of Reshape plugin is (21,8732,1). But in ssd, shape is (0,-1,21). Can u explain for me?

chenzhi1992 commented 6 years ago

You can print the output shape of all layers, including plugin and ssd. Their output shape is a little different.