关于 Huabot Brain

Huabot Brain 可以说是 caffe 的 GUI 工具。 Huabot Brain 提供 restful api 和一个基于 react 的 web 页面,使得开发者或者研究人员可以 方便的使用 caffe,查看训练状态,提供 demo 和应用。 可作为一个嵌入式的子系统,通过简单的 restful api 给原有的系统提供深度学习的能力。

来几张截图:

1、训练曲线

训练曲线

2、数据页面

3、Demo 页面

安装 Huabot Brain

# 设置 GOPATH
export GOPATH=$HOME/gopath
export PATH=$GOPATH/bin:$PATH

# 安装 Huabot Brain
go get -v github.com/Lupino/huabot-brain
# 安装 Train worker
go get -v github.com/Lupino/huabot-brain/tools/train_worker

启动 Huabot Brain

# 启动 Gearman job server
gearmand -d

# 启动 Huabot Brain (默认端口 3000)
cd $GOPATH/src/github.com/Lupino/huabot-brain
huabot-brain --gearmand=127.0.0.1:4730 --dbpath=dataset.db

# 启动 Train worker
train_worker --gearmand=127.0.0.1:4730

# 启动 Predict Worker (这一步要等训练一端时间或完成后,才可执行, xxx 为数字)
ln -s resourses/models/huabot-brain_iter_xxx.caffemodel resourses/models/huabot-brain.caffemodel
env GEARMAND_PORT=tcp://127.0.0.1:4730 python tools/predict_worker/main.py resources

提交数据

# 用 curl 提交数据
curl -XPOST -F file=@/path/to/image.png -F tag=xxx -F data_type=1 http://127.0.0.1:3000/api/datasets

# 用脚本提交数据
cd tools/datasets
python get_datasets.py

自己的数据集请使用 API 提交

训练 Huabot Brain

打开 Dashboard 点击 Train 就可以了。 也可以用命令

curl http://127.0.0.1:3000/api/train

详细信息请见 API 文档

预测图片的信息

但这里需要启动 predict worker。

通过 DEMO 页面,在这里面贴入一个图片的地址,回车就可以。

也可以用 API:

curl -d img_url=http://image http://127.0.0.1:3000/api/predict

结束语

就此饱览了 Huabot Brain 的盛宴。