caffe编写简单示例(C++,CMAKE)

看caffe相关书,在Blob部分可以尝试Blob的简单用法,我是在ubuntu上使用cmake尝试,使用的IDE是KDevelop。

其中cmakelists文件:

# 这里的项目名称为test_1,主文件为main.cpp
cmake_minimum_required(VERSION 2.6)
project(test_1)

set(Caffe_INCLUDE_DIRS /caffe所存储的相关地址/caffe/include /usr/local/cuda/include /caffe所存储的相关地址/caffe/build/src)
set(Caffe_LIBRARIES caffe boost_system glog)

include_directories(${Caffe_INCLUDE_DIRS})
link_directories(/caffe所存储的相关地址/caffe/build/lib)

add_executable(test_1 main.cpp)
target_link_libraries(test_1 ${Caffe_LIBRARIES})
    
install(TARGETS test_1 RUNTIME DESTINATION bin)

main.cpp

#include <iostream>
#include <vector>
#include <caffe/blob.hpp>
using namespace caffe;
using namespace std;
int main(void) {
    Blob<float> a;
    cout << "Size : " << a.shape_string() << endl;
    a.Reshape(1.0, 2.0, 3.0, 4.0);
    cout << "Size : " << a.shape_string() << endl;
    return 0;
}

输出结果

Size : (0)
Size : 1 2 3 4 (24)
*** Exited normally ***

参考:https://blog.csdn.net/mogoweb/article/details/79784720

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值