CNN图像分类DEMO
更新时间:2019-06-14
CNN图像分类DEMO
为了便于用户开发,FPGA 云服务器包装了CNN图像分类demo:
目录结构
fpga-cnn
|----demo
|----env.sh
|----fpga_demo.cpp
|----Makefile
|----run_fpga.sh
|----run.sh
|----models
|----ResNet-50
|----ResNet-50.caffemodel
|----ResNet-50.labels
|----ResNet-50.means.binaryproto
|----ResNet-50.prototxt
|----ResNet-50.test.jpg
|----dependency/
|----include/
|----lib/
编译方法
在fpga-cnn/demo目录下运行:make clean && make
运行方法
-
脚本用CPU模式运行ResNet-50分类
在fpga-cnn/demo目录下运行:
sh run.sh
-
脚本用FPGA模式运行ResNet-50分类
在fpga-cnn/demo目录下运行:sh run_fpga.sh
-
通过fpga_demo运行
./fpga_demo mode case
其中,mode参数可以是cpu或者fpga,case参数为模型目录和文件名称