ssd pytorch转 libtorch c++实现 (nms采用cuda并行实现)
背景:
因为所里面大多数同学在研究论文时用的比较多的还是pytorch,所以考虑在pc端的部署采用libtorch ,当然
libtorch比较新,所以还是要谨慎地采用,因为看到官方只用resnet等简单的网络做例子时就要有被坑的心里准备,但是,,项目紧急,不管了。。
只要懂得其中一些规则,尤其是jit 的规则还是挺好弄的,但是本人也是第一次真正接触到jit,所以也踩了很多坑
如果熟悉了里...
原创
2019-11-04 17:40:37 ·
3231 阅读 ·
63 评论