当前位置:   article > 正文

62、华为昇腾开发板Atlas 200I DK A2配置mmpose的hrnet模型推理python/c++_atlas200i推理模组

atlas200i推理模组

基本思想:适配mmpose模型,记录一下流水帐,环境配置和模型来自,请查看参考链接。

链接: https://pan.baidu.com/s/1IkiwuZf1anyKX1sZkYmD1g?pwd=i51s 提取码: i51s

一、转模型

  1. (base) root@davinci-mini:~/sxj731533730# atc --model=end2end.onnx --framework=5 --output=end2end --input_format=NCHW --input_shape="input:1,3,256,256" --log=error --soc_version=Ascend310B1
  2. ATC start working now, please wait for a moment.
  3. ...
  4. ATC run success, welcome to the next use.

python代码

  1. import time
  2. import cv2
  3. import numpy as np
  4. from ais_bench.infer.interface import InferSession
  5. model_path = "end2end.om"
  6. IMG_PATH = "ca110.jpeg"
  7. def bbox_xywh2cs(bbox, aspect_ratio, padding=1., pixel_std=200.):
  8. """Transform the bbox format from (x,y,w,h) into (center, scale)
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/weixin_40725706/article/detail/749747
推荐阅读
相关标签
  

闽ICP备14008679号