PyTorch使用OpenFace模块的nn4.small2.v1.t7模型进行人脸识别

Song • 6666 次浏览 • 0 个回复 • 2017年12月03日

Openface是一个基于深度神经网络的开源人脸识别系统。

OpenFace for Pytorch

我做了一个高级的代码在PyTorch中使用OpenFace。我使用torch-hdf5nn4.small2.v1.t7`转换为.hdf5文件。然后我从.hdf5文件读取图层信息,可以显示如下:

nn.Sequential {
  [input -> (1) -> (2) -> (3) -> (4) -> (5) -> (6) -> (7) -> (8) -> (9) -> (10) -> (11) -> (12) -> (13) -> (14) -> (15) -> (16) -> (17) -> (18) -> (19) -> (20) -> (21) -> (22) -> (23) -> (24) -> (25) -> (26) -> output]
  (1): nn.SpatialConvolutionMM(3 -> 64, 7x7, 2,2, 3,3)
  (2): nn.SpatialBatchNormalization (4D) (64)
  (3): nn.ReLU
  (4): nn.SpatialMaxPooling(3x3, 2,2, 1,1)
  (5): nn.SpatialCrossMapLRN
...

然后我用PyTorchPyTorch的nn.legacy采用的SpatialCrossMapLRN_temp.py)支持的PyTorch(参见loadOpenFace.py)手动编码图层,最后的模型是openface.pth(可能需要重命名为openface_nn4_small2_v1.pth),它可以通过loadOpenFace.py中的代码加载。

请参阅loadOpenFace.py的主要部分了解如何使用。只是,

net = prepareOpenFace(useCuda=True, gpuDevice=0, useMultiGPU=False).eval()
feature = net(input_tensor)    # input_tensor should be (batch_size, 3, 96, 96)

项目地址:OpenFace for Pytorch


原创文章,转载请注明 :PyTorch使用OpenFace模块的nn4.small2.v1.t7模型进行人脸识别 - pytorch中文网
原文出处: https://www.ptorch.com/news/88.html
问题交流群 :168117787
提交评论
要回复文章请先登录注册
用户评论
  • 没有评论
Pytorch是什么?关于Pytorch! PyTorch中的概率规划和统计推断PtStat