CNN图像分类DEMO说明
更新时间:2019-06-14
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
运行方法
1.脚本用CPU模式运行ResNet-50分类
在fpga-cnn/demo目录下运行:sh run.sh
2.脚本用FPGA模式运行ResNet-50分类
在fpga-cnn/demo目录下运行:sh run_fpga.sh
3.通过fpga_demo运行
./fpga_demo mode case
其中,mode参数可以是cpu或者fpga,case参数为模型目录和文件名称