NCNN部署例程 mxnet-gluoncv之simple_pose

叁歲伎倆 2023-02-28 09:00 36阅读 0赞

NCNN在官方源码中有一个例程，simple pose，今天试了一下效果。

整体来说，优化得太过，效果不行。

过程如下。

首先，在gluoncv中转换生成模型pose.param, pose.bin，python 源码如下，

其中，用  
export\_block('simple\_pose\_resnet18\_v1b', net, preprocess=True, layout='HWC')  
转换出来的不能成功部署到NCNN上，所以在python源码中我注释掉了。

# this one works OK for TestFcn_method2-symbol
    import mxnet as mx
    import gluoncv as gcv
    from gluoncv.utils import export_block
    
    net = gcv.model_zoo.get_model('simple_pose_resnet18_v1b', pretrained=True)
    #pose_net = model_zoo.get_model('simple_pose_resnet18_v1b', pretrained=True)
    net.hybridize()
    
    #try & uncomment the below code to generate:
    #it seems the below export method is NOT OK for NCNN
    #export_block('simple_pose_resnet18_v1b', net, preprocess=True, layout='HWC')
    
    #the below method is OK for NCNN
    data_shape = (1, 3, 512, 512)
    input_data = mx.nd.random.uniform(-1, 1, data_shape)
    _ = net(input_data)
    net.export('pose')
    
    # uncomment the below code to display
    print(net)
    
    print('Done.')
    
    # have a view of what has been done!
    import glob
    print(glob.glob('*.json') + glob.glob('*.params'))
    print('All Done.')

第二步

然后，用下面的命令来生成ncnn.param和ncnn.bin，生成后把名字改成pose.param, pose.bin。  
mxnet2ncnn pose-symbol.json pose-0000.params

第三步

此时，你就可以用ncnn中的simplepose来进行测试了，例如

simplepose COCO\_val2014\_000000324595.jpg

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3Rhbm14MjE5_size_16_color_FFFFFF_t_70][]

我直接用gluoncv测试的效果是这样的，

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3Rhbm14MjE5_size_16_color_FFFFFF_t_70 1][]

由于目前对NCNN的源码还不熟悉，所以也不清楚到底哪里出了问题。

对于NCNN的其他模型，作者介绍得最详细的我只找到这个，按照作者介绍的步骤，也成功部署了。

《详细记录YOLACT实例分割ncnn实现》  
[https://zhuanlan.zhihu.com/p/128974102][https_zhuanlan.zhihu.com_p_128974102]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3Rhbm14MjE5_size_16_color_FFFFFF_t_70]: /images/20230209/b9db3a6cb3bf442f9d967f8d182d1103.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3Rhbm14MjE5_size_16_color_FFFFFF_t_70 1]: /images/20230209/4f79aac4f56247bca397e1fbe56bc920.png
[https_zhuanlan.zhihu.com_p_128974102]: https://zhuanlan.zhihu.com/p/128974102