手动通知caffe快照操作等小技巧

caffe支持接受SIGHUP信号进行快照操作,所以命令就是:

# PID是caffe的进程号
kill -SIGHUP PID

# 查看PID的命令
ps -ef | grep caffe

# 或者使用pidof命令,这次一步完成
kill -SIGHUP $(pidof caffe)

使用caffe的solverstate继续训练的命令格式

./build/tools/caffe train --solver=solver.prototxt --snapshot=caffenet_train_iter_10000.solverstate

最简单caffe模型测试命令

import caffe
import numpy as np
import cv2
 
# 把图读进来解码
im = cv2.imread('000001.jpg',cv2.IMREAD_COLOR)
# 把图resize了 并转成网络input的shape大小
imblob=np.transpose(cv2.resize(im,(224,224)).reshape((1,224,224,3)),(0,3,1,2))
 
n = caffe.Net('deploy.prototxt','vgg16_iter_20000.caffemodel',caffe.TEST)
n.forward(data=imblob.astype(np.float32))

# 在直接取出输出层的结果即可,如
print n['cls_prob']

发表评论

电子邮件地址不会被公开。 必填项已用*标注