face_net模型转换为libtorch的心得

发布时间 2023-09-19 10:01:19作者: 澳大利亚树袋熊
Facenet demo:https://www.cnblogs.com/muyisun/p/13338098.html

如果要转为libtorch可调用,应该使用torch.jit中的script或trace。

script会保留一些控制流,trace则会擦除。可以使用.code查看过程。

torch_trace_model = torch.jit.script(model, input)#script trace
print(torch_trace_model.code)

但是无论如何,pytorch都无法转第三方库。

https://zhuanlan.zhihu.com/p/96397421

 

总而言之,经过调研,facenet无法转为C++下可调用的libtorch模型,因为其包含了众多控制流和第三方库。

这给了我们一个启示:在设计网络模型时,应当只包含一些简单的控制流或仅输出模型,不应当有第三库参与。