From 2cc14e48dccff7177d4a059a74a6f49a0cfb4bc6 Mon Sep 17 00:00:00 2001 From: Mustafa Karaki Date: Wed, 12 Jul 2017 06:59:21 +0200 Subject: [PATCH] Fix pycaffe import: Add missing __init__.py (#133) * Add missing __init__.py to fix pycaffe import. * Update __init__.py to last Caffe version --- 3rdparty/caffe/python/caffe/__init__.py | 8 ++++++++ 1 file changed, 8 insertions(+) create mode 100644 3rdparty/caffe/python/caffe/__init__.py diff --git a/3rdparty/caffe/python/caffe/__init__.py b/3rdparty/caffe/python/caffe/__init__.py new file mode 100644 index 00000000..776945ee --- /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 -- GitLab