MDK - Caffe

Ubuntu 16.04环境配置

所需硬件和软件的要求

所需硬件设备:

  • Linux(Ubuntu 16.04,64 位版本)
  • Intel i5 或更好的处理器(i7 首选)3.0 GHz 或更快
  • 8 GB 或更多 RAM
  • 至少一个 USB 3.0 端口

第三方软件:

  • Python 2.7 或 3.5
  • OpenCV 3.2.0
  • GNU 编译器(最低要求 g ++版本 4.9,首选 5.4)
  • Caffe 1.0
  • Cuda 8.0 和 CUDNN 6
  • GTI SDK 4.2

环境搭建

具体步骤可参考MDK - SSD - 环境安装。

与MDK - SSD中不同的是使用的代码不同。此版本的MDK使用的源码是https://github.com/BVLC/caffe中的v1.0源码，加上一些修改。具体修改在下文描述。

在caffe中添加2801的支持

根据下表中列出的说明更新现有的Caffe 1.0框架。所有必需的文件都位于caffe-installation目录中。

_images/mdk-caffe-replacement-table.png

如果替换caffe.proto后，因caffe.proto编译不通过请使用合并的方式修改caffe.proto或者使用GTIcaffeMDKv3.0.0.1中提供的caffe源码。

模型训练流程说明

具体步骤请参考GTI Caffe MDK User Guide v3.0.0.1_CN.pdf

注：相关资料及源码可从AI资料U盘中获取，如未购买请先购买，已购买过的但资料不是最新的可联系客服获取最新的资料。