Caffe2(2)----Eclipse环境中使用Caffe2

使用IDE开发深度学习的应用，可以事半功倍，Caffe2已经全面支持Python，这里介绍如何在Ubantu14.04下，利用EclipseCaffe2的二次开发或应用。

1.安装eclipse

具体请参考：ROS知识（6）----基于Eclipse开发

2.安装eclipse的插件PyDev

具体请参阅：几个常用的Eclipse插件

3建立PyDev新项目

a.点击"File->New->Other..."，弹出对话框，选择“PyDev Project”，如下图所示：

b.选择下一步，弹出如下对话框，在“Project Name”中输入项目名称，然后点击"Finish"。

c.设置caffe2的环境依赖

鼠标右击项目，选择“property”，弹出如下对话框，并选择“PyDev-Interpreter/Grammar”

d.点击“New Folder”添加两个Python的路径，分别为：

/usr/local/caffe2/python

/usr/local

e.添加后如下图所示，点击”Apply“，然后点击”OK“退出。

f.重启eclipse，重新导入项目。

4.测试Caffe2

a.在项目的src下添加python文件test1.py，然后添加如下代码

from caffe2.python import workspace, model_helper,cnn

import numpy as np

data_dim = 1000

label_dim = 10

print("Build Net")

# Create the input data

data = np.random.rand(label_dim, data_dim).astype(np.float32)

# Create labels for the data as integers [0, 9].

label = (np.random.rand(label_dim) * 10).astype(np.int32)

workspace.FeedBlob("data", data)

workspace.FeedBlob("label", label)

# Create model using a model helper

m = cnn.CNNModelHelper(name="my first net")

fc_1 = m.FC("data", "fc1", dim_in=data_dim, dim_out=10)

pred = m.Sigmoid(fc_1, "pred")

[softmax, loss] = m.SoftmaxWithLoss([pred, "label"], ["softmax", "loss"])

m.AddGradientOperators([loss])

workspace.RunNetOnce(m.param_init_net)

workspace.CreateNet(m.net)

# Run 100 x 10 iterations

print("Start to Train")

for j in range(0, 10):

    data = np.random.rand(label_dim, data_dim).astype(np.float32)

    label = (np.random.rand(label_dim) * 10).astype(np.int32)

    workspace.FeedBlob("data", data)

    workspace.FeedBlob("label", label)

    workspace.RunNet(m.name, 10)   # run for 10 times

    print("%d:%f"%(j, workspace.FetchBlob("loss")))

print("End Training")

b.点击工具栏的运行或调试按钮，如果一切ok，会输出一下的迭代信息：

Build Net

Start to Train

0:2.324265

1:2.382756

2:2.357759

3:2.298997

4:2.331089

5:2.285640

6:2.336110

7:2.274807

8:2.366172

9:2.337986

End Training

5.代码

下载：caffe2learning