models.py 文件源码

def create_models(buckets = [25, 50, 100], phase = caffe.TRAIN):

  transformers = create_spatial_transformers(buckets, phase)
  proposal_net = create_yolo(phase)
  recog = create_recognizer(phase)

  return proposal_net, transformers, recog