第一个caffe案例

原创

櫻木 2022-11-10 14:35:17 博主文章分类：caffe ©著作权

©著作权归作者所有：来自51CTO博客作者櫻木的原创作品，请联系作者获取转载授权，否则将追究法律责任

caffe案例

import caffe
from caffe import layers as L
from caffe import params as P 


def net(dbfile,batch_size,mean_value=0):
  d = caffe.NetSpec()
  d.data, d.label=L.Data(source=dbfile, backend = P.Data.LMDB, batch_size=batch_size, ntop=2, transform_param=dict(scale=0.00390625))
  d.ip1 = L.InnerProduct(d.data, num_output=500, weight_filler=dict(type='xavier'))
  d.relu1 = L.ReLU(d.ip1, in_place=True)
  d.ip2 = L.InnerProduct(d.relu1, num_output=10, weight_filler=dict(type='xavier'))
  d.loss = L.SoftmaxWithLoss(d.ip2, d.label)
  d.accu = L.Accuracy(d.ip2, d.label, include={'phase':caffe.TEST})
  return d.to_proto()
  
def main():
  with open( 'auto_train00.prototxt', 'w') as f:
    f.write(str(net( 'mnist/mnist_train_lmdb', 64)))
  with open('auto_test00.prototxt', 'w') as f:
    f.write(str(net('mnist/mnist_test_lmdb', 100)))
  solver = caffe.SGDSolver("hbk_mnist_solver_py.prototxt")
  solver.net.forward()
  solver.test_nets[0].forward()
  solver.step(1)

if __name__=="__main__":
  main()