AllentDan/LibtorchSegmentation

请问可以增加solov2的教程吗，球球了

gogostudy-daydayup opened this issue · 2 comments

gogostudy-daydayup commented

非常感谢大佬开源这个项目，这对于想在C++上利用libtorch部署的人的帮助真的很大。
最近在研究solov2，但网上很多资料都是基于python的，而关于在C++上部署的非常少，不知道大佬可不可以出个相关的教程指导一下？可能这要求有点过分啦哈哈哈，但还是希望大佬能考虑一下

AllentDan commented

Hi, @gogostudy-daydayup. 这个可以参考LibtorchTutorials，找个自己心仪的repo复现到C++上，应该不是特别难。

但是里面涉及到NMS，Libtorch的API复现NMS的话，可能速度上不够好。对于双阶段的网络，NMS的框很多的话，速度比numpy实现要快，但是不会比torchvision调用cuda快，还是要直接用cuda实现的速度会好很多。不确定对solov2影响怎么样

有时间的话我会考虑做下实例分割网络的

gogostudy-daydayup commented