diff --git a/3rdparty/caffe/python/caffe/__init__.py b/3rdparty/caffe/python/caffe/__init__.py new file mode 100644 index 0000000000000000000000000000000000000000..776945eec886cb52fb06eecbd9140e912c98c62d --- /dev/null +++ b/3rdparty/caffe/python/caffe/__init__.py @@ -0,0 +1,8 @@ +from .pycaffe import Net, SGDSolver, NesterovSolver, AdaGradSolver, RMSPropSolver, AdaDeltaSolver, AdamSolver, NCCL, Timer +from ._caffe import init_log, log, set_mode_cpu, set_mode_gpu, set_device, Layer, get_solver, layer_type_list, set_random_seed, solver_count, set_solver_count, solver_rank, set_solver_rank, set_multiprocess, has_nccl +from ._caffe import __version__ +from .proto.caffe_pb2 import TRAIN, TEST +from .classifier import Classifier +from .detector import Detector +from . import io +from .net_spec import layers, params, NetSpec, to_proto