当前位置:   article > 正文

Vitis AI 运行TensforFlow模型_vitis调用人工智能模型

vitis调用人工智能模型

在百度edgeboard fzu3上运行CIFAR10 Classification,介绍Vitis AI TensorFlow设计过程,将Python描述的网络模型运行在Xilinx DPU上。

CIFAR_10数据集

输入图片32x32x8 RGB images,完整CIFAR数据集有60k图片,将数据集进行划分,50k进行训练,10k用来验证。

DenseNet结构

DenseNet-121组成:

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-zxWB5HX7-1630813493654)(./pic/densenet.png)]

注:上图针对ImageNet。

步骤

  • 拷贝文件
cp -r /mnt/hgfs/ubuntu/DenseNetX/ ~/
  • 1
  • 下载模型

Vitis AI 1.2版本模型不能下载,使用Vitis AI1.3版本的模型,解压模型keras_model.zip,将k_model.h5拷贝到./files/build/keras_model 文件夹。

    声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/我家小花儿/article/detail/490021?site
    推荐阅读
    相关标签
      

    闽ICP备14008679号