__init__.py 434 B

12345678
  1. from .pycaffe import Net, SGDSolver, NesterovSolver, AdaGradSolver, RMSPropSolver, AdaDeltaSolver, AdamSolver
  2. from ._caffe import set_mode_cpu, set_mode_gpu, set_device, Layer, get_solver, layer_type_list, set_random_seed
  3. from ._caffe import __version__
  4. from .proto.caffe_pb2 import TRAIN, TEST
  5. from .classifier import Classifier
  6. from .detector import Detector
  7. from . import io
  8. from .net_spec import layers, params, NetSpec, to_proto