如何用TensorRT实现ZF(c++)

xiaxuesong  

2018-01-09

目前在做行人识别的项目,之前python使用的是vgg16和ZF,但是最近使用c++和tensorrt进行加速,给的samples只有vgg的没有ZF,看英伟达官方论坛也有人问到,但是说是ROIpooling层需要自己实现?弱问一句有没有大神已经改成功了?如何改?

补充:英伟达推出了tensorrt加速推理速度,并给出了一系列samples,其中包含了fasterrcnn的prototxt,但是我仿照fastercnn的方法改造zf的prototxt文件遇到了一系列的问题,当我到英伟达官方查找相关内容时,当时英伟达官方给出的回答是ROIpooling层需要自己实现,我不知道如何实现,是否一定要自己实现这个层(能否用其他的层代替,如果可以之前的caffe model还能继续使用吗)?不知道有没有人自己实现过caffe 到tensort转化的工作,是怎么改造的?

目前强行改造遇到参数不匹配的问题,fc6 kernel weights has count 37748736 but 51380224 was excepted

关注问题 1人已关注
 写回答 0人已回答