使用rknn进行facenet部署

文章目录

  • 开源仓库
  • pth转onnx
  • netron可视化
  • onnx转rknn
  • C++实现

开源仓库

https://github.com/bubbliiiing/facenet-pytorch

pth转onnx

修改facenet网络的forward函数代码
修改前

def forward(self, x, mode = "predict"):
        if mode == 

你可能感兴趣的:(深度学习模型部署,rknn,人脸识别)