vgg_16_deploy.py 文件源码

def construc_net():
  net = caffe.NetSpec()

  net.data = L.Input(shape = dict(dim = [10,3,224,224]))

  block_1 = _block_crp('1', 2, net, net.data, 64)
  block_2 = _block_crp('2', 2, net, block_1,  128)
  block_3 = _block_crp('3', 3, net, block_2,  256)
  block_4 = _block_crp('4', 3, net, block_3,  512)
  block_5 = _block_crp('5', 3, net, block_4,  512)

  block_6 = _block_frd('6', net, block_5, 4096)
  block_7 = _block_frd('7', net, block_6, 4096)

  net.fc8 = L.InnerProduct(block_7, num_output = 1000)
  net.prob = L.Softmax(net.fc8)

  return net.to_proto()