diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index 21bab1f7baea919e7548df5adbf4f312c7dacc75..236e319650e23972cac47ae2cad3cfc17a2fcecf 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -1,12 +1,8 @@ -- repo: local +- repo: https://github.com/PaddlePaddle/mirrors-yapf.git + sha: 0d79c0c469bab64f7229c9aca2b1186ef47f0e37 hooks: - id: yapf - name: yapf - entry: yapf - language: system - args: [-i, --style .style.yapf] files: \.py$ - - repo: https://github.com/pre-commit/pre-commit-hooks sha: a11d9314b22d8f8c7556443875b731ef05965464 hooks: @@ -18,6 +14,7 @@ - id: check-symlinks - id: check-added-large-files - repo: local + hooks: - id: copyright_checker name: copyright_checker diff --git a/.travis.yml b/.travis.yml index 7f181a905b9ae093bc183890ac3ef0b167fd0cdf..9c4eaf76656721581a71782743faf70ddba45f66 100644 --- a/.travis.yml +++ b/.travis.yml @@ -1,9 +1,7 @@ language: python python: - - '2.7' - '3.5' - - '3.6' script: - if [[ $TRAVIS_PYTHON_VERSION != 2.7 ]]; then /bin/bash ./tools/check_code_style.sh; fi diff --git a/setup.py b/setup.py index 51708204e15eac14a9bc992720bbbe63e32f2fda..490c42c95970cdd333de84e458eef910342076f1 100644 --- a/setup.py +++ b/setup.py @@ -11,8 +11,7 @@ setuptools.setup( version=x2paddle.__version__, author="dltp-sz", author_email="dltp-sz@baidu.com", - description= - "a toolkit for converting trained model to PaddlePaddle from other deep learning frameworks.", + description="a toolkit for converting trained model to PaddlePaddle from other deep learning frameworks.", long_description=long_description, long_description_content_type="text/plain", url="https://github.com/PaddlePaddle/x2paddle", @@ -23,6 +22,4 @@ setuptools.setup( "Operating System :: OS Independent", ], license='Apache 2.0', - entry_points={'console_scripts': [ - 'x2paddle=x2paddle.convert:main', - ]}) + entry_points={'console_scripts': ['x2paddle=x2paddle.convert:main', ]}) diff --git a/tools/merge_params.py b/tools/merge_params.py index e6a8a3d2a1d338cf64ceac9a9b1983d180334a58..1b2fa18df5153edfdcac012e3b84b1893803b0a5 100644 --- a/tools/merge_params.py +++ b/tools/merge_params.py @@ -5,12 +5,14 @@ model_dir = sys.argv[1] new_model_dir = sys.argv[2] exe = fluid.Executor(fluid.CPUPlace()) [inference_program, feed_target_names, - fetch_targets] = fluid.io.load_inference_model(dirname=model_dir, executor=exe) + fetch_targets] = fluid.io.load_inference_model( + dirname=model_dir, executor=exe) print(feed_target_names) -fluid.io.save_inference_model(dirname=new_model_dir, - feeded_var_names=feed_target_names, - target_vars=fetch_targets, - executor=exe, - main_program=inference_program, - params_filename="__params__") +fluid.io.save_inference_model( + dirname=new_model_dir, + feeded_var_names=feed_target_names, + target_vars=fetch_targets, + executor=exe, + main_program=inference_program, + params_filename="__params__") diff --git a/x2paddle/convert.py b/x2paddle/convert.py index efd665732e6b278fb862fc56df02b548eb7cdced..28a303bda9ea81899f03419dc9f538a5e5967a3d 100644 --- a/x2paddle/convert.py +++ b/x2paddle/convert.py @@ -48,8 +48,7 @@ def arg_parser(): "-f", type=_text_type, default=None, - help= - "define which deeplearning framework(tensorflow/caffe/onnx/paddle2onnx)" + help="define which deeplearning framework(tensorflow/caffe/onnx/paddle2onnx)" ) parser.add_argument( "--caffe_proto", @@ -126,7 +125,6 @@ def tf2paddle(model_path, optimizer.merge_bias() optimizer.optimize_sub_graph() - # optimizer.merge_batch_norm() # optimizer.merge_prelu() else: diff --git a/x2paddle/core/fluid_code.py b/x2paddle/core/fluid_code.py index 6f4df3fc89d25fea6fe9aeb9c46c77f7b30f4e68..72ea10788e9519f8ec4256c13220969b3a0b4959 100644 --- a/x2paddle/core/fluid_code.py +++ b/x2paddle/core/fluid_code.py @@ -46,8 +46,9 @@ class Layer(object): for input in self.inputs: if isinstance(input, GraphNode): if hasattr(input, "index"): - in_list += (input.layer_name + - "[{}]".format(input.index) + ", ") + in_list += ( + input.layer_name + "[{}]".format(input.index) + ", " + ) else: in_list += (input.layer_name + ", ") elif isinstance(input, six.string_types): @@ -71,8 +72,8 @@ class Layer(object): layer_code = layer_code + key + "={}, ".format(input) elif isinstance(self.inputs, GraphNode): if hasattr(self.inputs, "index"): - layer_code += (self.inputs.layer_name + - "[{}]".format(self.inputs.index)) + layer_code += ( + self.inputs.layer_name + "[{}]".format(self.inputs.index)) else: layer_code += (self.inputs.layer_name) if self.op != "=": diff --git a/x2paddle/core/op_mapper.py b/x2paddle/core/op_mapper.py index 77876bc6db485d91d66aa125d2efb30828e06bc0..d6649a54dc2cde78054781dc5b4b00b2b8522e61 100644 --- a/x2paddle/core/op_mapper.py +++ b/x2paddle/core/op_mapper.py @@ -64,10 +64,8 @@ def run_net(param_dir="./"): b = os.path.exists(os.path.join(param_dir, var.name)) return b - fluid.io.load_vars(exe, - param_dir, - fluid.default_main_program(), - predicate=if_exist) + fluid.io.load_vars( + exe, param_dir, fluid.default_main_program(), predicate=if_exist) class OpMapper(object): @@ -98,8 +96,8 @@ class OpMapper(object): def add_codes(self, codes, indent=0): if isinstance(codes, list): for code in codes: - self.paddle_codes += (self.tab * indent + code.strip('\n') + - '\n') + self.paddle_codes += ( + self.tab * indent + code.strip('\n') + '\n') elif isinstance(codes, str): self.paddle_codes += (self.tab * indent + codes.strip('\n') + '\n') else: @@ -135,24 +133,25 @@ class OpMapper(object): os.path.join(os.path.join(py_code_dir, var.name))) return b - fluid.io.load_vars(exe, - py_code_dir, - fluid.default_main_program(), - predicate=if_exist) + fluid.io.load_vars( + exe, + py_code_dir, + fluid.default_main_program(), + predicate=if_exist) if params_merge: - fluid.io.save_inference_model(dirname=os.path.join( - save_dir, "inference_model"), - feeded_var_names=input_names, - target_vars=outputs, - executor=exe, - params_filename="__params__") + fluid.io.save_inference_model( + dirname=os.path.join(save_dir, "inference_model"), + feeded_var_names=input_names, + target_vars=outputs, + executor=exe, + params_filename="__params__") else: - fluid.io.save_inference_model(dirname=os.path.join( - save_dir, "inference_model"), - feeded_var_names=input_names, - target_vars=outputs, - executor=exe, - params_filename=None) + fluid.io.save_inference_model( + dirname=os.path.join(save_dir, "inference_model"), + feeded_var_names=input_names, + target_vars=outputs, + executor=exe, + params_filename=None) except: raise Exception( "Paddle code was saved in {}/model.py, but seems there's wrong exist, please check model.py manually." diff --git a/x2paddle/decoder/caffe_decoder.py b/x2paddle/decoder/caffe_decoder.py index 28df01739b24293ba29b580dd652f274598d9ef1..cb9f59708fa0980b0a3bcb029319a33a3df18f92 100644 --- a/x2paddle/decoder/caffe_decoder.py +++ b/x2paddle/decoder/caffe_decoder.py @@ -49,13 +49,11 @@ class CaffeResolver(object): class CaffeGraphNode(GraphNode): def __init__(self, layer, type_str, layer_name=None): if layer_name is None: - super(CaffeGraphNode, - self).__init__(layer, - layer.name.replace('/', '_').replace('-', '_')) + super(CaffeGraphNode, self).__init__( + layer, layer.name.replace('/', '_').replace('-', '_')) else: - super(CaffeGraphNode, - self).__init__(layer, - layer_name.replace('/', '_').replace('-', '_')) + super(CaffeGraphNode, self).__init__( + layer, layer_name.replace('/', '_').replace('-', '_')) self.layer_type = type_str self.fluid_code = FluidCode() self.data = None @@ -268,8 +266,8 @@ class CaffeDecoder(object): c_i = blob.channels h = blob.height w = blob.width - data = np.asarray(list(blob.data), - dtype=np.float32).reshape(c_o, c_i, h, w) + data = np.asarray( + list(blob.data), dtype=np.float32).reshape(c_o, c_i, h, w) transformed.append(data) return transformed diff --git a/x2paddle/decoder/caffe_pb2.py b/x2paddle/decoder/caffe_pb2.py index 4db5a9f1ee86bc0aaaaf051041d4526d85b75ce6..5058c4c2bff8a8b1eab838c1b01780eb3102426b 100644 --- a/x2paddle/decoder/caffe_pb2.py +++ b/x2paddle/decoder/caffe_pb2.py @@ -27,22 +27,16 @@ _PHASE = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='TRAIN', - index=0, - number=0, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='TEST', - index=1, - number=1, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='TRAIN', index=0, number=0, serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='TEST', index=1, number=1, serialized_options=None, type=None), ], containing_type=None, serialized_options=None, serialized_start=21908, - serialized_end=21936, -) + serialized_end=21936, ) _sym_db.RegisterEnumDescriptor(_PHASE) Phase = enum_type_wrapper.EnumTypeWrapper(_PHASE) @@ -55,22 +49,23 @@ _EMITCONSTRAINT_EMITTYPE = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='CENTER', - index=0, - number=0, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='MIN_OVERLAP', - index=1, - number=1, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='CENTER', + index=0, + number=0, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='MIN_OVERLAP', + index=1, + number=1, + serialized_options=None, + type=None), ], containing_type=None, serialized_options=None, serialized_start=1146, - serialized_end=1185, -) + serialized_end=1185, ) _sym_db.RegisterEnumDescriptor(_EMITCONSTRAINT_EMITTYPE) _ANNOTATEDDATUM_ANNOTATIONTYPE = _descriptor.EnumDescriptor( @@ -79,17 +74,13 @@ _ANNOTATEDDATUM_ANNOTATIONTYPE = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='BBOX', - index=0, - number=0, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='BBOX', index=0, number=0, serialized_options=None, type=None), ], containing_type=None, serialized_options=None, serialized_start=1629, - serialized_end=1655, -) + serialized_end=1655, ) _sym_db.RegisterEnumDescriptor(_ANNOTATEDDATUM_ANNOTATIONTYPE) _FILLERPARAMETER_VARIANCENORM = _descriptor.EnumDescriptor( @@ -98,27 +89,29 @@ _FILLERPARAMETER_VARIANCENORM = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='FAN_IN', - index=0, - number=0, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='FAN_OUT', - index=1, - number=1, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='AVERAGE', - index=2, - number=2, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='FAN_IN', + index=0, + number=0, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='FAN_OUT', + index=1, + number=1, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='AVERAGE', + index=2, + number=2, + serialized_options=None, + type=None), ], containing_type=None, serialized_options=None, serialized_start=1872, - serialized_end=1924, -) + serialized_end=1924, ) _sym_db.RegisterEnumDescriptor(_FILLERPARAMETER_VARIANCENORM) _SOLVERPARAMETER_SNAPSHOTFORMAT = _descriptor.EnumDescriptor( @@ -127,22 +120,19 @@ _SOLVERPARAMETER_SNAPSHOTFORMAT = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='HDF5', - index=0, - number=0, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='BINARYPROTO', - index=1, - number=1, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='HDF5', index=0, number=0, serialized_options=None, type=None), + _descriptor.EnumValueDescriptor( + name='BINARYPROTO', + index=1, + number=1, + serialized_options=None, + type=None), ], containing_type=None, serialized_options=None, serialized_start=3480, - serialized_end=3523, -) + serialized_end=3523, ) _sym_db.RegisterEnumDescriptor(_SOLVERPARAMETER_SNAPSHOTFORMAT) _SOLVERPARAMETER_SOLVERMODE = _descriptor.EnumDescriptor( @@ -151,22 +141,15 @@ _SOLVERPARAMETER_SOLVERMODE = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='CPU', - index=0, - number=0, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='GPU', - index=1, - number=1, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='CPU', index=0, number=0, serialized_options=None, type=None), + _descriptor.EnumValueDescriptor( + name='GPU', index=1, number=1, serialized_options=None, type=None), ], containing_type=None, serialized_options=None, serialized_start=3525, - serialized_end=3555, -) + serialized_end=3555, ) _sym_db.RegisterEnumDescriptor(_SOLVERPARAMETER_SOLVERMODE) _SOLVERPARAMETER_SOLVERTYPE = _descriptor.EnumDescriptor( @@ -175,42 +158,39 @@ _SOLVERPARAMETER_SOLVERTYPE = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='SGD', - index=0, - number=0, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='NESTEROV', - index=1, - number=1, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='ADAGRAD', - index=2, - number=2, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='RMSPROP', - index=3, - number=3, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='ADADELTA', - index=4, - number=4, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='ADAM', - index=5, - number=5, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='SGD', index=0, number=0, serialized_options=None, type=None), + _descriptor.EnumValueDescriptor( + name='NESTEROV', + index=1, + number=1, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='ADAGRAD', + index=2, + number=2, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='RMSPROP', + index=3, + number=3, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='ADADELTA', + index=4, + number=4, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='ADAM', index=5, number=5, serialized_options=None, type=None), ], containing_type=None, serialized_options=None, serialized_start=3557, - serialized_end=3642, -) + serialized_end=3642, ) _sym_db.RegisterEnumDescriptor(_SOLVERPARAMETER_SOLVERTYPE) _PARAMSPEC_DIMCHECKMODE = _descriptor.EnumDescriptor( @@ -219,22 +199,23 @@ _PARAMSPEC_DIMCHECKMODE = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='STRICT', - index=0, - number=0, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='PERMISSIVE', - index=1, - number=1, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='STRICT', + index=0, + number=0, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='PERMISSIVE', + index=1, + number=1, + serialized_options=None, + type=None), ], containing_type=None, serialized_options=None, serialized_start=4131, - serialized_end=4173, -) + serialized_end=4173, ) _sym_db.RegisterEnumDescriptor(_PARAMSPEC_DIMCHECKMODE) _RESIZEPARAMETER_RESIZE_MODE = _descriptor.EnumDescriptor( @@ -243,27 +224,25 @@ _RESIZEPARAMETER_RESIZE_MODE = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='WARP', - index=0, - number=1, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='FIT_SMALL_SIZE', - index=1, - number=2, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='FIT_LARGE_SIZE_AND_PAD', - index=2, - number=3, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='WARP', index=0, number=1, serialized_options=None, type=None), + _descriptor.EnumValueDescriptor( + name='FIT_SMALL_SIZE', + index=1, + number=2, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='FIT_LARGE_SIZE_AND_PAD', + index=2, + number=3, + serialized_options=None, + type=None), ], containing_type=None, serialized_options=None, serialized_start=8049, - serialized_end=8120, -) + serialized_end=8120, ) _sym_db.RegisterEnumDescriptor(_RESIZEPARAMETER_RESIZE_MODE) _RESIZEPARAMETER_PAD_MODE = _descriptor.EnumDescriptor( @@ -272,27 +251,29 @@ _RESIZEPARAMETER_PAD_MODE = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='CONSTANT', - index=0, - number=1, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='MIRRORED', - index=1, - number=2, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='REPEAT_NEAREST', - index=2, - number=3, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='CONSTANT', + index=0, + number=1, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='MIRRORED', + index=1, + number=2, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='REPEAT_NEAREST', + index=2, + number=3, + serialized_options=None, + type=None), ], containing_type=None, serialized_options=None, serialized_start=8122, - serialized_end=8180, -) + serialized_end=8180, ) _sym_db.RegisterEnumDescriptor(_RESIZEPARAMETER_PAD_MODE) _RESIZEPARAMETER_INTERP_MODE = _descriptor.EnumDescriptor( @@ -301,37 +282,34 @@ _RESIZEPARAMETER_INTERP_MODE = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='LINEAR', - index=0, - number=1, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='AREA', - index=1, - number=2, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='NEAREST', - index=2, - number=3, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='CUBIC', - index=3, - number=4, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='LANCZOS4', - index=4, - number=5, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='LINEAR', + index=0, + number=1, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='AREA', index=1, number=2, serialized_options=None, type=None), + _descriptor.EnumValueDescriptor( + name='NEAREST', + index=2, + number=3, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='CUBIC', index=3, number=4, serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='LANCZOS4', + index=4, + number=5, + serialized_options=None, + type=None), ], containing_type=None, serialized_options=None, serialized_start=8182, - serialized_end=8255, -) + serialized_end=8255, ) _sym_db.RegisterEnumDescriptor(_RESIZEPARAMETER_INTERP_MODE) _LOSSPARAMETER_NORMALIZATIONMODE = _descriptor.EnumDescriptor( @@ -340,32 +318,24 @@ _LOSSPARAMETER_NORMALIZATIONMODE = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='FULL', - index=0, - number=0, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='VALID', - index=1, - number=1, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='BATCH_SIZE', - index=2, - number=2, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='NONE', - index=3, - number=3, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='FULL', index=0, number=0, serialized_options=None, type=None), + _descriptor.EnumValueDescriptor( + name='VALID', index=1, number=1, serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='BATCH_SIZE', + index=2, + number=2, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='NONE', index=3, number=3, serialized_options=None, type=None), ], containing_type=None, serialized_options=None, serialized_start=9202, - serialized_end=9268, -) + serialized_end=9268, ) _sym_db.RegisterEnumDescriptor(_LOSSPARAMETER_NORMALIZATIONMODE) _CONVOLUTIONPARAMETER_ENGINE = _descriptor.EnumDescriptor( @@ -374,27 +344,23 @@ _CONVOLUTIONPARAMETER_ENGINE = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='DEFAULT', - index=0, - number=0, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='CAFFE', - index=1, - number=1, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='CUDNN', - index=2, - number=2, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='DEFAULT', + index=0, + number=0, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='CAFFE', index=1, number=1, serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='CUDNN', index=2, number=2, serialized_options=None, + type=None), ], containing_type=None, serialized_options=None, serialized_start=10385, - serialized_end=10428, -) + serialized_end=10428, ) _sym_db.RegisterEnumDescriptor(_CONVOLUTIONPARAMETER_ENGINE) _DATAPARAMETER_DB = _descriptor.EnumDescriptor( @@ -403,22 +369,19 @@ _DATAPARAMETER_DB = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='LEVELDB', - index=0, - number=0, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='LMDB', - index=1, - number=1, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='LEVELDB', + index=0, + number=0, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='LMDB', index=1, number=1, serialized_options=None, type=None), ], containing_type=None, serialized_options=None, serialized_start=10746, - serialized_end=10773, -) + serialized_end=10773, ) _sym_db.RegisterEnumDescriptor(_DATAPARAMETER_DB) _ELTWISEPARAMETER_ELTWISEOP = _descriptor.EnumDescriptor( @@ -427,27 +390,17 @@ _ELTWISEPARAMETER_ELTWISEOP = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='PROD', - index=0, - number=0, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='SUM', - index=1, - number=1, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='MAX', - index=2, - number=2, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='PROD', index=0, number=0, serialized_options=None, type=None), + _descriptor.EnumValueDescriptor( + name='SUM', index=1, number=1, serialized_options=None, type=None), + _descriptor.EnumValueDescriptor( + name='MAX', index=2, number=2, serialized_options=None, type=None), ], containing_type=None, serialized_options=None, serialized_start=12106, - serialized_end=12145, -) + serialized_end=12145, ) _sym_db.RegisterEnumDescriptor(_ELTWISEPARAMETER_ELTWISEOP) _HINGELOSSPARAMETER_NORM = _descriptor.EnumDescriptor( @@ -456,22 +409,15 @@ _HINGELOSSPARAMETER_NORM = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='L1', - index=0, - number=1, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='L2', - index=1, - number=2, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='L1', index=0, number=1, serialized_options=None, type=None), + _descriptor.EnumValueDescriptor( + name='L2', index=1, number=2, serialized_options=None, type=None), ], containing_type=None, serialized_options=None, serialized_start=12680, - serialized_end=12702, -) + serialized_end=12702, ) _sym_db.RegisterEnumDescriptor(_HINGELOSSPARAMETER_NORM) _LRNPARAMETER_NORMREGION = _descriptor.EnumDescriptor( @@ -480,22 +426,23 @@ _LRNPARAMETER_NORMREGION = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='ACROSS_CHANNELS', - index=0, - number=0, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='WITHIN_CHANNEL', - index=1, - number=1, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='ACROSS_CHANNELS', + index=0, + number=0, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='WITHIN_CHANNEL', + index=1, + number=1, + serialized_options=None, + type=None), ], containing_type=None, serialized_options=None, serialized_start=13569, - serialized_end=13622, -) + serialized_end=13622, ) _sym_db.RegisterEnumDescriptor(_LRNPARAMETER_NORMREGION) _LRNPARAMETER_ENGINE = _descriptor.EnumDescriptor( @@ -504,27 +451,23 @@ _LRNPARAMETER_ENGINE = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='DEFAULT', - index=0, - number=0, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='CAFFE', - index=1, - number=1, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='CUDNN', - index=2, - number=2, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='DEFAULT', + index=0, + number=0, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='CAFFE', index=1, number=1, serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='CUDNN', index=2, number=2, serialized_options=None, + type=None), ], containing_type=None, serialized_options=None, serialized_start=10385, - serialized_end=10428, -) + serialized_end=10428, ) _sym_db.RegisterEnumDescriptor(_LRNPARAMETER_ENGINE) _MULTIBOXLOSSPARAMETER_LOCLOSSTYPE = _descriptor.EnumDescriptor( @@ -533,22 +476,19 @@ _MULTIBOXLOSSPARAMETER_LOCLOSSTYPE = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='L2', - index=0, - number=0, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='SMOOTH_L1', - index=1, - number=1, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='L2', index=0, number=0, serialized_options=None, type=None), + _descriptor.EnumValueDescriptor( + name='SMOOTH_L1', + index=1, + number=1, + serialized_options=None, + type=None), ], containing_type=None, serialized_options=None, serialized_start=14703, - serialized_end=14739, -) + serialized_end=14739, ) _sym_db.RegisterEnumDescriptor(_MULTIBOXLOSSPARAMETER_LOCLOSSTYPE) _MULTIBOXLOSSPARAMETER_CONFLOSSTYPE = _descriptor.EnumDescriptor( @@ -557,22 +497,23 @@ _MULTIBOXLOSSPARAMETER_CONFLOSSTYPE = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='SOFTMAX', - index=0, - number=0, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='LOGISTIC', - index=1, - number=1, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='SOFTMAX', + index=0, + number=0, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='LOGISTIC', + index=1, + number=1, + serialized_options=None, + type=None), ], containing_type=None, serialized_options=None, serialized_start=14741, - serialized_end=14782, -) + serialized_end=14782, ) _sym_db.RegisterEnumDescriptor(_MULTIBOXLOSSPARAMETER_CONFLOSSTYPE) _MULTIBOXLOSSPARAMETER_MATCHTYPE = _descriptor.EnumDescriptor( @@ -581,22 +522,23 @@ _MULTIBOXLOSSPARAMETER_MATCHTYPE = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='BIPARTITE', - index=0, - number=0, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='PER_PREDICTION', - index=1, - number=1, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='BIPARTITE', + index=0, + number=0, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='PER_PREDICTION', + index=1, + number=1, + serialized_options=None, + type=None), ], containing_type=None, serialized_options=None, serialized_start=14784, - serialized_end=14830, -) + serialized_end=14830, ) _sym_db.RegisterEnumDescriptor(_MULTIBOXLOSSPARAMETER_MATCHTYPE) _MULTIBOXLOSSPARAMETER_MININGTYPE = _descriptor.EnumDescriptor( @@ -605,27 +547,25 @@ _MULTIBOXLOSSPARAMETER_MININGTYPE = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='NONE', - index=0, - number=0, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='MAX_NEGATIVE', - index=1, - number=1, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='HARD_EXAMPLE', - index=2, - number=2, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='NONE', index=0, number=0, serialized_options=None, type=None), + _descriptor.EnumValueDescriptor( + name='MAX_NEGATIVE', + index=1, + number=1, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='HARD_EXAMPLE', + index=2, + number=2, + serialized_options=None, + type=None), ], containing_type=None, serialized_options=None, serialized_start=14832, - serialized_end=14890, -) + serialized_end=14890, ) _sym_db.RegisterEnumDescriptor(_MULTIBOXLOSSPARAMETER_MININGTYPE) _POOLINGPARAMETER_POOLMETHOD = _descriptor.EnumDescriptor( @@ -634,27 +574,21 @@ _POOLINGPARAMETER_POOLMETHOD = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='MAX', - index=0, - number=0, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='AVE', - index=1, - number=1, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='STOCHASTIC', - index=2, - number=2, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='MAX', index=0, number=0, serialized_options=None, type=None), + _descriptor.EnumValueDescriptor( + name='AVE', index=1, number=1, serialized_options=None, type=None), + _descriptor.EnumValueDescriptor( + name='STOCHASTIC', + index=2, + number=2, + serialized_options=None, + type=None), ], containing_type=None, serialized_options=None, serialized_start=15561, - serialized_end=15607, -) + serialized_end=15607, ) _sym_db.RegisterEnumDescriptor(_POOLINGPARAMETER_POOLMETHOD) _POOLINGPARAMETER_ENGINE = _descriptor.EnumDescriptor( @@ -663,27 +597,23 @@ _POOLINGPARAMETER_ENGINE = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='DEFAULT', - index=0, - number=0, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='CAFFE', - index=1, - number=1, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='CUDNN', - index=2, - number=2, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='DEFAULT', + index=0, + number=0, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='CAFFE', index=1, number=1, serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='CUDNN', index=2, number=2, serialized_options=None, + type=None), ], containing_type=None, serialized_options=None, serialized_start=10385, - serialized_end=10428, -) + serialized_end=10428, ) _sym_db.RegisterEnumDescriptor(_POOLINGPARAMETER_ENGINE) _PRIORBOXPARAMETER_CODETYPE = _descriptor.EnumDescriptor( @@ -692,27 +622,29 @@ _PRIORBOXPARAMETER_CODETYPE = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='CORNER', - index=0, - number=1, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='CENTER_SIZE', - index=1, - number=2, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='CORNER_SIZE', - index=2, - number=3, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='CORNER', + index=0, + number=1, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='CENTER_SIZE', + index=1, + number=2, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='CORNER_SIZE', + index=2, + number=3, + serialized_options=None, + type=None), ], containing_type=None, serialized_options=None, serialized_start=15980, - serialized_end=16036, -) + serialized_end=16036, ) _sym_db.RegisterEnumDescriptor(_PRIORBOXPARAMETER_CODETYPE) _REDUCTIONPARAMETER_REDUCTIONOP = _descriptor.EnumDescriptor( @@ -721,32 +653,20 @@ _REDUCTIONPARAMETER_REDUCTIONOP = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='SUM', - index=0, - number=1, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='ASUM', - index=1, - number=2, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='SUMSQ', - index=2, - number=3, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='MEAN', - index=3, - number=4, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='SUM', index=0, number=1, serialized_options=None, type=None), + _descriptor.EnumValueDescriptor( + name='ASUM', index=1, number=2, serialized_options=None, type=None), + _descriptor.EnumValueDescriptor( + name='SUMSQ', index=2, number=3, serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='MEAN', index=3, number=4, serialized_options=None, type=None), ], containing_type=None, serialized_options=None, serialized_start=16459, - serialized_end=16512, -) + serialized_end=16512, ) _sym_db.RegisterEnumDescriptor(_REDUCTIONPARAMETER_REDUCTIONOP) _RELUPARAMETER_ENGINE = _descriptor.EnumDescriptor( @@ -755,27 +675,23 @@ _RELUPARAMETER_ENGINE = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='DEFAULT', - index=0, - number=0, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='CAFFE', - index=1, - number=1, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='CUDNN', - index=2, - number=2, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='DEFAULT', + index=0, + number=0, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='CAFFE', index=1, number=1, serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='CUDNN', index=2, number=2, serialized_options=None, + type=None), ], containing_type=None, serialized_options=None, serialized_start=10385, - serialized_end=10428, -) + serialized_end=10428, ) _sym_db.RegisterEnumDescriptor(_RELUPARAMETER_ENGINE) _SIGMOIDPARAMETER_ENGINE = _descriptor.EnumDescriptor( @@ -784,27 +700,23 @@ _SIGMOIDPARAMETER_ENGINE = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='DEFAULT', - index=0, - number=0, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='CAFFE', - index=1, - number=1, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='CUDNN', - index=2, - number=2, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='DEFAULT', + index=0, + number=0, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='CAFFE', index=1, number=1, serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='CUDNN', index=2, number=2, serialized_options=None, + type=None), ], containing_type=None, serialized_options=None, serialized_start=10385, - serialized_end=10428, -) + serialized_end=10428, ) _sym_db.RegisterEnumDescriptor(_SIGMOIDPARAMETER_ENGINE) _SOFTMAXPARAMETER_ENGINE = _descriptor.EnumDescriptor( @@ -813,27 +725,23 @@ _SOFTMAXPARAMETER_ENGINE = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='DEFAULT', - index=0, - number=0, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='CAFFE', - index=1, - number=1, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='CUDNN', - index=2, - number=2, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='DEFAULT', + index=0, + number=0, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='CAFFE', index=1, number=1, serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='CUDNN', index=2, number=2, serialized_options=None, + type=None), ], containing_type=None, serialized_options=None, serialized_start=10385, - serialized_end=10428, -) + serialized_end=10428, ) _sym_db.RegisterEnumDescriptor(_SOFTMAXPARAMETER_ENGINE) _TANHPARAMETER_ENGINE = _descriptor.EnumDescriptor( @@ -842,27 +750,23 @@ _TANHPARAMETER_ENGINE = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='DEFAULT', - index=0, - number=0, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='CAFFE', - index=1, - number=1, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='CUDNN', - index=2, - number=2, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='DEFAULT', + index=0, + number=0, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='CAFFE', index=1, number=1, serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='CUDNN', index=2, number=2, serialized_options=None, + type=None), ], containing_type=None, serialized_options=None, serialized_start=10385, - serialized_end=10428, -) + serialized_end=10428, ) _sym_db.RegisterEnumDescriptor(_TANHPARAMETER_ENGINE) _VIDEODATAPARAMETER_VIDEOTYPE = _descriptor.EnumDescriptor( @@ -871,22 +775,20 @@ _VIDEODATAPARAMETER_VIDEOTYPE = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='WEBCAM', - index=0, - number=0, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='VIDEO', - index=1, - number=1, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='WEBCAM', + index=0, + number=0, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='VIDEO', index=1, number=1, serialized_options=None, + type=None), ], containing_type=None, serialized_options=None, serialized_start=17621, - serialized_end=17655, -) + serialized_end=17655, ) _sym_db.RegisterEnumDescriptor(_VIDEODATAPARAMETER_VIDEOTYPE) _SPPPARAMETER_POOLMETHOD = _descriptor.EnumDescriptor( @@ -895,27 +797,21 @@ _SPPPARAMETER_POOLMETHOD = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='MAX', - index=0, - number=0, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='AVE', - index=1, - number=1, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='STOCHASTIC', - index=2, - number=2, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='MAX', index=0, number=0, serialized_options=None, type=None), + _descriptor.EnumValueDescriptor( + name='AVE', index=1, number=1, serialized_options=None, type=None), + _descriptor.EnumValueDescriptor( + name='STOCHASTIC', + index=2, + number=2, + serialized_options=None, + type=None), ], containing_type=None, serialized_options=None, serialized_start=15561, - serialized_end=15607, -) + serialized_end=15607, ) _sym_db.RegisterEnumDescriptor(_SPPPARAMETER_POOLMETHOD) _SPPPARAMETER_ENGINE = _descriptor.EnumDescriptor( @@ -924,27 +820,23 @@ _SPPPARAMETER_ENGINE = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='DEFAULT', - index=0, - number=0, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='CAFFE', - index=1, - number=1, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='CUDNN', - index=2, - number=2, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='DEFAULT', + index=0, + number=0, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='CAFFE', index=1, number=1, serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='CUDNN', index=2, number=2, serialized_options=None, + type=None), ], containing_type=None, serialized_options=None, serialized_start=10385, - serialized_end=10428, -) + serialized_end=10428, ) _sym_db.RegisterEnumDescriptor(_SPPPARAMETER_ENGINE) _V1LAYERPARAMETER_LAYERTYPE = _descriptor.EnumDescriptor( @@ -953,212 +845,230 @@ _V1LAYERPARAMETER_LAYERTYPE = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='NONE', - index=0, - number=0, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='ABSVAL', - index=1, - number=35, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='ACCURACY', - index=2, - number=1, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='ARGMAX', - index=3, - number=30, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='BNLL', - index=4, - number=2, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='CONCAT', - index=5, - number=3, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='CONTRASTIVE_LOSS', - index=6, - number=37, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='CONVOLUTION', - index=7, - number=4, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='DATA', - index=8, - number=5, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='DECONVOLUTION', - index=9, - number=39, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='DROPOUT', - index=10, - number=6, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='DUMMY_DATA', - index=11, - number=32, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='EUCLIDEAN_LOSS', - index=12, - number=7, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='ELTWISE', - index=13, - number=25, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='EXP', - index=14, - number=38, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='FLATTEN', - index=15, - number=8, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='HDF5_DATA', - index=16, - number=9, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='HDF5_OUTPUT', - index=17, - number=10, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='HINGE_LOSS', - index=18, - number=28, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='IM2COL', - index=19, - number=11, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='IMAGE_DATA', - index=20, - number=12, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='INFOGAIN_LOSS', - index=21, - number=13, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='INNER_PRODUCT', - index=22, - number=14, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='LRN', - index=23, - number=15, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='MEMORY_DATA', - index=24, - number=29, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='MULTINOMIAL_LOGISTIC_LOSS', - index=25, - number=16, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='MVN', - index=26, - number=34, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='POOLING', - index=27, - number=17, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='POWER', - index=28, - number=26, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='RELU', - index=29, - number=18, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='SIGMOID', - index=30, - number=19, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='SIGMOID_CROSS_ENTROPY_LOSS', - index=31, - number=27, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='SILENCE', - index=32, - number=36, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='SOFTMAX', - index=33, - number=20, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='SOFTMAX_LOSS', - index=34, - number=21, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='SPLIT', - index=35, - number=22, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='SLICE', - index=36, - number=33, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='TANH', - index=37, - number=23, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='WINDOW_DATA', - index=38, - number=24, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='THRESHOLD', - index=39, - number=31, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='NONE', index=0, number=0, serialized_options=None, type=None), + _descriptor.EnumValueDescriptor( + name='ABSVAL', + index=1, + number=35, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='ACCURACY', + index=2, + number=1, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='ARGMAX', + index=3, + number=30, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='BNLL', index=4, number=2, serialized_options=None, type=None), + _descriptor.EnumValueDescriptor( + name='CONCAT', + index=5, + number=3, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='CONTRASTIVE_LOSS', + index=6, + number=37, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='CONVOLUTION', + index=7, + number=4, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='DATA', index=8, number=5, serialized_options=None, type=None), + _descriptor.EnumValueDescriptor( + name='DECONVOLUTION', + index=9, + number=39, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='DROPOUT', + index=10, + number=6, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='DUMMY_DATA', + index=11, + number=32, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='EUCLIDEAN_LOSS', + index=12, + number=7, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='ELTWISE', + index=13, + number=25, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='EXP', index=14, number=38, serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='FLATTEN', + index=15, + number=8, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='HDF5_DATA', + index=16, + number=9, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='HDF5_OUTPUT', + index=17, + number=10, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='HINGE_LOSS', + index=18, + number=28, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='IM2COL', + index=19, + number=11, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='IMAGE_DATA', + index=20, + number=12, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='INFOGAIN_LOSS', + index=21, + number=13, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='INNER_PRODUCT', + index=22, + number=14, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='LRN', index=23, number=15, serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='MEMORY_DATA', + index=24, + number=29, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='MULTINOMIAL_LOGISTIC_LOSS', + index=25, + number=16, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='MVN', index=26, number=34, serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='POOLING', + index=27, + number=17, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='POWER', + index=28, + number=26, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='RELU', + index=29, + number=18, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='SIGMOID', + index=30, + number=19, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='SIGMOID_CROSS_ENTROPY_LOSS', + index=31, + number=27, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='SILENCE', + index=32, + number=36, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='SOFTMAX', + index=33, + number=20, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='SOFTMAX_LOSS', + index=34, + number=21, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='SPLIT', + index=35, + number=22, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='SLICE', + index=36, + number=33, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='TANH', + index=37, + number=23, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='WINDOW_DATA', + index=38, + number=24, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='THRESHOLD', + index=39, + number=31, + serialized_options=None, + type=None), ], containing_type=None, serialized_options=None, serialized_start=20104, - serialized_end=20704, -) + serialized_end=20704, ) _sym_db.RegisterEnumDescriptor(_V1LAYERPARAMETER_LAYERTYPE) _V1LAYERPARAMETER_DIMCHECKMODE = _descriptor.EnumDescriptor( @@ -1167,22 +1077,23 @@ _V1LAYERPARAMETER_DIMCHECKMODE = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='STRICT', - index=0, - number=0, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='PERMISSIVE', - index=1, - number=1, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='STRICT', + index=0, + number=0, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='PERMISSIVE', + index=1, + number=1, + serialized_options=None, + type=None), ], containing_type=None, serialized_options=None, serialized_start=4131, - serialized_end=4173, -) + serialized_end=4173, ) _sym_db.RegisterEnumDescriptor(_V1LAYERPARAMETER_DIMCHECKMODE) _V0LAYERPARAMETER_POOLMETHOD = _descriptor.EnumDescriptor( @@ -1191,27 +1102,21 @@ _V0LAYERPARAMETER_POOLMETHOD = _descriptor.EnumDescriptor( filename=None, file=DESCRIPTOR, values=[ - _descriptor.EnumValueDescriptor(name='MAX', - index=0, - number=0, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='AVE', - index=1, - number=1, - serialized_options=None, - type=None), - _descriptor.EnumValueDescriptor(name='STOCHASTIC', - index=2, - number=2, - serialized_options=None, - type=None), + _descriptor.EnumValueDescriptor( + name='MAX', index=0, number=0, serialized_options=None, type=None), + _descriptor.EnumValueDescriptor( + name='AVE', index=1, number=1, serialized_options=None, type=None), + _descriptor.EnumValueDescriptor( + name='STOCHASTIC', + index=2, + number=2, + serialized_options=None, + type=None), ], containing_type=None, serialized_options=None, serialized_start=15561, - serialized_end=15607, -) + serialized_end=15607, ) _sym_db.RegisterEnumDescriptor(_V0LAYERPARAMETER_POOLMETHOD) _BLOBSHAPE = _descriptor.Descriptor( @@ -1221,22 +1126,23 @@ _BLOBSHAPE = _descriptor.Descriptor( file=DESCRIPTOR, containing_type=None, fields=[ - _descriptor.FieldDescriptor(name='dim', - full_name='caffe.BlobShape.dim', - index=0, - number=1, - type=3, - cpp_type=2, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=_b('\020\001'), - file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='dim', + full_name='caffe.BlobShape.dim', + index=0, + number=1, + type=3, + cpp_type=2, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=_b('\020\001'), + file=DESCRIPTOR), ], extensions=[], nested_types=[], @@ -1247,8 +1153,7 @@ _BLOBSHAPE = _descriptor.Descriptor( extension_ranges=[], oneofs=[], serialized_start=22, - serialized_end=50, -) + serialized_end=50, ) _BLOBPROTO = _descriptor.Descriptor( name='BlobProto', @@ -1256,537 +1161,17 @@ _BLOBPROTO = _descriptor.Descriptor( filename=None, file=DESCRIPTOR, containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='shape', - full_name='caffe.BlobProto.shape', - index=0, - number=7, - type=11, - cpp_type=10, - label=1, - has_default_value=False, - default_value=None, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='data', - full_name='caffe.BlobProto.data', - index=1, - number=5, - type=2, - cpp_type=6, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=_b('\020\001'), - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='diff', - full_name='caffe.BlobProto.diff', - index=2, - number=6, - type=2, - cpp_type=6, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=_b('\020\001'), - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='double_data', - full_name='caffe.BlobProto.double_data', - index=3, - number=8, - type=1, - cpp_type=5, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=_b('\020\001'), - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='double_diff', - full_name='caffe.BlobProto.double_diff', - index=4, - number=9, - type=1, - cpp_type=5, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=_b('\020\001'), - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='num', - full_name='caffe.BlobProto.num', - index=5, - number=1, - type=5, - cpp_type=1, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='channels', - full_name='caffe.BlobProto.channels', - index=6, - number=2, - type=5, - cpp_type=1, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='height', - full_name='caffe.BlobProto.height', - index=7, - number=3, - type=5, - cpp_type=1, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='width', - full_name='caffe.BlobProto.width', - index=8, - number=4, - type=5, - cpp_type=1, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=53, - serialized_end=257, -) - -_BLOBPROTOVECTOR = _descriptor.Descriptor( - name='BlobProtoVector', - full_name='caffe.BlobProtoVector', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='blobs', - full_name='caffe.BlobProtoVector.blobs', - index=0, - number=1, - type=11, - cpp_type=10, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=259, - serialized_end=309, -) - -_DATUM = _descriptor.Descriptor( - name='Datum', - full_name='caffe.Datum', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='channels', - full_name='caffe.Datum.channels', - index=0, - number=1, - type=5, - cpp_type=1, - label=1, - has_default_value=False, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='height', - full_name='caffe.Datum.height', - index=1, - number=2, - type=5, - cpp_type=1, - label=1, - has_default_value=False, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='width', - full_name='caffe.Datum.width', - index=2, - number=3, - type=5, - cpp_type=1, - label=1, - has_default_value=False, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='data', - full_name='caffe.Datum.data', - index=3, - number=4, - type=12, - cpp_type=9, - label=1, - has_default_value=False, - default_value=_b(""), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='label', - full_name='caffe.Datum.label', - index=4, - number=5, - type=5, - cpp_type=1, - label=1, - has_default_value=False, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='float_data', - full_name='caffe.Datum.float_data', - index=5, - number=6, - type=2, - cpp_type=6, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='encoded', - full_name='caffe.Datum.encoded', - index=6, - number=7, - type=8, - cpp_type=7, - label=1, - has_default_value=True, - default_value=False, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=312, - serialized_end=441, -) - -_LABELMAPITEM = _descriptor.Descriptor( - name='LabelMapItem', - full_name='caffe.LabelMapItem', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='name', - full_name='caffe.LabelMapItem.name', - index=0, - number=1, - type=9, - cpp_type=9, - label=1, - has_default_value=False, - default_value=_b("").decode('utf-8'), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='label', - full_name='caffe.LabelMapItem.label', - index=1, - number=2, - type=5, - cpp_type=1, - label=1, - has_default_value=False, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='display_name', - full_name='caffe.LabelMapItem.display_name', - index=2, - number=3, - type=9, - cpp_type=9, - label=1, - has_default_value=False, - default_value=_b("").decode('utf-8'), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=443, - serialized_end=508, -) - -_LABELMAP = _descriptor.Descriptor( - name='LabelMap', - full_name='caffe.LabelMap', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='item', - full_name='caffe.LabelMap.item', - index=0, - number=1, - type=11, - cpp_type=10, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=510, - serialized_end=555, -) - -_SAMPLER = _descriptor.Descriptor( - name='Sampler', - full_name='caffe.Sampler', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='min_scale', - full_name='caffe.Sampler.min_scale', - index=0, - number=1, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(1), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='max_scale', - full_name='caffe.Sampler.max_scale', - index=1, - number=2, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(1), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='min_aspect_ratio', - full_name='caffe.Sampler.min_aspect_ratio', - index=2, - number=3, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(1), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='max_aspect_ratio', - full_name='caffe.Sampler.max_aspect_ratio', - index=3, - number=4, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(1), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=557, - serialized_end=668, -) - -_SAMPLECONSTRAINT = _descriptor.Descriptor( - name='SampleConstraint', - full_name='caffe.SampleConstraint', - filename=None, - file=DESCRIPTOR, - containing_type=None, fields=[ _descriptor.FieldDescriptor( - name='min_jaccard_overlap', - full_name='caffe.SampleConstraint.min_jaccard_overlap', + name='shape', + full_name='caffe.BlobProto.shape', index=0, - number=1, - type=2, - cpp_type=6, + number=7, + type=11, + cpp_type=10, label=1, has_default_value=False, - default_value=float(0), + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -1795,83 +1180,83 @@ _SAMPLECONSTRAINT = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='max_jaccard_overlap', - full_name='caffe.SampleConstraint.max_jaccard_overlap', + name='data', + full_name='caffe.BlobProto.data', index=1, - number=2, + number=5, type=2, cpp_type=6, - label=1, + label=3, has_default_value=False, - default_value=float(0), + default_value=[], message_type=None, enum_type=None, containing_type=None, is_extension=False, extension_scope=None, - serialized_options=None, + serialized_options=_b('\020\001'), file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='min_sample_coverage', - full_name='caffe.SampleConstraint.min_sample_coverage', + name='diff', + full_name='caffe.BlobProto.diff', index=2, - number=3, + number=6, type=2, cpp_type=6, - label=1, + label=3, has_default_value=False, - default_value=float(0), + default_value=[], message_type=None, enum_type=None, containing_type=None, is_extension=False, extension_scope=None, - serialized_options=None, + serialized_options=_b('\020\001'), file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='max_sample_coverage', - full_name='caffe.SampleConstraint.max_sample_coverage', + name='double_data', + full_name='caffe.BlobProto.double_data', index=3, - number=4, - type=2, - cpp_type=6, - label=1, + number=8, + type=1, + cpp_type=5, + label=3, has_default_value=False, - default_value=float(0), + default_value=[], message_type=None, enum_type=None, containing_type=None, is_extension=False, extension_scope=None, - serialized_options=None, + serialized_options=_b('\020\001'), file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='min_object_coverage', - full_name='caffe.SampleConstraint.min_object_coverage', + name='double_diff', + full_name='caffe.BlobProto.double_diff', index=4, - number=5, - type=2, - cpp_type=6, - label=1, + number=9, + type=1, + cpp_type=5, + label=3, has_default_value=False, - default_value=float(0), + default_value=[], message_type=None, enum_type=None, containing_type=None, is_extension=False, extension_scope=None, - serialized_options=None, + serialized_options=_b('\020\001'), file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='max_object_coverage', - full_name='caffe.SampleConstraint.max_object_coverage', + name='num', + full_name='caffe.BlobProto.num', index=5, - number=6, - type=2, - cpp_type=6, + number=1, + type=5, + cpp_type=1, label=1, - has_default_value=False, - default_value=float(0), + has_default_value=True, + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -1879,36 +1264,16 @@ _SAMPLECONSTRAINT = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=671, - serialized_end=863, -) - -_BATCHSAMPLER = _descriptor.Descriptor( - name='BatchSampler', - full_name='caffe.BatchSampler', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ _descriptor.FieldDescriptor( - name='use_original_image', - full_name='caffe.BatchSampler.use_original_image', - index=0, - number=1, - type=8, - cpp_type=7, + name='channels', + full_name='caffe.BlobProto.channels', + index=6, + number=2, + type=5, + cpp_type=1, label=1, has_default_value=True, - default_value=True, + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -1916,32 +1281,33 @@ _BATCHSAMPLER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='sampler', - full_name='caffe.BatchSampler.sampler', - index=1, - number=2, - type=11, - cpp_type=10, - label=1, - has_default_value=False, - default_value=None, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='sample_constraint', - full_name='caffe.BatchSampler.sample_constraint', - index=2, + name='height', + full_name='caffe.BlobProto.height', + index=7, number=3, - type=11, - cpp_type=10, + type=5, + cpp_type=1, label=1, - has_default_value=False, - default_value=None, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='width', + full_name='caffe.BlobProto.width', + index=8, + number=4, + type=5, + cpp_type=1, + label=1, + has_default_value=True, + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -1949,38 +1315,6 @@ _BATCHSAMPLER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='max_sample', - full_name='caffe.BatchSampler.max_sample', - index=3, - number=4, - type=13, - cpp_type=3, - label=1, - has_default_value=False, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='max_trials', - full_name='caffe.BatchSampler.max_trials', - index=4, - number=5, - type=13, - cpp_type=3, - label=1, - has_default_value=True, - default_value=100, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), ], extensions=[], nested_types=[], @@ -1990,43 +1324,26 @@ _BATCHSAMPLER = _descriptor.Descriptor( syntax='proto2', extension_ranges=[], oneofs=[], - serialized_start=866, - serialized_end=1044, -) + serialized_start=53, + serialized_end=257, ) -_EMITCONSTRAINT = _descriptor.Descriptor( - name='EmitConstraint', - full_name='caffe.EmitConstraint', +_BLOBPROTOVECTOR = _descriptor.Descriptor( + name='BlobProtoVector', + full_name='caffe.BlobProtoVector', filename=None, file=DESCRIPTOR, containing_type=None, fields=[ - _descriptor.FieldDescriptor(name='emit_type', - full_name='caffe.EmitConstraint.emit_type', - index=0, - number=1, - type=14, - cpp_type=8, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='emit_overlap', - full_name='caffe.EmitConstraint.emit_overlap', - index=1, - number=2, - type=2, - cpp_type=6, - label=1, + name='blobs', + full_name='caffe.BlobProtoVector.blobs', + index=0, + number=1, + type=11, + cpp_type=10, + label=3, has_default_value=False, - default_value=float(0), + default_value=[], message_type=None, enum_type=None, containing_type=None, @@ -2037,228 +1354,25 @@ _EMITCONSTRAINT = _descriptor.Descriptor( ], extensions=[], nested_types=[], - enum_types=[ - _EMITCONSTRAINT_EMITTYPE, - ], + enum_types=[], serialized_options=None, is_extendable=False, syntax='proto2', extension_ranges=[], oneofs=[], - serialized_start=1047, - serialized_end=1185, -) + serialized_start=259, + serialized_end=309, ) -_NORMALIZEDBBOX = _descriptor.Descriptor( - name='NormalizedBBox', - full_name='caffe.NormalizedBBox', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='xmin', - full_name='caffe.NormalizedBBox.xmin', - index=0, - number=1, - type=2, - cpp_type=6, - label=1, - has_default_value=False, - default_value=float(0), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='ymin', - full_name='caffe.NormalizedBBox.ymin', - index=1, - number=2, - type=2, - cpp_type=6, - label=1, - has_default_value=False, - default_value=float(0), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='xmax', - full_name='caffe.NormalizedBBox.xmax', - index=2, - number=3, - type=2, - cpp_type=6, - label=1, - has_default_value=False, - default_value=float(0), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='ymax', - full_name='caffe.NormalizedBBox.ymax', - index=3, - number=4, - type=2, - cpp_type=6, - label=1, - has_default_value=False, - default_value=float(0), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='label', - full_name='caffe.NormalizedBBox.label', - index=4, - number=5, - type=5, - cpp_type=1, - label=1, - has_default_value=False, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='difficult', - full_name='caffe.NormalizedBBox.difficult', - index=5, - number=6, - type=8, - cpp_type=7, - label=1, - has_default_value=False, - default_value=False, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='score', - full_name='caffe.NormalizedBBox.score', - index=6, - number=7, - type=2, - cpp_type=6, - label=1, - has_default_value=False, - default_value=float(0), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='size', - full_name='caffe.NormalizedBBox.size', - index=7, - number=8, - type=2, - cpp_type=6, - label=1, - has_default_value=False, - default_value=float(0), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=1188, - serialized_end=1323, -) - -_ANNOTATION = _descriptor.Descriptor( - name='Annotation', - full_name='caffe.Annotation', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='instance_id', - full_name='caffe.Annotation.instance_id', - index=0, - number=1, - type=5, - cpp_type=1, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='bbox', - full_name='caffe.Annotation.bbox', - index=1, - number=2, - type=11, - cpp_type=10, - label=1, - has_default_value=False, - default_value=None, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=1325, - serialized_end=1398, -) - -_ANNOTATIONGROUP = _descriptor.Descriptor( - name='AnnotationGroup', - full_name='caffe.AnnotationGroup', +_DATUM = _descriptor.Descriptor( + name='Datum', + full_name='caffe.Datum', filename=None, file=DESCRIPTOR, containing_type=None, fields=[ _descriptor.FieldDescriptor( - name='group_label', - full_name='caffe.AnnotationGroup.group_label', + name='channels', + full_name='caffe.Datum.channels', index=0, number=1, type=5, @@ -2274,15 +1388,15 @@ _ANNOTATIONGROUP = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='annotation', - full_name='caffe.AnnotationGroup.annotation', + name='height', + full_name='caffe.Datum.height', index=1, number=2, - type=11, - cpp_type=10, - label=3, + type=5, + cpp_type=1, + label=1, has_default_value=False, - default_value=[], + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -2290,68 +1404,16 @@ _ANNOTATIONGROUP = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=1400, - serialized_end=1477, -) - -_ANNOTATEDDATUM = _descriptor.Descriptor( - name='AnnotatedDatum', - full_name='caffe.AnnotatedDatum', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='datum', - full_name='caffe.AnnotatedDatum.datum', - index=0, - number=1, - type=11, - cpp_type=10, - label=1, - has_default_value=False, - default_value=None, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='type', - full_name='caffe.AnnotatedDatum.type', - index=1, - number=2, - type=14, - cpp_type=8, - label=1, - has_default_value=False, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='annotation_group', - full_name='caffe.AnnotatedDatum.annotation_group', + name='width', + full_name='caffe.Datum.width', index=2, number=3, - type=11, - cpp_type=10, - label=3, + type=5, + cpp_type=1, + label=1, has_default_value=False, - default_value=[], + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -2359,38 +1421,16 @@ _ANNOTATEDDATUM = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[ - _ANNOTATEDDATUM_ANNOTATIONTYPE, - ], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=1480, - serialized_end=1655, -) - -_FILLERPARAMETER = _descriptor.Descriptor( - name='FillerParameter', - full_name='caffe.FillerParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ _descriptor.FieldDescriptor( - name='type', - full_name='caffe.FillerParameter.type', - index=0, - number=1, - type=9, + name='data', + full_name='caffe.Datum.data', + index=3, + number=4, + type=12, cpp_type=9, label=1, - has_default_value=True, - default_value=_b("constant").decode('utf-8'), + has_default_value=False, + default_value=_b(""), message_type=None, enum_type=None, containing_type=None, @@ -2398,111 +1438,15 @@ _FILLERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='value', - full_name='caffe.FillerParameter.value', - index=1, - number=2, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(0), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='min', - full_name='caffe.FillerParameter.min', - index=2, - number=3, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(0), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='max', - full_name='caffe.FillerParameter.max', - index=3, - number=4, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(1), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='mean', - full_name='caffe.FillerParameter.mean', - index=4, - number=5, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(0), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='std', - full_name='caffe.FillerParameter.std', - index=5, - number=6, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(1), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='sparse', - full_name='caffe.FillerParameter.sparse', - index=6, - number=7, - type=5, - cpp_type=1, - label=1, - has_default_value=True, - default_value=-1, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='variance_norm', - full_name='caffe.FillerParameter.variance_norm', - index=7, - number=8, - type=14, - cpp_type=8, + name='label', + full_name='caffe.Datum.label', + index=4, + number=5, + type=5, + cpp_type=1, label=1, - has_default_value=True, + has_default_value=False, default_value=0, message_type=None, enum_type=None, @@ -2511,97 +1455,28 @@ _FILLERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[ - _FILLERPARAMETER_VARIANCENORM, - ], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=1658, - serialized_end=1924, -) - -_NETPARAMETER = _descriptor.Descriptor( - name='NetParameter', - full_name='caffe.NetParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='name', - full_name='caffe.NetParameter.name', - index=0, - number=1, - type=9, - cpp_type=9, - label=1, - has_default_value=False, - default_value=_b("").decode('utf-8'), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='input', - full_name='caffe.NetParameter.input', - index=1, - number=3, - type=9, - cpp_type=9, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='input_shape', - full_name='caffe.NetParameter.input_shape', - index=2, - number=8, - type=11, - cpp_type=10, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='input_dim', - full_name='caffe.NetParameter.input_dim', - index=3, - number=4, - type=5, - cpp_type=1, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='force_backward', - full_name='caffe.NetParameter.force_backward', - index=4, - number=5, + name='float_data', + full_name='caffe.Datum.float_data', + index=5, + number=6, + type=2, + cpp_type=6, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='encoded', + full_name='caffe.Datum.encoded', + index=6, + number=7, type=8, cpp_type=7, label=1, @@ -2614,70 +1489,6 @@ _NETPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='state', - full_name='caffe.NetParameter.state', - index=5, - number=6, - type=11, - cpp_type=10, - label=1, - has_default_value=False, - default_value=None, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='debug_info', - full_name='caffe.NetParameter.debug_info', - index=6, - number=7, - type=8, - cpp_type=7, - label=1, - has_default_value=True, - default_value=False, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='layer', - full_name='caffe.NetParameter.layer', - index=7, - number=100, - type=11, - cpp_type=10, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='layers', - full_name='caffe.NetParameter.layers', - index=8, - number=2, - type=11, - cpp_type=10, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), ], extensions=[], nested_types=[], @@ -2687,91 +1498,26 @@ _NETPARAMETER = _descriptor.Descriptor( syntax='proto2', extension_ranges=[], oneofs=[], - serialized_start=1927, - serialized_end=2197, -) + serialized_start=312, + serialized_end=441, ) -_SOLVERPARAMETER = _descriptor.Descriptor( - name='SolverParameter', - full_name='caffe.SolverParameter', +_LABELMAPITEM = _descriptor.Descriptor( + name='LabelMapItem', + full_name='caffe.LabelMapItem', filename=None, file=DESCRIPTOR, containing_type=None, fields=[ - _descriptor.FieldDescriptor(name='net', - full_name='caffe.SolverParameter.net', - index=0, - number=24, - type=9, - cpp_type=9, - label=1, - has_default_value=False, - default_value=_b("").decode('utf-8'), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='net_param', - full_name='caffe.SolverParameter.net_param', - index=1, - number=25, - type=11, - cpp_type=10, - label=1, - has_default_value=False, - default_value=None, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='train_net', - full_name='caffe.SolverParameter.train_net', - index=2, - number=1, - type=9, - cpp_type=9, - label=1, - has_default_value=False, - default_value=_b("").decode('utf-8'), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='test_net', - full_name='caffe.SolverParameter.test_net', - index=3, - number=2, - type=9, - cpp_type=9, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='train_net_param', - full_name='caffe.SolverParameter.train_net_param', - index=4, - number=21, - type=11, - cpp_type=10, + name='name', + full_name='caffe.LabelMapItem.name', + index=0, + number=1, + type=9, + cpp_type=9, label=1, has_default_value=False, - default_value=None, + default_value=_b("").decode('utf-8'), message_type=None, enum_type=None, containing_type=None, @@ -2780,15 +1526,15 @@ _SOLVERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='test_net_param', - full_name='caffe.SolverParameter.test_net_param', - index=5, - number=22, - type=11, - cpp_type=10, - label=3, + name='label', + full_name='caffe.LabelMapItem.label', + index=1, + number=2, + type=5, + cpp_type=1, + label=1, has_default_value=False, - default_value=[], + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -2797,15 +1543,15 @@ _SOLVERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='train_state', - full_name='caffe.SolverParameter.train_state', - index=6, - number=26, - type=11, - cpp_type=10, + name='display_name', + full_name='caffe.LabelMapItem.display_name', + index=2, + number=3, + type=9, + cpp_type=9, label=1, has_default_value=False, - default_value=None, + default_value=_b("").decode('utf-8'), message_type=None, enum_type=None, containing_type=None, @@ -2813,11 +1559,30 @@ _SOLVERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=443, + serialized_end=508, ) + +_LABELMAP = _descriptor.Descriptor( + name='LabelMap', + full_name='caffe.LabelMap', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ _descriptor.FieldDescriptor( - name='test_state', - full_name='caffe.SolverParameter.test_state', - index=7, - number=27, + name='item', + full_name='caffe.LabelMap.item', + index=0, + number=1, type=11, cpp_type=10, label=3, @@ -2830,16 +1595,35 @@ _SOLVERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=510, + serialized_end=555, ) + +_SAMPLER = _descriptor.Descriptor( + name='Sampler', + full_name='caffe.Sampler', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ _descriptor.FieldDescriptor( - name='eval_type', - full_name='caffe.SolverParameter.eval_type', - index=8, - number=41, - type=9, - cpp_type=9, + name='min_scale', + full_name='caffe.Sampler.min_scale', + index=0, + number=1, + type=2, + cpp_type=6, label=1, has_default_value=True, - default_value=_b("classification").decode('utf-8'), + default_value=float(1), message_type=None, enum_type=None, containing_type=None, @@ -2848,15 +1632,15 @@ _SOLVERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='ap_version', - full_name='caffe.SolverParameter.ap_version', - index=9, - number=42, - type=9, - cpp_type=9, + name='max_scale', + full_name='caffe.Sampler.max_scale', + index=1, + number=2, + type=2, + cpp_type=6, label=1, has_default_value=True, - default_value=_b("Integral").decode('utf-8'), + default_value=float(1), message_type=None, enum_type=None, containing_type=None, @@ -2865,15 +1649,15 @@ _SOLVERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='show_per_class_result', - full_name='caffe.SolverParameter.show_per_class_result', - index=10, - number=44, - type=8, - cpp_type=7, + name='min_aspect_ratio', + full_name='caffe.Sampler.min_aspect_ratio', + index=2, + number=3, + type=2, + cpp_type=6, label=1, has_default_value=True, - default_value=False, + default_value=float(1), message_type=None, enum_type=None, containing_type=None, @@ -2881,32 +1665,16 @@ _SOLVERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='test_iter', - full_name='caffe.SolverParameter.test_iter', - index=11, - number=3, - type=5, - cpp_type=1, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='test_interval', - full_name='caffe.SolverParameter.test_interval', - index=12, + name='max_aspect_ratio', + full_name='caffe.Sampler.max_aspect_ratio', + index=3, number=4, - type=5, - cpp_type=1, + type=2, + cpp_type=6, label=1, has_default_value=True, - default_value=0, + default_value=float(1), message_type=None, enum_type=None, containing_type=None, @@ -2914,16 +1682,35 @@ _SOLVERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=557, + serialized_end=668, ) + +_SAMPLECONSTRAINT = _descriptor.Descriptor( + name='SampleConstraint', + full_name='caffe.SampleConstraint', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ _descriptor.FieldDescriptor( - name='test_compute_loss', - full_name='caffe.SolverParameter.test_compute_loss', - index=13, - number=19, - type=8, - cpp_type=7, + name='min_jaccard_overlap', + full_name='caffe.SampleConstraint.min_jaccard_overlap', + index=0, + number=1, + type=2, + cpp_type=6, label=1, - has_default_value=True, - default_value=False, + has_default_value=False, + default_value=float(0), message_type=None, enum_type=None, containing_type=None, @@ -2932,15 +1719,15 @@ _SOLVERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='test_initialization', - full_name='caffe.SolverParameter.test_initialization', - index=14, - number=32, - type=8, - cpp_type=7, + name='max_jaccard_overlap', + full_name='caffe.SampleConstraint.max_jaccard_overlap', + index=1, + number=2, + type=2, + cpp_type=6, label=1, - has_default_value=True, - default_value=True, + has_default_value=False, + default_value=float(0), message_type=None, enum_type=None, containing_type=None, @@ -2948,48 +1735,16 @@ _SOLVERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='base_lr', - full_name='caffe.SolverParameter.base_lr', - index=15, - number=5, - type=2, - cpp_type=6, - label=1, - has_default_value=False, - default_value=float(0), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='display', - full_name='caffe.SolverParameter.display', - index=16, - number=6, - type=5, - cpp_type=1, - label=1, - has_default_value=False, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='average_loss', - full_name='caffe.SolverParameter.average_loss', - index=17, - number=33, - type=5, - cpp_type=1, + name='min_sample_coverage', + full_name='caffe.SampleConstraint.min_sample_coverage', + index=2, + number=3, + type=2, + cpp_type=6, label=1, - has_default_value=True, - default_value=1, + has_default_value=False, + default_value=float(0), message_type=None, enum_type=None, containing_type=None, @@ -2997,107 +1752,11 @@ _SOLVERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='max_iter', - full_name='caffe.SolverParameter.max_iter', - index=18, - number=7, - type=5, - cpp_type=1, - label=1, - has_default_value=False, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='iter_size', - full_name='caffe.SolverParameter.iter_size', - index=19, - number=36, - type=5, - cpp_type=1, - label=1, - has_default_value=True, - default_value=1, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='lr_policy', - full_name='caffe.SolverParameter.lr_policy', - index=20, - number=8, - type=9, - cpp_type=9, - label=1, - has_default_value=False, - default_value=_b("").decode('utf-8'), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='gamma', - full_name='caffe.SolverParameter.gamma', - index=21, - number=9, - type=2, - cpp_type=6, - label=1, - has_default_value=False, - default_value=float(0), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='power', - full_name='caffe.SolverParameter.power', - index=22, - number=10, - type=2, - cpp_type=6, - label=1, - has_default_value=False, - default_value=float(0), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='momentum', - full_name='caffe.SolverParameter.momentum', - index=23, - number=11, - type=2, - cpp_type=6, - label=1, - has_default_value=False, - default_value=float(0), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='weight_decay', - full_name='caffe.SolverParameter.weight_decay', - index=24, - number=12, + name='max_sample_coverage', + full_name='caffe.SampleConstraint.max_sample_coverage', + index=3, + number=4, type=2, cpp_type=6, label=1, @@ -3111,15 +1770,15 @@ _SOLVERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='regularization_type', - full_name='caffe.SolverParameter.regularization_type', - index=25, - number=29, - type=9, - cpp_type=9, + name='min_object_coverage', + full_name='caffe.SampleConstraint.min_object_coverage', + index=4, + number=5, + type=2, + cpp_type=6, label=1, - has_default_value=True, - default_value=_b("L2").decode('utf-8'), + has_default_value=False, + default_value=float(0), message_type=None, enum_type=None, containing_type=None, @@ -3127,48 +1786,16 @@ _SOLVERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='stepsize', - full_name='caffe.SolverParameter.stepsize', - index=26, - number=13, - type=5, - cpp_type=1, - label=1, - has_default_value=False, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='stepvalue', - full_name='caffe.SolverParameter.stepvalue', - index=27, - number=34, - type=5, - cpp_type=1, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='plateau_winsize', - full_name='caffe.SolverParameter.plateau_winsize', - index=28, - number=43, - type=5, - cpp_type=1, - label=3, + name='max_object_coverage', + full_name='caffe.SampleConstraint.max_object_coverage', + index=5, + number=6, + type=2, + cpp_type=6, + label=1, has_default_value=False, - default_value=[], + default_value=float(0), message_type=None, enum_type=None, containing_type=None, @@ -3176,16 +1803,35 @@ _SOLVERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=671, + serialized_end=863, ) + +_BATCHSAMPLER = _descriptor.Descriptor( + name='BatchSampler', + full_name='caffe.BatchSampler', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ _descriptor.FieldDescriptor( - name='clip_gradients', - full_name='caffe.SolverParameter.clip_gradients', - index=29, - number=35, - type=2, - cpp_type=6, + name='use_original_image', + full_name='caffe.BatchSampler.use_original_image', + index=0, + number=1, + type=8, + cpp_type=7, label=1, has_default_value=True, - default_value=float(-1), + default_value=True, message_type=None, enum_type=None, containing_type=None, @@ -3193,32 +1839,16 @@ _SOLVERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='snapshot', - full_name='caffe.SolverParameter.snapshot', - index=30, - number=14, - type=5, - cpp_type=1, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='snapshot_prefix', - full_name='caffe.SolverParameter.snapshot_prefix', - index=31, - number=15, - type=9, - cpp_type=9, + name='sampler', + full_name='caffe.BatchSampler.sampler', + index=1, + number=2, + type=11, + cpp_type=10, label=1, has_default_value=False, - default_value=_b("").decode('utf-8'), + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -3227,15 +1857,15 @@ _SOLVERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='snapshot_diff', - full_name='caffe.SolverParameter.snapshot_diff', - index=32, - number=16, - type=8, - cpp_type=7, + name='sample_constraint', + full_name='caffe.BatchSampler.sample_constraint', + index=2, + number=3, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=False, + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -3244,15 +1874,15 @@ _SOLVERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='snapshot_format', - full_name='caffe.SolverParameter.snapshot_format', - index=33, - number=37, - type=14, - cpp_type=8, + name='max_sample', + full_name='caffe.BatchSampler.max_sample', + index=3, + number=4, + type=13, + cpp_type=3, label=1, - has_default_value=True, - default_value=1, + has_default_value=False, + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -3261,15 +1891,15 @@ _SOLVERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='solver_mode', - full_name='caffe.SolverParameter.solver_mode', - index=34, - number=17, - type=14, - cpp_type=8, + name='max_trials', + full_name='caffe.BatchSampler.max_trials', + index=4, + number=5, + type=13, + cpp_type=3, label=1, has_default_value=True, - default_value=1, + default_value=100, message_type=None, enum_type=None, containing_type=None, @@ -3277,32 +1907,35 @@ _SOLVERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='device_id', - full_name='caffe.SolverParameter.device_id', - index=35, - number=18, - type=5, - cpp_type=1, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=866, + serialized_end=1044, ) + +_EMITCONSTRAINT = _descriptor.Descriptor( + name='EmitConstraint', + full_name='caffe.EmitConstraint', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ _descriptor.FieldDescriptor( - name='random_seed', - full_name='caffe.SolverParameter.random_seed', - index=36, - number=20, - type=3, - cpp_type=2, + name='emit_type', + full_name='caffe.EmitConstraint.emit_type', + index=0, + number=1, + type=14, + cpp_type=8, label=1, has_default_value=True, - default_value=-1, + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -3310,235 +1943,16 @@ _SOLVERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='type', - full_name='caffe.SolverParameter.type', - index=37, - number=40, - type=9, - cpp_type=9, - label=1, - has_default_value=True, - default_value=_b("SGD").decode('utf-8'), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='delta', - full_name='caffe.SolverParameter.delta', - index=38, - number=31, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(1e-08), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='momentum2', - full_name='caffe.SolverParameter.momentum2', - index=39, - number=39, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(0.999), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='rms_decay', - full_name='caffe.SolverParameter.rms_decay', - index=40, - number=38, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(0.99), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='debug_info', - full_name='caffe.SolverParameter.debug_info', - index=41, - number=23, - type=8, - cpp_type=7, - label=1, - has_default_value=True, - default_value=False, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor( - name='snapshot_after_train', - full_name='caffe.SolverParameter.snapshot_after_train', - index=42, - number=28, - type=8, - cpp_type=7, - label=1, - has_default_value=True, - default_value=True, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor( - name='solver_type', - full_name='caffe.SolverParameter.solver_type', - index=43, - number=30, - type=14, - cpp_type=8, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[ - _SOLVERPARAMETER_SNAPSHOTFORMAT, - _SOLVERPARAMETER_SOLVERMODE, - _SOLVERPARAMETER_SOLVERTYPE, - ], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=2200, - serialized_end=3642, -) - -_SOLVERSTATE = _descriptor.Descriptor( - name='SolverState', - full_name='caffe.SolverState', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='iter', - full_name='caffe.SolverState.iter', - index=0, - number=1, - type=5, - cpp_type=1, - label=1, - has_default_value=False, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='learned_net', - full_name='caffe.SolverState.learned_net', - index=1, - number=2, - type=9, - cpp_type=9, - label=1, - has_default_value=False, - default_value=_b("").decode('utf-8'), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='history', - full_name='caffe.SolverState.history', - index=2, - number=3, - type=11, - cpp_type=10, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='current_step', - full_name='caffe.SolverState.current_step', - index=3, - number=4, - type=5, - cpp_type=1, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='minimum_loss', - full_name='caffe.SolverState.minimum_loss', - index=4, - number=5, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(1e+38), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor( - name='iter_last_event', - full_name='caffe.SolverState.iter_last_event', - index=5, - number=6, - type=5, - cpp_type=1, + name='emit_overlap', + full_name='caffe.EmitConstraint.emit_overlap', + index=1, + number=2, + type=2, + cpp_type=6, label=1, - has_default_value=True, - default_value=0, + has_default_value=False, + default_value=float(0), message_type=None, enum_type=None, containing_type=None, @@ -3549,367 +1963,32 @@ _SOLVERSTATE = _descriptor.Descriptor( ], extensions=[], nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=3645, - serialized_end=3810, -) - -_NETSTATE = _descriptor.Descriptor( - name='NetState', - full_name='caffe.NetState', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='phase', - full_name='caffe.NetState.phase', - index=0, - number=1, - type=14, - cpp_type=8, - label=1, - has_default_value=True, - default_value=1, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='level', - full_name='caffe.NetState.level', - index=1, - number=2, - type=5, - cpp_type=1, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='stage', - full_name='caffe.NetState.stage', - index=2, - number=3, - type=9, - cpp_type=9, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=3812, - serialized_end=3890, -) - -_NETSTATERULE = _descriptor.Descriptor( - name='NetStateRule', - full_name='caffe.NetStateRule', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='phase', - full_name='caffe.NetStateRule.phase', - index=0, - number=1, - type=14, - cpp_type=8, - label=1, - has_default_value=False, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='min_level', - full_name='caffe.NetStateRule.min_level', - index=1, - number=2, - type=5, - cpp_type=1, - label=1, - has_default_value=False, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='max_level', - full_name='caffe.NetStateRule.max_level', - index=2, - number=3, - type=5, - cpp_type=1, - label=1, - has_default_value=False, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='stage', - full_name='caffe.NetStateRule.stage', - index=3, - number=4, - type=9, - cpp_type=9, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='not_stage', - full_name='caffe.NetStateRule.not_stage', - index=4, - number=5, - type=9, - cpp_type=9, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=3892, - serialized_end=4007, -) - -_PARAMSPEC = _descriptor.Descriptor( - name='ParamSpec', - full_name='caffe.ParamSpec', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='name', - full_name='caffe.ParamSpec.name', - index=0, - number=1, - type=9, - cpp_type=9, - label=1, - has_default_value=False, - default_value=_b("").decode('utf-8'), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='share_mode', - full_name='caffe.ParamSpec.share_mode', - index=1, - number=2, - type=14, - cpp_type=8, - label=1, - has_default_value=False, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='lr_mult', - full_name='caffe.ParamSpec.lr_mult', - index=2, - number=3, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(1), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='decay_mult', - full_name='caffe.ParamSpec.decay_mult', - index=3, - number=4, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(1), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[ - _PARAMSPEC_DIMCHECKMODE, - ], + enum_types=[_EMITCONSTRAINT_EMITTYPE, ], serialized_options=None, is_extendable=False, syntax='proto2', extension_ranges=[], oneofs=[], - serialized_start=4010, - serialized_end=4173, -) + serialized_start=1047, + serialized_end=1185, ) -_LAYERPARAMETER = _descriptor.Descriptor( - name='LayerParameter', - full_name='caffe.LayerParameter', +_NORMALIZEDBBOX = _descriptor.Descriptor( + name='NormalizedBBox', + full_name='caffe.NormalizedBBox', filename=None, file=DESCRIPTOR, containing_type=None, fields=[ - _descriptor.FieldDescriptor(name='name', - full_name='caffe.LayerParameter.name', - index=0, - number=1, - type=9, - cpp_type=9, - label=1, - has_default_value=False, - default_value=_b("").decode('utf-8'), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='type', - full_name='caffe.LayerParameter.type', - index=1, - number=2, - type=9, - cpp_type=9, - label=1, - has_default_value=False, - default_value=_b("").decode('utf-8'), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='bottom', - full_name='caffe.LayerParameter.bottom', - index=2, - number=3, - type=9, - cpp_type=9, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='top', - full_name='caffe.LayerParameter.top', - index=3, - number=4, - type=9, - cpp_type=9, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='phase', - full_name='caffe.LayerParameter.phase', - index=4, - number=10, - type=14, - cpp_type=8, - label=1, - has_default_value=False, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='loss_weight', - full_name='caffe.LayerParameter.loss_weight', - index=5, - number=5, + name='xmin', + full_name='caffe.NormalizedBBox.xmin', + index=0, + number=1, type=2, cpp_type=6, - label=3, + label=1, has_default_value=False, - default_value=[], + default_value=float(0), message_type=None, enum_type=None, containing_type=None, @@ -3917,48 +1996,16 @@ _LAYERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='param', - full_name='caffe.LayerParameter.param', - index=6, - number=6, - type=11, - cpp_type=10, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='blobs', - full_name='caffe.LayerParameter.blobs', - index=7, - number=7, - type=11, - cpp_type=10, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='propagate_down', - full_name='caffe.LayerParameter.propagate_down', - index=8, - number=11, - type=8, - cpp_type=7, - label=3, + name='ymin', + full_name='caffe.NormalizedBBox.ymin', + index=1, + number=2, + type=2, + cpp_type=6, + label=1, has_default_value=False, - default_value=[], + default_value=float(0), message_type=None, enum_type=None, containing_type=None, @@ -3966,48 +2013,16 @@ _LAYERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='include', - full_name='caffe.LayerParameter.include', - index=9, - number=8, - type=11, - cpp_type=10, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='exclude', - full_name='caffe.LayerParameter.exclude', - index=10, - number=9, - type=11, - cpp_type=10, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='transform_param', - full_name='caffe.LayerParameter.transform_param', - index=11, - number=100, - type=11, - cpp_type=10, + name='xmax', + full_name='caffe.NormalizedBBox.xmax', + index=2, + number=3, + type=2, + cpp_type=6, label=1, has_default_value=False, - default_value=None, + default_value=float(0), message_type=None, enum_type=None, containing_type=None, @@ -4015,32 +2030,16 @@ _LAYERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='loss_param', - full_name='caffe.LayerParameter.loss_param', - index=12, - number=101, - type=11, - cpp_type=10, - label=1, - has_default_value=False, - default_value=None, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='accuracy_param', - full_name='caffe.LayerParameter.accuracy_param', - index=13, - number=102, - type=11, - cpp_type=10, + name='ymax', + full_name='caffe.NormalizedBBox.ymax', + index=3, + number=4, + type=2, + cpp_type=6, label=1, has_default_value=False, - default_value=None, + default_value=float(0), message_type=None, enum_type=None, containing_type=None, @@ -4049,15 +2048,15 @@ _LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='annotated_data_param', - full_name='caffe.LayerParameter.annotated_data_param', - index=14, - number=200, - type=11, - cpp_type=10, + name='label', + full_name='caffe.NormalizedBBox.label', + index=4, + number=5, + type=5, + cpp_type=1, label=1, has_default_value=False, - default_value=None, + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -4066,15 +2065,15 @@ _LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='argmax_param', - full_name='caffe.LayerParameter.argmax_param', - index=15, - number=103, - type=11, - cpp_type=10, + name='difficult', + full_name='caffe.NormalizedBBox.difficult', + index=5, + number=6, + type=8, + cpp_type=7, label=1, has_default_value=False, - default_value=None, + default_value=False, message_type=None, enum_type=None, containing_type=None, @@ -4083,15 +2082,15 @@ _LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='batch_norm_param', - full_name='caffe.LayerParameter.batch_norm_param', - index=16, - number=139, - type=11, - cpp_type=10, + name='score', + full_name='caffe.NormalizedBBox.score', + index=6, + number=7, + type=2, + cpp_type=6, label=1, has_default_value=False, - default_value=None, + default_value=float(0), message_type=None, enum_type=None, containing_type=None, @@ -4099,49 +2098,16 @@ _LAYERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='bias_param', - full_name='caffe.LayerParameter.bias_param', - index=17, - number=141, - type=11, - cpp_type=10, - label=1, - has_default_value=False, - default_value=None, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='concat_param', - full_name='caffe.LayerParameter.concat_param', - index=18, - number=104, - type=11, - cpp_type=10, + name='size', + full_name='caffe.NormalizedBBox.size', + index=7, + number=8, + type=2, + cpp_type=6, label=1, has_default_value=False, - default_value=None, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor( - name='contrastive_loss_param', - full_name='caffe.LayerParameter.contrastive_loss_param', - index=19, - number=105, - type=11, - cpp_type=10, - label=1, - has_default_value=False, - default_value=None, + default_value=float(0), message_type=None, enum_type=None, containing_type=None, @@ -4149,16 +2115,35 @@ _LAYERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=1188, + serialized_end=1323, ) + +_ANNOTATION = _descriptor.Descriptor( + name='Annotation', + full_name='caffe.Annotation', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ _descriptor.FieldDescriptor( - name='convolution_param', - full_name='caffe.LayerParameter.convolution_param', - index=20, - number=106, - type=11, - cpp_type=10, + name='instance_id', + full_name='caffe.Annotation.instance_id', + index=0, + number=1, + type=5, + cpp_type=1, label=1, - has_default_value=False, - default_value=None, + has_default_value=True, + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -4166,43 +2151,11 @@ _LAYERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='crop_param', - full_name='caffe.LayerParameter.crop_param', - index=21, - number=144, - type=11, - cpp_type=10, - label=1, - has_default_value=False, - default_value=None, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='data_param', - full_name='caffe.LayerParameter.data_param', - index=22, - number=107, - type=11, - cpp_type=10, - label=1, - has_default_value=False, - default_value=None, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='detection_evaluate_param', - full_name='caffe.LayerParameter.detection_evaluate_param', - index=23, - number=205, + name='bbox', + full_name='caffe.Annotation.bbox', + index=1, + number=2, type=11, cpp_type=10, label=1, @@ -4215,16 +2168,35 @@ _LAYERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=1325, + serialized_end=1398, ) + +_ANNOTATIONGROUP = _descriptor.Descriptor( + name='AnnotationGroup', + full_name='caffe.AnnotationGroup', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ _descriptor.FieldDescriptor( - name='detection_output_param', - full_name='caffe.LayerParameter.detection_output_param', - index=24, - number=204, - type=11, - cpp_type=10, + name='group_label', + full_name='caffe.AnnotationGroup.group_label', + index=0, + number=1, + type=5, + cpp_type=1, label=1, has_default_value=False, - default_value=None, + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -4233,15 +2205,15 @@ _LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='dropout_param', - full_name='caffe.LayerParameter.dropout_param', - index=25, - number=108, + name='annotation', + full_name='caffe.AnnotationGroup.annotation', + index=1, + number=2, type=11, cpp_type=10, - label=1, + label=3, has_default_value=False, - default_value=None, + default_value=[], message_type=None, enum_type=None, containing_type=None, @@ -4249,11 +2221,30 @@ _LAYERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=1400, + serialized_end=1477, ) + +_ANNOTATEDDATUM = _descriptor.Descriptor( + name='AnnotatedDatum', + full_name='caffe.AnnotatedDatum', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ _descriptor.FieldDescriptor( - name='dummy_data_param', - full_name='caffe.LayerParameter.dummy_data_param', - index=26, - number=109, + name='datum', + full_name='caffe.AnnotatedDatum.datum', + index=0, + number=1, type=11, cpp_type=10, label=1, @@ -4267,15 +2258,15 @@ _LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='eltwise_param', - full_name='caffe.LayerParameter.eltwise_param', - index=27, - number=110, - type=11, - cpp_type=10, + name='type', + full_name='caffe.AnnotatedDatum.type', + index=1, + number=2, + type=14, + cpp_type=8, label=1, has_default_value=False, - default_value=None, + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -4283,32 +2274,16 @@ _LAYERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='elu_param', - full_name='caffe.LayerParameter.elu_param', - index=28, - number=140, - type=11, - cpp_type=10, - label=1, - has_default_value=False, - default_value=None, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='embed_param', - full_name='caffe.LayerParameter.embed_param', - index=29, - number=137, + name='annotation_group', + full_name='caffe.AnnotatedDatum.annotation_group', + index=2, + number=3, type=11, cpp_type=10, - label=1, + label=3, has_default_value=False, - default_value=None, + default_value=[], message_type=None, enum_type=None, containing_type=None, @@ -4316,32 +2291,35 @@ _LAYERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='exp_param', - full_name='caffe.LayerParameter.exp_param', - index=30, - number=111, - type=11, - cpp_type=10, - label=1, - has_default_value=False, - default_value=None, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[_ANNOTATEDDATUM_ANNOTATIONTYPE, ], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=1480, + serialized_end=1655, ) + +_FILLERPARAMETER = _descriptor.Descriptor( + name='FillerParameter', + full_name='caffe.FillerParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ _descriptor.FieldDescriptor( - name='flatten_param', - full_name='caffe.LayerParameter.flatten_param', - index=31, - number=135, - type=11, - cpp_type=10, + name='type', + full_name='caffe.FillerParameter.type', + index=0, + number=1, + type=9, + cpp_type=9, label=1, - has_default_value=False, - default_value=None, + has_default_value=True, + default_value=_b("constant").decode('utf-8'), message_type=None, enum_type=None, containing_type=None, @@ -4350,15 +2328,15 @@ _LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='hdf5_data_param', - full_name='caffe.LayerParameter.hdf5_data_param', - index=32, - number=112, - type=11, - cpp_type=10, + name='value', + full_name='caffe.FillerParameter.value', + index=1, + number=2, + type=2, + cpp_type=6, label=1, - has_default_value=False, - default_value=None, + has_default_value=True, + default_value=float(0), message_type=None, enum_type=None, containing_type=None, @@ -4367,15 +2345,15 @@ _LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='hdf5_output_param', - full_name='caffe.LayerParameter.hdf5_output_param', - index=33, - number=113, - type=11, - cpp_type=10, + name='min', + full_name='caffe.FillerParameter.min', + index=2, + number=3, + type=2, + cpp_type=6, label=1, - has_default_value=False, - default_value=None, + has_default_value=True, + default_value=float(0), message_type=None, enum_type=None, containing_type=None, @@ -4384,15 +2362,15 @@ _LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='hinge_loss_param', - full_name='caffe.LayerParameter.hinge_loss_param', - index=34, - number=114, - type=11, - cpp_type=10, + name='max', + full_name='caffe.FillerParameter.max', + index=3, + number=4, + type=2, + cpp_type=6, label=1, - has_default_value=False, - default_value=None, + has_default_value=True, + default_value=float(1), message_type=None, enum_type=None, containing_type=None, @@ -4401,15 +2379,15 @@ _LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='image_data_param', - full_name='caffe.LayerParameter.image_data_param', - index=35, - number=115, - type=11, - cpp_type=10, + name='mean', + full_name='caffe.FillerParameter.mean', + index=4, + number=5, + type=2, + cpp_type=6, label=1, - has_default_value=False, - default_value=None, + has_default_value=True, + default_value=float(0), message_type=None, enum_type=None, containing_type=None, @@ -4418,15 +2396,15 @@ _LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='infogain_loss_param', - full_name='caffe.LayerParameter.infogain_loss_param', - index=36, - number=116, - type=11, - cpp_type=10, + name='std', + full_name='caffe.FillerParameter.std', + index=5, + number=6, + type=2, + cpp_type=6, label=1, - has_default_value=False, - default_value=None, + has_default_value=True, + default_value=float(1), message_type=None, enum_type=None, containing_type=None, @@ -4435,15 +2413,15 @@ _LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='inner_product_param', - full_name='caffe.LayerParameter.inner_product_param', - index=37, - number=117, - type=11, - cpp_type=10, + name='sparse', + full_name='caffe.FillerParameter.sparse', + index=6, + number=7, + type=5, + cpp_type=1, label=1, - has_default_value=False, - default_value=None, + has_default_value=True, + default_value=-1, message_type=None, enum_type=None, containing_type=None, @@ -4452,15 +2430,15 @@ _LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='input_param', - full_name='caffe.LayerParameter.input_param', - index=38, - number=143, - type=11, - cpp_type=10, + name='variance_norm', + full_name='caffe.FillerParameter.variance_norm', + index=7, + number=8, + type=14, + cpp_type=8, label=1, - has_default_value=False, - default_value=None, + has_default_value=True, + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -4468,48 +2446,35 @@ _LAYERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='log_param', - full_name='caffe.LayerParameter.log_param', - index=39, - number=134, - type=11, - cpp_type=10, - label=1, - has_default_value=False, - default_value=None, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='lrn_param', - full_name='caffe.LayerParameter.lrn_param', - index=40, - number=118, - type=11, - cpp_type=10, - label=1, - has_default_value=False, - default_value=None, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[_FILLERPARAMETER_VARIANCENORM, ], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=1658, + serialized_end=1924, ) + +_NETPARAMETER = _descriptor.Descriptor( + name='NetParameter', + full_name='caffe.NetParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ _descriptor.FieldDescriptor( - name='memory_data_param', - full_name='caffe.LayerParameter.memory_data_param', - index=41, - number=119, - type=11, - cpp_type=10, + name='name', + full_name='caffe.NetParameter.name', + index=0, + number=1, + type=9, + cpp_type=9, label=1, has_default_value=False, - default_value=None, + default_value=_b("").decode('utf-8'), message_type=None, enum_type=None, containing_type=None, @@ -4518,15 +2483,15 @@ _LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='multibox_loss_param', - full_name='caffe.LayerParameter.multibox_loss_param', - index=42, - number=201, - type=11, - cpp_type=10, - label=1, + name='input', + full_name='caffe.NetParameter.input', + index=1, + number=3, + type=9, + cpp_type=9, + label=3, has_default_value=False, - default_value=None, + default_value=[], message_type=None, enum_type=None, containing_type=None, @@ -4534,48 +2499,16 @@ _LAYERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='mvn_param', - full_name='caffe.LayerParameter.mvn_param', - index=43, - number=120, - type=11, - cpp_type=10, - label=1, - has_default_value=False, - default_value=None, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='norm_param', - full_name='caffe.LayerParameter.norm_param', - index=44, - number=206, - type=11, - cpp_type=10, - label=1, - has_default_value=False, - default_value=None, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='parameter_param', - full_name='caffe.LayerParameter.parameter_param', - index=45, - number=145, + name='input_shape', + full_name='caffe.NetParameter.input_shape', + index=2, + number=8, type=11, cpp_type=10, - label=1, + label=3, has_default_value=False, - default_value=None, + default_value=[], message_type=None, enum_type=None, containing_type=None, @@ -4584,15 +2517,15 @@ _LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='permute_param', - full_name='caffe.LayerParameter.permute_param', - index=46, - number=202, - type=11, - cpp_type=10, - label=1, + name='input_dim', + full_name='caffe.NetParameter.input_dim', + index=3, + number=4, + type=5, + cpp_type=1, + label=3, has_default_value=False, - default_value=None, + default_value=[], message_type=None, enum_type=None, containing_type=None, @@ -4601,15 +2534,15 @@ _LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='pooling_param', - full_name='caffe.LayerParameter.pooling_param', - index=47, - number=121, - type=11, - cpp_type=10, + name='force_backward', + full_name='caffe.NetParameter.force_backward', + index=4, + number=5, + type=8, + cpp_type=7, label=1, - has_default_value=False, - default_value=None, + has_default_value=True, + default_value=False, message_type=None, enum_type=None, containing_type=None, @@ -4618,10 +2551,10 @@ _LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='power_param', - full_name='caffe.LayerParameter.power_param', - index=48, - number=122, + name='state', + full_name='caffe.NetParameter.state', + index=5, + number=6, type=11, cpp_type=10, label=1, @@ -4635,15 +2568,15 @@ _LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='prelu_param', - full_name='caffe.LayerParameter.prelu_param', - index=49, - number=131, - type=11, - cpp_type=10, + name='debug_info', + full_name='caffe.NetParameter.debug_info', + index=6, + number=7, + type=8, + cpp_type=7, label=1, - has_default_value=False, - default_value=None, + has_default_value=True, + default_value=False, message_type=None, enum_type=None, containing_type=None, @@ -4652,15 +2585,15 @@ _LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='prior_box_param', - full_name='caffe.LayerParameter.prior_box_param', - index=50, - number=203, + name='layer', + full_name='caffe.NetParameter.layer', + index=7, + number=100, type=11, cpp_type=10, - label=1, + label=3, has_default_value=False, - default_value=None, + default_value=[], message_type=None, enum_type=None, containing_type=None, @@ -4669,15 +2602,15 @@ _LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='python_param', - full_name='caffe.LayerParameter.python_param', - index=51, - number=130, + name='layers', + full_name='caffe.NetParameter.layers', + index=8, + number=2, type=11, cpp_type=10, - label=1, + label=3, has_default_value=False, - default_value=None, + default_value=[], message_type=None, enum_type=None, containing_type=None, @@ -4685,16 +2618,35 @@ _LAYERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=1927, + serialized_end=2197, ) + +_SOLVERPARAMETER = _descriptor.Descriptor( + name='SolverParameter', + full_name='caffe.SolverParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ _descriptor.FieldDescriptor( - name='recurrent_param', - full_name='caffe.LayerParameter.recurrent_param', - index=52, - number=146, - type=11, - cpp_type=10, + name='net', + full_name='caffe.SolverParameter.net', + index=0, + number=24, + type=9, + cpp_type=9, label=1, has_default_value=False, - default_value=None, + default_value=_b("").decode('utf-8'), message_type=None, enum_type=None, containing_type=None, @@ -4703,10 +2655,10 @@ _LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='reduction_param', - full_name='caffe.LayerParameter.reduction_param', - index=53, - number=136, + name='net_param', + full_name='caffe.SolverParameter.net_param', + index=1, + number=25, type=11, cpp_type=10, label=1, @@ -4719,32 +2671,16 @@ _LAYERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='relu_param', - full_name='caffe.LayerParameter.relu_param', - index=54, - number=123, - type=11, - cpp_type=10, - label=1, - has_default_value=False, - default_value=None, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='reshape_param', - full_name='caffe.LayerParameter.reshape_param', - index=55, - number=133, - type=11, - cpp_type=10, + name='train_net', + full_name='caffe.SolverParameter.train_net', + index=2, + number=1, + type=9, + cpp_type=9, label=1, has_default_value=False, - default_value=None, + default_value=_b("").decode('utf-8'), message_type=None, enum_type=None, containing_type=None, @@ -4753,15 +2689,15 @@ _LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='scale_param', - full_name='caffe.LayerParameter.scale_param', - index=56, - number=142, - type=11, - cpp_type=10, - label=1, + name='test_net', + full_name='caffe.SolverParameter.test_net', + index=3, + number=2, + type=9, + cpp_type=9, + label=3, has_default_value=False, - default_value=None, + default_value=[], message_type=None, enum_type=None, containing_type=None, @@ -4770,10 +2706,10 @@ _LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='shuffle_channel_param', - full_name='caffe.LayerParameter.shuffle_channel_param', - index=57, - number=208, + name='train_net_param', + full_name='caffe.SolverParameter.train_net_param', + index=4, + number=21, type=11, cpp_type=10, label=1, @@ -4787,15 +2723,15 @@ _LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='sigmoid_param', - full_name='caffe.LayerParameter.sigmoid_param', - index=58, - number=124, + name='test_net_param', + full_name='caffe.SolverParameter.test_net_param', + index=5, + number=22, type=11, cpp_type=10, - label=1, + label=3, has_default_value=False, - default_value=None, + default_value=[], message_type=None, enum_type=None, containing_type=None, @@ -4804,10 +2740,10 @@ _LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='softmax_param', - full_name='caffe.LayerParameter.softmax_param', - index=59, - number=125, + name='train_state', + full_name='caffe.SolverParameter.train_state', + index=6, + number=26, type=11, cpp_type=10, label=1, @@ -4820,32 +2756,16 @@ _LAYERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='spp_param', - full_name='caffe.LayerParameter.spp_param', - index=60, - number=132, - type=11, - cpp_type=10, - label=1, - has_default_value=False, - default_value=None, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='slice_param', - full_name='caffe.LayerParameter.slice_param', - index=61, - number=126, + name='test_state', + full_name='caffe.SolverParameter.test_state', + index=7, + number=27, type=11, cpp_type=10, - label=1, + label=3, has_default_value=False, - default_value=None, + default_value=[], message_type=None, enum_type=None, containing_type=None, @@ -4853,32 +2773,16 @@ _LAYERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='tanh_param', - full_name='caffe.LayerParameter.tanh_param', - index=62, - number=127, - type=11, - cpp_type=10, - label=1, - has_default_value=False, - default_value=None, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='threshold_param', - full_name='caffe.LayerParameter.threshold_param', - index=63, - number=128, - type=11, - cpp_type=10, + name='eval_type', + full_name='caffe.SolverParameter.eval_type', + index=8, + number=41, + type=9, + cpp_type=9, label=1, - has_default_value=False, - default_value=None, + has_default_value=True, + default_value=_b("classification").decode('utf-8'), message_type=None, enum_type=None, containing_type=None, @@ -4886,32 +2790,16 @@ _LAYERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='tile_param', - full_name='caffe.LayerParameter.tile_param', - index=64, - number=138, - type=11, - cpp_type=10, - label=1, - has_default_value=False, - default_value=None, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='video_data_param', - full_name='caffe.LayerParameter.video_data_param', - index=65, - number=207, - type=11, - cpp_type=10, + name='ap_version', + full_name='caffe.SolverParameter.ap_version', + index=9, + number=42, + type=9, + cpp_type=9, label=1, - has_default_value=False, - default_value=None, + has_default_value=True, + default_value=_b("Integral").decode('utf-8'), message_type=None, enum_type=None, containing_type=None, @@ -4920,15 +2808,32 @@ _LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='window_data_param', - full_name='caffe.LayerParameter.window_data_param', - index=66, - number=129, - type=11, - cpp_type=10, + name='show_per_class_result', + full_name='caffe.SolverParameter.show_per_class_result', + index=10, + number=44, + type=8, + cpp_type=7, label=1, + has_default_value=True, + default_value=False, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='test_iter', + full_name='caffe.SolverParameter.test_iter', + index=11, + number=3, + type=5, + cpp_type=1, + label=3, has_default_value=False, - default_value=None, + default_value=[], message_type=None, enum_type=None, containing_type=None, @@ -4936,36 +2841,16 @@ _LAYERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=4176, - serialized_end=7263, -) - -_TRANSFORMATIONPARAMETER = _descriptor.Descriptor( - name='TransformationParameter', - full_name='caffe.TransformationParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ _descriptor.FieldDescriptor( - name='scale', - full_name='caffe.TransformationParameter.scale', - index=0, - number=1, - type=2, - cpp_type=6, + name='test_interval', + full_name='caffe.SolverParameter.test_interval', + index=12, + number=4, + type=5, + cpp_type=1, label=1, has_default_value=True, - default_value=float(1), + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -4974,10 +2859,10 @@ _TRANSFORMATIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='mirror', - full_name='caffe.TransformationParameter.mirror', - index=1, - number=2, + name='test_compute_loss', + full_name='caffe.SolverParameter.test_compute_loss', + index=13, + number=19, type=8, cpp_type=7, label=1, @@ -4991,15 +2876,15 @@ _TRANSFORMATIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='crop_size', - full_name='caffe.TransformationParameter.crop_size', - index=2, - number=3, - type=13, - cpp_type=3, + name='test_initialization', + full_name='caffe.SolverParameter.test_initialization', + index=14, + number=32, + type=8, + cpp_type=7, label=1, has_default_value=True, - default_value=0, + default_value=True, message_type=None, enum_type=None, containing_type=None, @@ -5008,15 +2893,15 @@ _TRANSFORMATIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='crop_h', - full_name='caffe.TransformationParameter.crop_h', - index=3, - number=11, - type=13, - cpp_type=3, + name='base_lr', + full_name='caffe.SolverParameter.base_lr', + index=15, + number=5, + type=2, + cpp_type=6, label=1, - has_default_value=True, - default_value=0, + has_default_value=False, + default_value=float(0), message_type=None, enum_type=None, containing_type=None, @@ -5025,14 +2910,14 @@ _TRANSFORMATIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='crop_w', - full_name='caffe.TransformationParameter.crop_w', - index=4, - number=12, - type=13, - cpp_type=3, + name='display', + full_name='caffe.SolverParameter.display', + index=16, + number=6, + type=5, + cpp_type=1, label=1, - has_default_value=True, + has_default_value=False, default_value=0, message_type=None, enum_type=None, @@ -5042,15 +2927,15 @@ _TRANSFORMATIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='mean_file', - full_name='caffe.TransformationParameter.mean_file', - index=5, - number=4, - type=9, - cpp_type=9, + name='average_loss', + full_name='caffe.SolverParameter.average_loss', + index=17, + number=33, + type=5, + cpp_type=1, label=1, - has_default_value=False, - default_value=_b("").decode('utf-8'), + has_default_value=True, + default_value=1, message_type=None, enum_type=None, containing_type=None, @@ -5059,15 +2944,15 @@ _TRANSFORMATIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='mean_value', - full_name='caffe.TransformationParameter.mean_value', - index=6, - number=5, - type=2, - cpp_type=6, - label=3, + name='max_iter', + full_name='caffe.SolverParameter.max_iter', + index=18, + number=7, + type=5, + cpp_type=1, + label=1, has_default_value=False, - default_value=[], + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -5076,15 +2961,15 @@ _TRANSFORMATIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='force_color', - full_name='caffe.TransformationParameter.force_color', - index=7, - number=6, - type=8, - cpp_type=7, + name='iter_size', + full_name='caffe.SolverParameter.iter_size', + index=19, + number=36, + type=5, + cpp_type=1, label=1, has_default_value=True, - default_value=False, + default_value=1, message_type=None, enum_type=None, containing_type=None, @@ -5093,15 +2978,15 @@ _TRANSFORMATIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='force_gray', - full_name='caffe.TransformationParameter.force_gray', - index=8, - number=7, - type=8, - cpp_type=7, + name='lr_policy', + full_name='caffe.SolverParameter.lr_policy', + index=20, + number=8, + type=9, + cpp_type=9, label=1, - has_default_value=True, - default_value=False, + has_default_value=False, + default_value=_b("").decode('utf-8'), message_type=None, enum_type=None, containing_type=None, @@ -5110,15 +2995,15 @@ _TRANSFORMATIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='resize_param', - full_name='caffe.TransformationParameter.resize_param', - index=9, - number=8, - type=11, - cpp_type=10, + name='gamma', + full_name='caffe.SolverParameter.gamma', + index=21, + number=9, + type=2, + cpp_type=6, label=1, has_default_value=False, - default_value=None, + default_value=float(0), message_type=None, enum_type=None, containing_type=None, @@ -5127,15 +3012,15 @@ _TRANSFORMATIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='noise_param', - full_name='caffe.TransformationParameter.noise_param', - index=10, - number=9, - type=11, - cpp_type=10, + name='power', + full_name='caffe.SolverParameter.power', + index=22, + number=10, + type=2, + cpp_type=6, label=1, has_default_value=False, - default_value=None, + default_value=float(0), message_type=None, enum_type=None, containing_type=None, @@ -5144,15 +3029,15 @@ _TRANSFORMATIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='distort_param', - full_name='caffe.TransformationParameter.distort_param', - index=11, - number=13, - type=11, - cpp_type=10, + name='momentum', + full_name='caffe.SolverParameter.momentum', + index=23, + number=11, + type=2, + cpp_type=6, label=1, has_default_value=False, - default_value=None, + default_value=float(0), message_type=None, enum_type=None, containing_type=None, @@ -5161,15 +3046,15 @@ _TRANSFORMATIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='expand_param', - full_name='caffe.TransformationParameter.expand_param', - index=12, - number=14, - type=11, - cpp_type=10, + name='weight_decay', + full_name='caffe.SolverParameter.weight_decay', + index=24, + number=12, + type=2, + cpp_type=6, label=1, has_default_value=False, - default_value=None, + default_value=float(0), message_type=None, enum_type=None, containing_type=None, @@ -5178,15 +3063,15 @@ _TRANSFORMATIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='emit_constraint', - full_name='caffe.TransformationParameter.emit_constraint', - index=13, - number=10, - type=11, - cpp_type=10, + name='regularization_type', + full_name='caffe.SolverParameter.regularization_type', + index=25, + number=29, + type=9, + cpp_type=9, label=1, - has_default_value=False, - default_value=None, + has_default_value=True, + default_value=_b("L2").decode('utf-8'), message_type=None, enum_type=None, containing_type=None, @@ -5194,52 +3079,16 @@ _TRANSFORMATIONPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=7266, - serialized_end=7724, -) - -_RESIZEPARAMETER = _descriptor.Descriptor( - name='ResizeParameter', - full_name='caffe.ResizeParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='prob', - full_name='caffe.ResizeParameter.prob', - index=0, - number=1, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(1), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='resize_mode', - full_name='caffe.ResizeParameter.resize_mode', - index=1, - number=2, - type=14, - cpp_type=8, + name='stepsize', + full_name='caffe.SolverParameter.stepsize', + index=26, + number=13, + type=5, + cpp_type=1, label=1, - has_default_value=True, - default_value=1, + has_default_value=False, + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -5247,48 +3096,16 @@ _RESIZEPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='height', - full_name='caffe.ResizeParameter.height', - index=2, - number=3, - type=13, - cpp_type=3, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='width', - full_name='caffe.ResizeParameter.width', - index=3, - number=4, - type=13, - cpp_type=3, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='height_scale', - full_name='caffe.ResizeParameter.height_scale', - index=4, - number=8, - type=13, - cpp_type=3, - label=1, - has_default_value=True, - default_value=0, + name='stepvalue', + full_name='caffe.SolverParameter.stepvalue', + index=27, + number=34, + type=5, + cpp_type=1, + label=3, + has_default_value=False, + default_value=[], message_type=None, enum_type=None, containing_type=None, @@ -5297,15 +3114,15 @@ _RESIZEPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='width_scale', - full_name='caffe.ResizeParameter.width_scale', - index=5, - number=9, - type=13, - cpp_type=3, - label=1, - has_default_value=True, - default_value=0, + name='plateau_winsize', + full_name='caffe.SolverParameter.plateau_winsize', + index=28, + number=43, + type=5, + cpp_type=1, + label=3, + has_default_value=False, + default_value=[], message_type=None, enum_type=None, containing_type=None, @@ -5313,48 +3130,16 @@ _RESIZEPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='pad_mode', - full_name='caffe.ResizeParameter.pad_mode', - index=6, - number=5, - type=14, - cpp_type=8, - label=1, - has_default_value=True, - default_value=1, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='pad_value', - full_name='caffe.ResizeParameter.pad_value', - index=7, - number=6, - type=2, - cpp_type=6, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='interp_mode', - full_name='caffe.ResizeParameter.interp_mode', - index=8, - number=7, - type=14, - cpp_type=8, - label=3, - has_default_value=False, - default_value=[], + name='clip_gradients', + full_name='caffe.SolverParameter.clip_gradients', + index=29, + number=35, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(-1), message_type=None, enum_type=None, containing_type=None, @@ -5362,40 +3147,16 @@ _RESIZEPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[ - _RESIZEPARAMETER_RESIZE_MODE, - _RESIZEPARAMETER_PAD_MODE, - _RESIZEPARAMETER_INTERP_MODE, - ], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=7727, - serialized_end=8255, -) - -_SALTPEPPERPARAMETER = _descriptor.Descriptor( - name='SaltPepperParameter', - full_name='caffe.SaltPepperParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ _descriptor.FieldDescriptor( - name='fraction', - full_name='caffe.SaltPepperParameter.fraction', - index=0, - number=1, - type=2, - cpp_type=6, + name='snapshot', + full_name='caffe.SolverParameter.snapshot', + index=30, + number=14, + type=5, + cpp_type=1, label=1, has_default_value=True, - default_value=float(0), + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -5403,196 +3164,16 @@ _SALTPEPPERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='value', - full_name='caffe.SaltPepperParameter.value', - index=1, - number=2, - type=2, - cpp_type=6, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=8257, - serialized_end=8314, -) - -_NOISEPARAMETER = _descriptor.Descriptor( - name='NoiseParameter', - full_name='caffe.NoiseParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='prob', - full_name='caffe.NoiseParameter.prob', - index=0, - number=1, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(0), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='hist_eq', - full_name='caffe.NoiseParameter.hist_eq', - index=1, - number=2, - type=8, - cpp_type=7, - label=1, - has_default_value=True, - default_value=False, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='inverse', - full_name='caffe.NoiseParameter.inverse', - index=2, - number=3, - type=8, - cpp_type=7, - label=1, - has_default_value=True, - default_value=False, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='decolorize', - full_name='caffe.NoiseParameter.decolorize', - index=3, - number=4, - type=8, - cpp_type=7, - label=1, - has_default_value=True, - default_value=False, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='gauss_blur', - full_name='caffe.NoiseParameter.gauss_blur', - index=4, - number=5, - type=8, - cpp_type=7, - label=1, - has_default_value=True, - default_value=False, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='jpeg', - full_name='caffe.NoiseParameter.jpeg', - index=5, - number=6, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(-1), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='posterize', - full_name='caffe.NoiseParameter.posterize', - index=6, - number=7, - type=8, - cpp_type=7, - label=1, - has_default_value=True, - default_value=False, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='erode', - full_name='caffe.NoiseParameter.erode', - index=7, - number=8, - type=8, - cpp_type=7, - label=1, - has_default_value=True, - default_value=False, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='saltpepper', - full_name='caffe.NoiseParameter.saltpepper', - index=8, - number=9, - type=8, - cpp_type=7, - label=1, - has_default_value=True, - default_value=False, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='saltpepper_param', - full_name='caffe.NoiseParameter.saltpepper_param', - index=9, - number=10, - type=11, - cpp_type=10, + name='snapshot_prefix', + full_name='caffe.SolverParameter.snapshot_prefix', + index=31, + number=15, + type=9, + cpp_type=9, label=1, has_default_value=False, - default_value=None, + default_value=_b("").decode('utf-8'), message_type=None, enum_type=None, containing_type=None, @@ -5600,27 +3181,11 @@ _NOISEPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='clahe', - full_name='caffe.NoiseParameter.clahe', - index=10, - number=11, - type=8, - cpp_type=7, - label=1, - has_default_value=True, - default_value=False, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='convert_to_hsv', - full_name='caffe.NoiseParameter.convert_to_hsv', - index=11, - number=12, + name='snapshot_diff', + full_name='caffe.SolverParameter.snapshot_diff', + index=32, + number=16, type=8, cpp_type=7, label=1, @@ -5634,15 +3199,15 @@ _NOISEPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='convert_to_lab', - full_name='caffe.NoiseParameter.convert_to_lab', - index=12, - number=13, - type=8, - cpp_type=7, + name='snapshot_format', + full_name='caffe.SolverParameter.snapshot_format', + index=33, + number=37, + type=14, + cpp_type=8, label=1, has_default_value=True, - default_value=False, + default_value=1, message_type=None, enum_type=None, containing_type=None, @@ -5650,36 +3215,16 @@ _NOISEPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=8317, - serialized_end=8683, -) - -_DISTORTIONPARAMETER = _descriptor.Descriptor( - name='DistortionParameter', - full_name='caffe.DistortionParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ _descriptor.FieldDescriptor( - name='brightness_prob', - full_name='caffe.DistortionParameter.brightness_prob', - index=0, - number=1, - type=2, - cpp_type=6, + name='solver_mode', + full_name='caffe.SolverParameter.solver_mode', + index=34, + number=17, + type=14, + cpp_type=8, label=1, has_default_value=True, - default_value=float(0), + default_value=1, message_type=None, enum_type=None, containing_type=None, @@ -5688,15 +3233,15 @@ _DISTORTIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='brightness_delta', - full_name='caffe.DistortionParameter.brightness_delta', - index=1, - number=2, - type=2, - cpp_type=6, + name='device_id', + full_name='caffe.SolverParameter.device_id', + index=35, + number=18, + type=5, + cpp_type=1, label=1, has_default_value=True, - default_value=float(0), + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -5705,15 +3250,15 @@ _DISTORTIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='contrast_prob', - full_name='caffe.DistortionParameter.contrast_prob', - index=2, - number=3, - type=2, - cpp_type=6, + name='random_seed', + full_name='caffe.SolverParameter.random_seed', + index=36, + number=20, + type=3, + cpp_type=2, label=1, has_default_value=True, - default_value=float(0), + default_value=-1, message_type=None, enum_type=None, containing_type=None, @@ -5722,15 +3267,15 @@ _DISTORTIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='contrast_lower', - full_name='caffe.DistortionParameter.contrast_lower', - index=3, - number=4, - type=2, - cpp_type=6, + name='type', + full_name='caffe.SolverParameter.type', + index=37, + number=40, + type=9, + cpp_type=9, label=1, has_default_value=True, - default_value=float(0), + default_value=_b("SGD").decode('utf-8'), message_type=None, enum_type=None, containing_type=None, @@ -5739,15 +3284,15 @@ _DISTORTIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='contrast_upper', - full_name='caffe.DistortionParameter.contrast_upper', - index=4, - number=5, + name='delta', + full_name='caffe.SolverParameter.delta', + index=38, + number=31, type=2, cpp_type=6, label=1, has_default_value=True, - default_value=float(0), + default_value=float(1e-08), message_type=None, enum_type=None, containing_type=None, @@ -5756,15 +3301,15 @@ _DISTORTIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='hue_prob', - full_name='caffe.DistortionParameter.hue_prob', - index=5, - number=6, + name='momentum2', + full_name='caffe.SolverParameter.momentum2', + index=39, + number=39, type=2, cpp_type=6, label=1, has_default_value=True, - default_value=float(0), + default_value=float(0.999), message_type=None, enum_type=None, containing_type=None, @@ -5773,15 +3318,15 @@ _DISTORTIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='hue_delta', - full_name='caffe.DistortionParameter.hue_delta', - index=6, - number=7, + name='rms_decay', + full_name='caffe.SolverParameter.rms_decay', + index=40, + number=38, type=2, cpp_type=6, label=1, has_default_value=True, - default_value=float(0), + default_value=float(0.99), message_type=None, enum_type=None, containing_type=None, @@ -5790,32 +3335,15 @@ _DISTORTIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='saturation_prob', - full_name='caffe.DistortionParameter.saturation_prob', - index=7, - number=8, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(0), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor( - name='saturation_lower', - full_name='caffe.DistortionParameter.saturation_lower', - index=8, - number=9, - type=2, - cpp_type=6, + name='debug_info', + full_name='caffe.SolverParameter.debug_info', + index=41, + number=23, + type=8, + cpp_type=7, label=1, has_default_value=True, - default_value=float(0), + default_value=False, message_type=None, enum_type=None, containing_type=None, @@ -5824,15 +3352,15 @@ _DISTORTIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='saturation_upper', - full_name='caffe.DistortionParameter.saturation_upper', - index=9, - number=10, - type=2, - cpp_type=6, + name='snapshot_after_train', + full_name='caffe.SolverParameter.snapshot_after_train', + index=42, + number=28, + type=8, + cpp_type=7, label=1, has_default_value=True, - default_value=float(0), + default_value=True, message_type=None, enum_type=None, containing_type=None, @@ -5841,15 +3369,15 @@ _DISTORTIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='random_order_prob', - full_name='caffe.DistortionParameter.random_order_prob', - index=10, - number=11, - type=2, - cpp_type=6, + name='solver_type', + full_name='caffe.SolverParameter.solver_type', + index=43, + number=30, + type=14, + cpp_type=8, label=1, has_default_value=True, - default_value=float(0), + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -5860,49 +3388,53 @@ _DISTORTIONPARAMETER = _descriptor.Descriptor( ], extensions=[], nested_types=[], - enum_types=[], + enum_types=[ + _SOLVERPARAMETER_SNAPSHOTFORMAT, + _SOLVERPARAMETER_SOLVERMODE, + _SOLVERPARAMETER_SOLVERTYPE, + ], serialized_options=None, is_extendable=False, syntax='proto2', extension_ranges=[], oneofs=[], - serialized_start=8686, - serialized_end=9003, -) + serialized_start=2200, + serialized_end=3642, ) -_EXPANSIONPARAMETER = _descriptor.Descriptor( - name='ExpansionParameter', - full_name='caffe.ExpansionParameter', +_SOLVERSTATE = _descriptor.Descriptor( + name='SolverState', + full_name='caffe.SolverState', filename=None, file=DESCRIPTOR, containing_type=None, fields=[ - _descriptor.FieldDescriptor(name='prob', - full_name='caffe.ExpansionParameter.prob', - index=0, - number=1, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(1), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='max_expand_ratio', - full_name='caffe.ExpansionParameter.max_expand_ratio', + name='iter', + full_name='caffe.SolverState.iter', + index=0, + number=1, + type=5, + cpp_type=1, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='learned_net', + full_name='caffe.SolverState.learned_net', index=1, number=2, - type=2, - cpp_type=6, + type=9, + cpp_type=9, label=1, - has_default_value=True, - default_value=float(1), + has_default_value=False, + default_value=_b("").decode('utf-8'), message_type=None, enum_type=None, containing_type=None, @@ -5910,35 +3442,32 @@ _EXPANSIONPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=9005, - serialized_end=9071, -) - -_LOSSPARAMETER = _descriptor.Descriptor( - name='LossParameter', - full_name='caffe.LossParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ _descriptor.FieldDescriptor( - name='ignore_label', - full_name='caffe.LossParameter.ignore_label', - index=0, - number=1, + name='history', + full_name='caffe.SolverState.history', + index=2, + number=3, + type=11, + cpp_type=10, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='current_step', + full_name='caffe.SolverState.current_step', + index=3, + number=4, type=5, cpp_type=1, label=1, - has_default_value=False, + has_default_value=True, default_value=0, message_type=None, enum_type=None, @@ -5948,15 +3477,15 @@ _LOSSPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='normalization', - full_name='caffe.LossParameter.normalization', - index=1, - number=3, - type=14, - cpp_type=8, + name='minimum_loss', + full_name='caffe.SolverState.minimum_loss', + index=4, + number=5, + type=2, + cpp_type=6, label=1, has_default_value=True, - default_value=1, + default_value=float(1e+38), message_type=None, enum_type=None, containing_type=None, @@ -5964,85 +3493,15 @@ _LOSSPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='normalize', - full_name='caffe.LossParameter.normalize', - index=2, - number=2, - type=8, - cpp_type=7, - label=1, - has_default_value=False, - default_value=False, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[ - _LOSSPARAMETER_NORMALIZATIONMODE, - ], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=9074, - serialized_end=9268, -) - -_ACCURACYPARAMETER = _descriptor.Descriptor( - name='AccuracyParameter', - full_name='caffe.AccuracyParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='top_k', - full_name='caffe.AccuracyParameter.top_k', - index=0, - number=1, - type=13, - cpp_type=3, - label=1, - has_default_value=True, - default_value=1, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='axis', - full_name='caffe.AccuracyParameter.axis', - index=1, - number=2, - type=5, - cpp_type=1, - label=1, - has_default_value=True, - default_value=1, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='ignore_label', - full_name='caffe.AccuracyParameter.ignore_label', - index=2, - number=3, + name='iter_last_event', + full_name='caffe.SolverState.iter_last_event', + index=5, + number=6, type=5, cpp_type=1, label=1, - has_default_value=False, + has_default_value=True, default_value=0, message_type=None, enum_type=None, @@ -6060,27 +3519,26 @@ _ACCURACYPARAMETER = _descriptor.Descriptor( syntax='proto2', extension_ranges=[], oneofs=[], - serialized_start=9270, - serialized_end=9346, -) + serialized_start=3645, + serialized_end=3810, ) -_ANNOTATEDDATAPARAMETER = _descriptor.Descriptor( - name='AnnotatedDataParameter', - full_name='caffe.AnnotatedDataParameter', +_NETSTATE = _descriptor.Descriptor( + name='NetState', + full_name='caffe.NetState', filename=None, file=DESCRIPTOR, containing_type=None, fields=[ _descriptor.FieldDescriptor( - name='batch_sampler', - full_name='caffe.AnnotatedDataParameter.batch_sampler', + name='phase', + full_name='caffe.NetState.phase', index=0, number=1, - type=11, - cpp_type=10, - label=3, - has_default_value=False, - default_value=[], + type=14, + cpp_type=8, + label=1, + has_default_value=True, + default_value=1, message_type=None, enum_type=None, containing_type=None, @@ -6089,15 +3547,15 @@ _ANNOTATEDDATAPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='label_map_file', - full_name='caffe.AnnotatedDataParameter.label_map_file', + name='level', + full_name='caffe.NetState.level', index=1, number=2, - type=9, - cpp_type=9, + type=5, + cpp_type=1, label=1, - has_default_value=False, - default_value=_b("").decode('utf-8'), + has_default_value=True, + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -6106,15 +3564,15 @@ _ANNOTATEDDATAPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='anno_type', - full_name='caffe.AnnotatedDataParameter.anno_type', + name='stage', + full_name='caffe.NetState.stage', index=2, number=3, - type=14, - cpp_type=8, - label=1, + type=9, + cpp_type=9, + label=3, has_default_value=False, - default_value=0, + default_value=[], message_type=None, enum_type=None, containing_type=None, @@ -6131,27 +3589,26 @@ _ANNOTATEDDATAPARAMETER = _descriptor.Descriptor( syntax='proto2', extension_ranges=[], oneofs=[], - serialized_start=9349, - serialized_end=9498, -) + serialized_start=3812, + serialized_end=3890, ) -_ARGMAXPARAMETER = _descriptor.Descriptor( - name='ArgMaxParameter', - full_name='caffe.ArgMaxParameter', +_NETSTATERULE = _descriptor.Descriptor( + name='NetStateRule', + full_name='caffe.NetStateRule', filename=None, file=DESCRIPTOR, containing_type=None, fields=[ _descriptor.FieldDescriptor( - name='out_max_val', - full_name='caffe.ArgMaxParameter.out_max_val', + name='phase', + full_name='caffe.NetStateRule.phase', index=0, number=1, - type=8, - cpp_type=7, + type=14, + cpp_type=8, label=1, - has_default_value=True, - default_value=False, + has_default_value=False, + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -6159,84 +3616,67 @@ _ARGMAXPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='top_k', - full_name='caffe.ArgMaxParameter.top_k', - index=1, - number=2, - type=13, - cpp_type=3, - label=1, - has_default_value=True, - default_value=1, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='axis', - full_name='caffe.ArgMaxParameter.axis', - index=2, - number=3, - type=5, - cpp_type=1, - label=1, - has_default_value=False, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=9500, - serialized_end=9577, -) - -_CONCATPARAMETER = _descriptor.Descriptor( - name='ConcatParameter', - full_name='caffe.ConcatParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='axis', - full_name='caffe.ConcatParameter.axis', - index=0, - number=2, - type=5, - cpp_type=1, - label=1, - has_default_value=True, - default_value=1, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='concat_dim', - full_name='caffe.ConcatParameter.concat_dim', + name='min_level', + full_name='caffe.NetStateRule.min_level', index=1, - number=1, - type=13, - cpp_type=3, + number=2, + type=5, + cpp_type=1, label=1, - has_default_value=True, - default_value=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_level', + full_name='caffe.NetStateRule.max_level', + index=2, + number=3, + type=5, + cpp_type=1, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='stage', + full_name='caffe.NetStateRule.stage', + index=3, + number=4, + type=9, + cpp_type=9, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='not_stage', + full_name='caffe.NetStateRule.not_stage', + index=4, + number=5, + type=9, + cpp_type=9, + label=3, + has_default_value=False, + default_value=[], message_type=None, enum_type=None, containing_type=None, @@ -6253,27 +3693,26 @@ _CONCATPARAMETER = _descriptor.Descriptor( syntax='proto2', extension_ranges=[], oneofs=[], - serialized_start=9579, - serialized_end=9636, -) + serialized_start=3892, + serialized_end=4007, ) -_BATCHNORMPARAMETER = _descriptor.Descriptor( - name='BatchNormParameter', - full_name='caffe.BatchNormParameter', +_PARAMSPEC = _descriptor.Descriptor( + name='ParamSpec', + full_name='caffe.ParamSpec', filename=None, file=DESCRIPTOR, containing_type=None, fields=[ _descriptor.FieldDescriptor( - name='use_global_stats', - full_name='caffe.BatchNormParameter.use_global_stats', + name='name', + full_name='caffe.ParamSpec.name', index=0, number=1, - type=8, - cpp_type=7, + type=9, + cpp_type=9, label=1, has_default_value=False, - default_value=False, + default_value=_b("").decode('utf-8'), message_type=None, enum_type=None, containing_type=None, @@ -6282,15 +3721,15 @@ _BATCHNORMPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='moving_average_fraction', - full_name='caffe.BatchNormParameter.moving_average_fraction', + name='share_mode', + full_name='caffe.ParamSpec.share_mode', index=1, number=2, - type=2, - cpp_type=6, + type=14, + cpp_type=8, label=1, - has_default_value=True, - default_value=float(0.999), + has_default_value=False, + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -6298,115 +3737,11 @@ _BATCHNORMPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='eps', - full_name='caffe.BatchNormParameter.eps', - index=2, - number=3, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(1e-05), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=9638, - serialized_end=9744, -) - -_BIASPARAMETER = _descriptor.Descriptor( - name='BiasParameter', - full_name='caffe.BiasParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='axis', - full_name='caffe.BiasParameter.axis', - index=0, - number=1, - type=5, - cpp_type=1, - label=1, - has_default_value=True, - default_value=1, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='num_axes', - full_name='caffe.BiasParameter.num_axes', - index=1, - number=2, - type=5, - cpp_type=1, - label=1, - has_default_value=True, - default_value=1, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='filler', - full_name='caffe.BiasParameter.filler', - index=2, - number=3, - type=11, - cpp_type=10, - label=1, - has_default_value=False, - default_value=None, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=9746, - serialized_end=9839, -) - -_CONTRASTIVELOSSPARAMETER = _descriptor.Descriptor( - name='ContrastiveLossParameter', - full_name='caffe.ContrastiveLossParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ _descriptor.FieldDescriptor( - name='margin', - full_name='caffe.ContrastiveLossParameter.margin', - index=0, - number=1, + name='lr_mult', + full_name='caffe.ParamSpec.lr_mult', + index=2, + number=3, type=2, cpp_type=6, label=1, @@ -6420,15 +3755,15 @@ _CONTRASTIVELOSSPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='legacy_version', - full_name='caffe.ContrastiveLossParameter.legacy_version', - index=1, - number=2, - type=8, - cpp_type=7, + name='decay_mult', + full_name='caffe.ParamSpec.decay_mult', + index=3, + number=4, + type=2, + cpp_type=6, label=1, has_default_value=True, - default_value=False, + default_value=float(1), message_type=None, enum_type=None, containing_type=None, @@ -6439,33 +3774,32 @@ _CONTRASTIVELOSSPARAMETER = _descriptor.Descriptor( ], extensions=[], nested_types=[], - enum_types=[], + enum_types=[_PARAMSPEC_DIMCHECKMODE, ], serialized_options=None, is_extendable=False, syntax='proto2', extension_ranges=[], oneofs=[], - serialized_start=9841, - serialized_end=9917, -) + serialized_start=4010, + serialized_end=4173, ) -_CONVOLUTIONPARAMETER = _descriptor.Descriptor( - name='ConvolutionParameter', - full_name='caffe.ConvolutionParameter', +_LAYERPARAMETER = _descriptor.Descriptor( + name='LayerParameter', + full_name='caffe.LayerParameter', filename=None, file=DESCRIPTOR, containing_type=None, fields=[ _descriptor.FieldDescriptor( - name='num_output', - full_name='caffe.ConvolutionParameter.num_output', + name='name', + full_name='caffe.LayerParameter.name', index=0, number=1, - type=13, - cpp_type=3, + type=9, + cpp_type=9, label=1, has_default_value=False, - default_value=0, + default_value=_b("").decode('utf-8'), message_type=None, enum_type=None, containing_type=None, @@ -6474,15 +3808,15 @@ _CONVOLUTIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='bias_term', - full_name='caffe.ConvolutionParameter.bias_term', + name='type', + full_name='caffe.LayerParameter.type', index=1, number=2, - type=8, - cpp_type=7, + type=9, + cpp_type=9, label=1, - has_default_value=True, - default_value=True, + has_default_value=False, + default_value=_b("").decode('utf-8'), message_type=None, enum_type=None, containing_type=None, @@ -6490,29 +3824,30 @@ _CONVOLUTIONPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='pad', - full_name='caffe.ConvolutionParameter.pad', - index=2, - number=3, - type=13, - cpp_type=3, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='kernel_size', - full_name='caffe.ConvolutionParameter.kernel_size', + name='bottom', + full_name='caffe.LayerParameter.bottom', + index=2, + number=3, + type=9, + cpp_type=9, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='top', + full_name='caffe.LayerParameter.top', index=3, number=4, - type=13, - cpp_type=3, + type=9, + cpp_type=9, label=3, has_default_value=False, default_value=[], @@ -6524,15 +3859,15 @@ _CONVOLUTIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='stride', - full_name='caffe.ConvolutionParameter.stride', + name='phase', + full_name='caffe.LayerParameter.phase', index=4, - number=6, - type=13, - cpp_type=3, - label=3, + number=10, + type=14, + cpp_type=8, + label=1, has_default_value=False, - default_value=[], + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -6541,12 +3876,12 @@ _CONVOLUTIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='dilation', - full_name='caffe.ConvolutionParameter.dilation', + name='loss_weight', + full_name='caffe.LayerParameter.loss_weight', index=5, - number=18, - type=13, - cpp_type=3, + number=5, + type=2, + cpp_type=6, label=3, has_default_value=False, default_value=[], @@ -6558,15 +3893,15 @@ _CONVOLUTIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='pad_h', - full_name='caffe.ConvolutionParameter.pad_h', + name='param', + full_name='caffe.LayerParameter.param', index=6, - number=9, - type=13, - cpp_type=3, - label=1, - has_default_value=True, - default_value=0, + number=6, + type=11, + cpp_type=10, + label=3, + has_default_value=False, + default_value=[], message_type=None, enum_type=None, containing_type=None, @@ -6575,15 +3910,15 @@ _CONVOLUTIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='pad_w', - full_name='caffe.ConvolutionParameter.pad_w', + name='blobs', + full_name='caffe.LayerParameter.blobs', index=7, - number=10, - type=13, - cpp_type=3, - label=1, - has_default_value=True, - default_value=0, + number=7, + type=11, + cpp_type=10, + label=3, + has_default_value=False, + default_value=[], message_type=None, enum_type=None, containing_type=None, @@ -6592,15 +3927,15 @@ _CONVOLUTIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='kernel_h', - full_name='caffe.ConvolutionParameter.kernel_h', + name='propagate_down', + full_name='caffe.LayerParameter.propagate_down', index=8, number=11, - type=13, - cpp_type=3, - label=1, + type=8, + cpp_type=7, + label=3, has_default_value=False, - default_value=0, + default_value=[], message_type=None, enum_type=None, containing_type=None, @@ -6609,15 +3944,15 @@ _CONVOLUTIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='kernel_w', - full_name='caffe.ConvolutionParameter.kernel_w', + name='include', + full_name='caffe.LayerParameter.include', index=9, - number=12, - type=13, - cpp_type=3, - label=1, + number=8, + type=11, + cpp_type=10, + label=3, has_default_value=False, - default_value=0, + default_value=[], message_type=None, enum_type=None, containing_type=None, @@ -6626,15 +3961,15 @@ _CONVOLUTIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='stride_h', - full_name='caffe.ConvolutionParameter.stride_h', + name='exclude', + full_name='caffe.LayerParameter.exclude', index=10, - number=13, - type=13, - cpp_type=3, - label=1, + number=9, + type=11, + cpp_type=10, + label=3, has_default_value=False, - default_value=0, + default_value=[], message_type=None, enum_type=None, containing_type=None, @@ -6643,15 +3978,15 @@ _CONVOLUTIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='stride_w', - full_name='caffe.ConvolutionParameter.stride_w', + name='transform_param', + full_name='caffe.LayerParameter.transform_param', index=11, - number=14, - type=13, - cpp_type=3, + number=100, + type=11, + cpp_type=10, label=1, has_default_value=False, - default_value=0, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -6660,15 +3995,15 @@ _CONVOLUTIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='group', - full_name='caffe.ConvolutionParameter.group', + name='loss_param', + full_name='caffe.LayerParameter.loss_param', index=12, - number=5, - type=13, - cpp_type=3, + number=101, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=1, + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -6677,10 +4012,10 @@ _CONVOLUTIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='weight_filler', - full_name='caffe.ConvolutionParameter.weight_filler', + name='accuracy_param', + full_name='caffe.LayerParameter.accuracy_param', index=13, - number=7, + number=102, type=11, cpp_type=10, label=1, @@ -6694,10 +4029,10 @@ _CONVOLUTIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='bias_filler', - full_name='caffe.ConvolutionParameter.bias_filler', + name='annotated_data_param', + full_name='caffe.LayerParameter.annotated_data_param', index=14, - number=8, + number=200, type=11, cpp_type=10, label=1, @@ -6711,15 +4046,15 @@ _CONVOLUTIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='engine', - full_name='caffe.ConvolutionParameter.engine', + name='argmax_param', + full_name='caffe.LayerParameter.argmax_param', index=15, - number=15, - type=14, - cpp_type=8, + number=103, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=0, + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -6727,32 +4062,16 @@ _CONVOLUTIONPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='axis', - full_name='caffe.ConvolutionParameter.axis', - index=16, - number=16, - type=5, - cpp_type=1, - label=1, - has_default_value=True, - default_value=1, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='force_nd_im2col', - full_name='caffe.ConvolutionParameter.force_nd_im2col', - index=17, - number=17, - type=8, - cpp_type=7, + name='batch_norm_param', + full_name='caffe.LayerParameter.batch_norm_param', + index=16, + number=139, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=False, + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -6760,218 +4079,16 @@ _CONVOLUTIONPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[ - _CONVOLUTIONPARAMETER_ENGINE, - ], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=9920, - serialized_end=10428, -) - -_CROPPARAMETER = _descriptor.Descriptor( - name='CropParameter', - full_name='caffe.CropParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='axis', - full_name='caffe.CropParameter.axis', - index=0, - number=1, - type=5, - cpp_type=1, - label=1, - has_default_value=True, - default_value=2, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='offset', - full_name='caffe.CropParameter.offset', - index=1, - number=2, - type=13, - cpp_type=3, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=10430, - serialized_end=10478, -) - -_DATAPARAMETER = _descriptor.Descriptor( - name='DataParameter', - full_name='caffe.DataParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='source', - full_name='caffe.DataParameter.source', - index=0, - number=1, - type=9, - cpp_type=9, - label=1, - has_default_value=False, - default_value=_b("").decode('utf-8'), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='batch_size', - full_name='caffe.DataParameter.batch_size', - index=1, - number=4, - type=13, - cpp_type=3, - label=1, - has_default_value=False, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='rand_skip', - full_name='caffe.DataParameter.rand_skip', - index=2, - number=7, - type=13, - cpp_type=3, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='backend', - full_name='caffe.DataParameter.backend', - index=3, - number=8, - type=14, - cpp_type=8, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='scale', - full_name='caffe.DataParameter.scale', - index=4, - number=2, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(1), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='mean_file', - full_name='caffe.DataParameter.mean_file', - index=5, - number=3, - type=9, - cpp_type=9, - label=1, - has_default_value=False, - default_value=_b("").decode('utf-8'), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='crop_size', - full_name='caffe.DataParameter.crop_size', - index=6, - number=5, - type=13, - cpp_type=3, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='mirror', - full_name='caffe.DataParameter.mirror', - index=7, - number=6, - type=8, - cpp_type=7, - label=1, - has_default_value=True, - default_value=False, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='force_encoded_color', - full_name='caffe.DataParameter.force_encoded_color', - index=8, - number=9, - type=8, - cpp_type=7, + name='bias_param', + full_name='caffe.LayerParameter.bias_param', + index=17, + number=141, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=False, + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -6979,54 +4096,16 @@ _DATAPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='prefetch', - full_name='caffe.DataParameter.prefetch', - index=9, - number=10, - type=13, - cpp_type=3, - label=1, - has_default_value=True, - default_value=4, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[ - _DATAPARAMETER_DB, - ], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=10481, - serialized_end=10773, -) - -_DETECTIONEVALUATEPARAMETER = _descriptor.Descriptor( - name='DetectionEvaluateParameter', - full_name='caffe.DetectionEvaluateParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ _descriptor.FieldDescriptor( - name='num_classes', - full_name='caffe.DetectionEvaluateParameter.num_classes', - index=0, - number=1, - type=13, - cpp_type=3, + name='concat_param', + full_name='caffe.LayerParameter.concat_param', + index=18, + number=104, + type=11, + cpp_type=10, label=1, has_default_value=False, - default_value=0, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -7035,15 +4114,15 @@ _DETECTIONEVALUATEPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='background_label_id', - full_name='caffe.DetectionEvaluateParameter.background_label_id', - index=1, - number=2, - type=13, - cpp_type=3, + name='contrastive_loss_param', + full_name='caffe.LayerParameter.contrastive_loss_param', + index=19, + number=105, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=0, + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -7052,15 +4131,15 @@ _DETECTIONEVALUATEPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='overlap_threshold', - full_name='caffe.DetectionEvaluateParameter.overlap_threshold', - index=2, - number=3, - type=2, - cpp_type=6, + name='convolution_param', + full_name='caffe.LayerParameter.convolution_param', + index=20, + number=106, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=float(0.5), + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -7069,15 +4148,15 @@ _DETECTIONEVALUATEPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='evaluate_difficult_gt', - full_name='caffe.DetectionEvaluateParameter.evaluate_difficult_gt', - index=3, - number=4, - type=8, - cpp_type=7, + name='crop_param', + full_name='caffe.LayerParameter.crop_param', + index=21, + number=144, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=True, + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -7086,15 +4165,15 @@ _DETECTIONEVALUATEPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='name_size_file', - full_name='caffe.DetectionEvaluateParameter.name_size_file', - index=4, - number=5, - type=9, - cpp_type=9, + name='data_param', + full_name='caffe.LayerParameter.data_param', + index=22, + number=107, + type=11, + cpp_type=10, label=1, has_default_value=False, - default_value=_b("").decode('utf-8'), + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -7103,10 +4182,10 @@ _DETECTIONEVALUATEPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='resize_param', - full_name='caffe.DetectionEvaluateParameter.resize_param', - index=5, - number=6, + name='detection_evaluate_param', + full_name='caffe.LayerParameter.detection_evaluate_param', + index=23, + number=205, type=11, cpp_type=10, label=1, @@ -7119,37 +4198,17 @@ _DETECTIONEVALUATEPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=10776, - serialized_end=10996, -) - -_NONMAXIMUMSUPPRESSIONPARAMETER = _descriptor.Descriptor( - name='NonMaximumSuppressionParameter', - full_name='caffe.NonMaximumSuppressionParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ _descriptor.FieldDescriptor( - name='nms_threshold', - full_name='caffe.NonMaximumSuppressionParameter.nms_threshold', - index=0, - number=1, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(0.3), - message_type=None, + name='detection_output_param', + full_name='caffe.LayerParameter.detection_output_param', + index=24, + number=204, + type=11, + cpp_type=10, + label=1, + has_default_value=False, + default_value=None, + message_type=None, enum_type=None, containing_type=None, is_extension=False, @@ -7157,15 +4216,15 @@ _NONMAXIMUMSUPPRESSIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='top_k', - full_name='caffe.NonMaximumSuppressionParameter.top_k', - index=1, - number=2, - type=5, - cpp_type=1, + name='dropout_param', + full_name='caffe.LayerParameter.dropout_param', + index=25, + number=108, + type=11, + cpp_type=10, label=1, has_default_value=False, - default_value=0, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -7174,15 +4233,15 @@ _NONMAXIMUMSUPPRESSIONPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='eta', - full_name='caffe.NonMaximumSuppressionParameter.eta', - index=2, - number=3, - type=2, - cpp_type=6, + name='dummy_data_param', + full_name='caffe.LayerParameter.dummy_data_param', + index=26, + number=109, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=float(1), + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -7190,36 +4249,16 @@ _NONMAXIMUMSUPPRESSIONPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=10998, - serialized_end=11089, -) - -_SAVEOUTPUTPARAMETER = _descriptor.Descriptor( - name='SaveOutputParameter', - full_name='caffe.SaveOutputParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ _descriptor.FieldDescriptor( - name='output_directory', - full_name='caffe.SaveOutputParameter.output_directory', - index=0, - number=1, - type=9, - cpp_type=9, + name='eltwise_param', + full_name='caffe.LayerParameter.eltwise_param', + index=27, + number=110, + type=11, + cpp_type=10, label=1, has_default_value=False, - default_value=_b("").decode('utf-8'), + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -7228,15 +4267,15 @@ _SAVEOUTPUTPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='output_name_prefix', - full_name='caffe.SaveOutputParameter.output_name_prefix', - index=1, - number=2, - type=9, - cpp_type=9, + name='elu_param', + full_name='caffe.LayerParameter.elu_param', + index=28, + number=140, + type=11, + cpp_type=10, label=1, has_default_value=False, - default_value=_b("").decode('utf-8'), + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -7245,15 +4284,15 @@ _SAVEOUTPUTPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='output_format', - full_name='caffe.SaveOutputParameter.output_format', - index=2, - number=3, - type=9, - cpp_type=9, + name='embed_param', + full_name='caffe.LayerParameter.embed_param', + index=29, + number=137, + type=11, + cpp_type=10, label=1, has_default_value=False, - default_value=_b("").decode('utf-8'), + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -7262,15 +4301,15 @@ _SAVEOUTPUTPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='label_map_file', - full_name='caffe.SaveOutputParameter.label_map_file', - index=3, - number=4, - type=9, - cpp_type=9, + name='exp_param', + full_name='caffe.LayerParameter.exp_param', + index=30, + number=111, + type=11, + cpp_type=10, label=1, has_default_value=False, - default_value=_b("").decode('utf-8'), + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -7279,15 +4318,15 @@ _SAVEOUTPUTPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='name_size_file', - full_name='caffe.SaveOutputParameter.name_size_file', - index=4, - number=5, - type=9, - cpp_type=9, + name='flatten_param', + full_name='caffe.LayerParameter.flatten_param', + index=31, + number=135, + type=11, + cpp_type=10, label=1, has_default_value=False, - default_value=_b("").decode('utf-8'), + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -7296,15 +4335,15 @@ _SAVEOUTPUTPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='num_test_image', - full_name='caffe.SaveOutputParameter.num_test_image', - index=5, - number=6, - type=13, - cpp_type=3, + name='hdf5_data_param', + full_name='caffe.LayerParameter.hdf5_data_param', + index=32, + number=112, + type=11, + cpp_type=10, label=1, has_default_value=False, - default_value=0, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -7313,10 +4352,10 @@ _SAVEOUTPUTPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='resize_param', - full_name='caffe.SaveOutputParameter.resize_param', - index=6, - number=7, + name='hdf5_output_param', + full_name='caffe.LayerParameter.hdf5_output_param', + index=33, + number=113, type=11, cpp_type=10, label=1, @@ -7329,36 +4368,16 @@ _SAVEOUTPUTPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=11092, - serialized_end=11308, -) - -_DETECTIONOUTPUTPARAMETER = _descriptor.Descriptor( - name='DetectionOutputParameter', - full_name='caffe.DetectionOutputParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ _descriptor.FieldDescriptor( - name='num_classes', - full_name='caffe.DetectionOutputParameter.num_classes', - index=0, - number=1, - type=13, - cpp_type=3, + name='hinge_loss_param', + full_name='caffe.LayerParameter.hinge_loss_param', + index=34, + number=114, + type=11, + cpp_type=10, label=1, has_default_value=False, - default_value=0, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -7367,15 +4386,15 @@ _DETECTIONOUTPUTPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='share_location', - full_name='caffe.DetectionOutputParameter.share_location', - index=1, - number=2, - type=8, - cpp_type=7, + name='image_data_param', + full_name='caffe.LayerParameter.image_data_param', + index=35, + number=115, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=True, + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -7384,15 +4403,15 @@ _DETECTIONOUTPUTPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='background_label_id', - full_name='caffe.DetectionOutputParameter.background_label_id', - index=2, - number=3, - type=5, - cpp_type=1, + name='infogain_loss_param', + full_name='caffe.LayerParameter.infogain_loss_param', + index=36, + number=116, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=0, + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -7401,10 +4420,10 @@ _DETECTIONOUTPUTPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='nms_param', - full_name='caffe.DetectionOutputParameter.nms_param', - index=3, - number=4, + name='inner_product_param', + full_name='caffe.LayerParameter.inner_product_param', + index=37, + number=117, type=11, cpp_type=10, label=1, @@ -7418,10 +4437,10 @@ _DETECTIONOUTPUTPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='save_output_param', - full_name='caffe.DetectionOutputParameter.save_output_param', - index=4, - number=5, + name='input_param', + full_name='caffe.LayerParameter.input_param', + index=38, + number=143, type=11, cpp_type=10, label=1, @@ -7435,15 +4454,15 @@ _DETECTIONOUTPUTPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='code_type', - full_name='caffe.DetectionOutputParameter.code_type', - index=5, - number=6, - type=14, - cpp_type=8, + name='log_param', + full_name='caffe.LayerParameter.log_param', + index=39, + number=134, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=1, + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -7452,16 +4471,15 @@ _DETECTIONOUTPUTPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='variance_encoded_in_target', - full_name= - 'caffe.DetectionOutputParameter.variance_encoded_in_target', - index=6, - number=8, - type=8, - cpp_type=7, + name='lrn_param', + full_name='caffe.LayerParameter.lrn_param', + index=40, + number=118, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=False, + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -7470,15 +4488,15 @@ _DETECTIONOUTPUTPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='keep_top_k', - full_name='caffe.DetectionOutputParameter.keep_top_k', - index=7, - number=7, - type=5, - cpp_type=1, + name='memory_data_param', + full_name='caffe.LayerParameter.memory_data_param', + index=41, + number=119, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=-1, + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -7487,15 +4505,15 @@ _DETECTIONOUTPUTPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='confidence_threshold', - full_name='caffe.DetectionOutputParameter.confidence_threshold', - index=8, - number=9, - type=2, - cpp_type=6, + name='multibox_loss_param', + full_name='caffe.LayerParameter.multibox_loss_param', + index=42, + number=201, + type=11, + cpp_type=10, label=1, has_default_value=False, - default_value=float(0), + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -7504,15 +4522,15 @@ _DETECTIONOUTPUTPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='visualize', - full_name='caffe.DetectionOutputParameter.visualize', - index=9, - number=10, - type=8, - cpp_type=7, + name='mvn_param', + full_name='caffe.LayerParameter.mvn_param', + index=43, + number=120, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=False, + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -7521,15 +4539,15 @@ _DETECTIONOUTPUTPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='visualize_threshold', - full_name='caffe.DetectionOutputParameter.visualize_threshold', - index=10, - number=11, - type=2, - cpp_type=6, + name='norm_param', + full_name='caffe.LayerParameter.norm_param', + index=44, + number=206, + type=11, + cpp_type=10, label=1, has_default_value=False, - default_value=float(0), + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -7538,15 +4556,15 @@ _DETECTIONOUTPUTPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='save_file', - full_name='caffe.DetectionOutputParameter.save_file', - index=11, - number=12, - type=9, - cpp_type=9, + name='parameter_param', + full_name='caffe.LayerParameter.parameter_param', + index=45, + number=145, + type=11, + cpp_type=10, label=1, has_default_value=False, - default_value=_b("").decode('utf-8'), + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -7554,36 +4572,16 @@ _DETECTIONOUTPUTPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=11311, - serialized_end=11766, -) - -_DROPOUTPARAMETER = _descriptor.Descriptor( - name='DropoutParameter', - full_name='caffe.DropoutParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ _descriptor.FieldDescriptor( - name='dropout_ratio', - full_name='caffe.DropoutParameter.dropout_ratio', - index=0, - number=1, - type=2, - cpp_type=6, + name='permute_param', + full_name='caffe.LayerParameter.permute_param', + index=46, + number=202, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=float(0.5), + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -7591,36 +4589,16 @@ _DROPOUTPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=11768, - serialized_end=11814, -) - -_DUMMYDATAPARAMETER = _descriptor.Descriptor( - name='DummyDataParameter', - full_name='caffe.DummyDataParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ _descriptor.FieldDescriptor( - name='data_filler', - full_name='caffe.DummyDataParameter.data_filler', - index=0, - number=1, + name='pooling_param', + full_name='caffe.LayerParameter.pooling_param', + index=47, + number=121, type=11, cpp_type=10, - label=3, + label=1, has_default_value=False, - default_value=[], + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -7628,48 +4606,16 @@ _DUMMYDATAPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='shape', - full_name='caffe.DummyDataParameter.shape', - index=1, - number=6, - type=11, - cpp_type=10, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='num', - full_name='caffe.DummyDataParameter.num', - index=2, - number=2, - type=13, - cpp_type=3, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='channels', - full_name='caffe.DummyDataParameter.channels', - index=3, - number=3, - type=13, - cpp_type=3, - label=3, + name='power_param', + full_name='caffe.LayerParameter.power_param', + index=48, + number=122, + type=11, + cpp_type=10, + label=1, has_default_value=False, - default_value=[], + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -7677,68 +4623,16 @@ _DUMMYDATAPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='height', - full_name='caffe.DummyDataParameter.height', - index=4, - number=4, - type=13, - cpp_type=3, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='width', - full_name='caffe.DummyDataParameter.width', - index=5, - number=5, - type=13, - cpp_type=3, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=11817, - serialized_end=11977, -) - -_ELTWISEPARAMETER = _descriptor.Descriptor( - name='EltwiseParameter', - full_name='caffe.EltwiseParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ _descriptor.FieldDescriptor( - name='operation', - full_name='caffe.EltwiseParameter.operation', - index=0, - number=1, - type=14, - cpp_type=8, + name='prelu_param', + full_name='caffe.LayerParameter.prelu_param', + index=49, + number=131, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=1, + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -7746,32 +4640,16 @@ _ELTWISEPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='coeff', - full_name='caffe.EltwiseParameter.coeff', - index=1, - number=2, - type=2, - cpp_type=6, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='stable_prod_grad', - full_name='caffe.EltwiseParameter.stable_prod_grad', - index=2, - number=3, - type=8, - cpp_type=7, + name='prior_box_param', + full_name='caffe.LayerParameter.prior_box_param', + index=50, + number=203, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=True, + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -7779,117 +4657,11 @@ _ELTWISEPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[ - _ELTWISEPARAMETER_ELTWISEOP, - ], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=11980, - serialized_end=12145, -) - -_ELUPARAMETER = _descriptor.Descriptor( - name='ELUParameter', - full_name='caffe.ELUParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='alpha', - full_name='caffe.ELUParameter.alpha', - index=0, - number=1, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(1), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=12147, - serialized_end=12179, -) - -_EMBEDPARAMETER = _descriptor.Descriptor( - name='EmbedParameter', - full_name='caffe.EmbedParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='num_output', - full_name='caffe.EmbedParameter.num_output', - index=0, - number=1, - type=13, - cpp_type=3, - label=1, - has_default_value=False, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='input_dim', - full_name='caffe.EmbedParameter.input_dim', - index=1, - number=2, - type=13, - cpp_type=3, - label=1, - has_default_value=False, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='bias_term', - full_name='caffe.EmbedParameter.bias_term', - index=2, - number=3, - type=8, - cpp_type=7, - label=1, - has_default_value=True, - default_value=True, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='weight_filler', - full_name='caffe.EmbedParameter.weight_filler', - index=3, - number=4, + name='python_param', + full_name='caffe.LayerParameter.python_param', + index=51, + number=130, type=11, cpp_type=10, label=1, @@ -7903,10 +4675,10 @@ _EMBEDPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='bias_filler', - full_name='caffe.EmbedParameter.bias_filler', - index=4, - number=5, + name='recurrent_param', + full_name='caffe.LayerParameter.recurrent_param', + index=52, + number=146, type=11, cpp_type=10, label=1, @@ -7919,172 +4691,16 @@ _EMBEDPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=12182, - serialized_end=12354, -) - -_EXPPARAMETER = _descriptor.Descriptor( - name='ExpParameter', - full_name='caffe.ExpParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='base', - full_name='caffe.ExpParameter.base', - index=0, - number=1, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(-1), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='scale', - full_name='caffe.ExpParameter.scale', - index=1, - number=2, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(1), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='shift', - full_name='caffe.ExpParameter.shift', - index=2, - number=3, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(0), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=12356, - serialized_end=12424, -) - -_FLATTENPARAMETER = _descriptor.Descriptor( - name='FlattenParameter', - full_name='caffe.FlattenParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='axis', - full_name='caffe.FlattenParameter.axis', - index=0, - number=1, - type=5, - cpp_type=1, - label=1, - has_default_value=True, - default_value=1, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='end_axis', - full_name='caffe.FlattenParameter.end_axis', - index=1, - number=2, - type=5, - cpp_type=1, - label=1, - has_default_value=True, - default_value=-1, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=12426, - serialized_end=12483, -) - -_HDF5DATAPARAMETER = _descriptor.Descriptor( - name='HDF5DataParameter', - full_name='caffe.HDF5DataParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='source', - full_name='caffe.HDF5DataParameter.source', - index=0, - number=1, - type=9, - cpp_type=9, - label=1, - has_default_value=False, - default_value=_b("").decode('utf-8'), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='batch_size', - full_name='caffe.HDF5DataParameter.batch_size', - index=1, - number=2, - type=13, - cpp_type=3, + name='reduction_param', + full_name='caffe.LayerParameter.reduction_param', + index=53, + number=136, + type=11, + cpp_type=10, label=1, has_default_value=False, - default_value=0, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -8092,52 +4708,16 @@ _HDF5DATAPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='shuffle', - full_name='caffe.HDF5DataParameter.shuffle', - index=2, - number=3, - type=8, - cpp_type=7, - label=1, - has_default_value=True, - default_value=False, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=12485, - serialized_end=12564, -) - -_HDF5OUTPUTPARAMETER = _descriptor.Descriptor( - name='HDF5OutputParameter', - full_name='caffe.HDF5OutputParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ _descriptor.FieldDescriptor( - name='file_name', - full_name='caffe.HDF5OutputParameter.file_name', - index=0, - number=1, - type=9, - cpp_type=9, + name='relu_param', + full_name='caffe.LayerParameter.relu_param', + index=54, + number=123, + type=11, + cpp_type=10, label=1, has_default_value=False, - default_value=_b("").decode('utf-8'), + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -8145,90 +4725,16 @@ _HDF5OUTPUTPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=12566, - serialized_end=12606, -) - -_HINGELOSSPARAMETER = _descriptor.Descriptor( - name='HingeLossParameter', - full_name='caffe.HingeLossParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='norm', - full_name='caffe.HingeLossParameter.norm', - index=0, - number=1, - type=14, - cpp_type=8, - label=1, - has_default_value=True, - default_value=1, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[ - _HINGELOSSPARAMETER_NORM, - ], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=12608, - serialized_end=12702, -) - -_IMAGEDATAPARAMETER = _descriptor.Descriptor( - name='ImageDataParameter', - full_name='caffe.ImageDataParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='source', - full_name='caffe.ImageDataParameter.source', - index=0, - number=1, - type=9, - cpp_type=9, - label=1, - has_default_value=False, - default_value=_b("").decode('utf-8'), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='batch_size', - full_name='caffe.ImageDataParameter.batch_size', - index=1, - number=4, - type=13, - cpp_type=3, + name='reshape_param', + full_name='caffe.LayerParameter.reshape_param', + index=55, + number=133, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=1, + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -8237,15 +4743,15 @@ _IMAGEDATAPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='rand_skip', - full_name='caffe.ImageDataParameter.rand_skip', - index=2, - number=7, - type=13, - cpp_type=3, + name='scale_param', + full_name='caffe.LayerParameter.scale_param', + index=56, + number=142, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=0, + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -8254,15 +4760,15 @@ _IMAGEDATAPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='shuffle', - full_name='caffe.ImageDataParameter.shuffle', - index=3, - number=8, - type=8, - cpp_type=7, + name='shuffle_channel_param', + full_name='caffe.LayerParameter.shuffle_channel_param', + index=57, + number=208, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=False, + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -8271,15 +4777,15 @@ _IMAGEDATAPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='new_height', - full_name='caffe.ImageDataParameter.new_height', - index=4, - number=9, - type=13, - cpp_type=3, + name='sigmoid_param', + full_name='caffe.LayerParameter.sigmoid_param', + index=58, + number=124, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=0, + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -8288,15 +4794,15 @@ _IMAGEDATAPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='new_width', - full_name='caffe.ImageDataParameter.new_width', - index=5, - number=10, - type=13, - cpp_type=3, + name='softmax_param', + full_name='caffe.LayerParameter.softmax_param', + index=59, + number=125, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=0, + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -8305,15 +4811,15 @@ _IMAGEDATAPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='is_color', - full_name='caffe.ImageDataParameter.is_color', - index=6, - number=11, - type=8, - cpp_type=7, + name='spp_param', + full_name='caffe.LayerParameter.spp_param', + index=60, + number=132, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=True, + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -8321,32 +4827,16 @@ _IMAGEDATAPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='scale', - full_name='caffe.ImageDataParameter.scale', - index=7, - number=2, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(1), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='mean_file', - full_name='caffe.ImageDataParameter.mean_file', - index=8, - number=3, - type=9, - cpp_type=9, + name='slice_param', + full_name='caffe.LayerParameter.slice_param', + index=61, + number=126, + type=11, + cpp_type=10, label=1, has_default_value=False, - default_value=_b("").decode('utf-8'), + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -8355,15 +4845,15 @@ _IMAGEDATAPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='crop_size', - full_name='caffe.ImageDataParameter.crop_size', - index=9, - number=5, - type=13, - cpp_type=3, + name='tanh_param', + full_name='caffe.LayerParameter.tanh_param', + index=62, + number=127, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=0, + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -8371,32 +4861,16 @@ _IMAGEDATAPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='mirror', - full_name='caffe.ImageDataParameter.mirror', - index=10, - number=6, - type=8, - cpp_type=7, - label=1, - has_default_value=True, - default_value=False, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='root_folder', - full_name='caffe.ImageDataParameter.root_folder', - index=11, - number=12, - type=9, - cpp_type=9, + name='threshold_param', + full_name='caffe.LayerParameter.threshold_param', + index=63, + number=128, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=_b("").decode('utf-8'), + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -8404,36 +4878,50 @@ _IMAGEDATAPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=12705, - serialized_end=12984, -) - -_INFOGAINLOSSPARAMETER = _descriptor.Descriptor( - name='InfogainLossParameter', - full_name='caffe.InfogainLossParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ _descriptor.FieldDescriptor( - name='source', - full_name='caffe.InfogainLossParameter.source', - index=0, - number=1, - type=9, - cpp_type=9, + name='tile_param', + full_name='caffe.LayerParameter.tile_param', + index=64, + number=138, + type=11, + cpp_type=10, label=1, has_default_value=False, - default_value=_b("").decode('utf-8'), + default_value=None, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='video_data_param', + full_name='caffe.LayerParameter.video_data_param', + index=65, + number=207, + type=11, + cpp_type=10, + label=1, + has_default_value=False, + default_value=None, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='window_data_param', + full_name='caffe.LayerParameter.window_data_param', + index=66, + number=129, + type=11, + cpp_type=10, + label=1, + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -8450,27 +4938,26 @@ _INFOGAINLOSSPARAMETER = _descriptor.Descriptor( syntax='proto2', extension_ranges=[], oneofs=[], - serialized_start=12986, - serialized_end=13025, -) + serialized_start=4176, + serialized_end=7263, ) -_INNERPRODUCTPARAMETER = _descriptor.Descriptor( - name='InnerProductParameter', - full_name='caffe.InnerProductParameter', +_TRANSFORMATIONPARAMETER = _descriptor.Descriptor( + name='TransformationParameter', + full_name='caffe.TransformationParameter', filename=None, file=DESCRIPTOR, containing_type=None, fields=[ _descriptor.FieldDescriptor( - name='num_output', - full_name='caffe.InnerProductParameter.num_output', + name='scale', + full_name='caffe.TransformationParameter.scale', index=0, number=1, - type=13, - cpp_type=3, + type=2, + cpp_type=6, label=1, - has_default_value=False, - default_value=0, + has_default_value=True, + default_value=float(1), message_type=None, enum_type=None, containing_type=None, @@ -8479,15 +4966,15 @@ _INNERPRODUCTPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='bias_term', - full_name='caffe.InnerProductParameter.bias_term', + name='mirror', + full_name='caffe.TransformationParameter.mirror', index=1, number=2, type=8, cpp_type=7, label=1, has_default_value=True, - default_value=True, + default_value=False, message_type=None, enum_type=None, containing_type=None, @@ -8496,15 +4983,15 @@ _INNERPRODUCTPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='weight_filler', - full_name='caffe.InnerProductParameter.weight_filler', + name='crop_size', + full_name='caffe.TransformationParameter.crop_size', index=2, number=3, - type=11, - cpp_type=10, + type=13, + cpp_type=3, label=1, - has_default_value=False, - default_value=None, + has_default_value=True, + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -8513,15 +5000,15 @@ _INNERPRODUCTPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='bias_filler', - full_name='caffe.InnerProductParameter.bias_filler', + name='crop_h', + full_name='caffe.TransformationParameter.crop_h', index=3, - number=4, - type=11, - cpp_type=10, + number=11, + type=13, + cpp_type=3, label=1, - has_default_value=False, - default_value=None, + has_default_value=True, + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -8530,15 +5017,15 @@ _INNERPRODUCTPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='axis', - full_name='caffe.InnerProductParameter.axis', + name='crop_w', + full_name='caffe.TransformationParameter.crop_w', index=4, - number=5, - type=5, - cpp_type=1, + number=12, + type=13, + cpp_type=3, label=1, has_default_value=True, - default_value=1, + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -8547,15 +5034,15 @@ _INNERPRODUCTPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='transpose', - full_name='caffe.InnerProductParameter.transpose', + name='mean_file', + full_name='caffe.TransformationParameter.mean_file', index=5, - number=6, - type=8, - cpp_type=7, + number=4, + type=9, + cpp_type=9, label=1, - has_default_value=True, - default_value=False, + has_default_value=False, + default_value=_b("").decode('utf-8'), message_type=None, enum_type=None, containing_type=None, @@ -8563,259 +5050,16 @@ _INNERPRODUCTPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=13028, - serialized_end=13231, -) - -_INPUTPARAMETER = _descriptor.Descriptor( - name='InputParameter', - full_name='caffe.InputParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='shape', - full_name='caffe.InputParameter.shape', - index=0, - number=1, - type=11, - cpp_type=10, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=13233, - serialized_end=13282, -) - -_LOGPARAMETER = _descriptor.Descriptor( - name='LogParameter', - full_name='caffe.LogParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='base', - full_name='caffe.LogParameter.base', - index=0, - number=1, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(-1), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='scale', - full_name='caffe.LogParameter.scale', - index=1, - number=2, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(1), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='shift', - full_name='caffe.LogParameter.shift', - index=2, - number=3, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(0), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=13284, - serialized_end=13352, -) - -_LRNPARAMETER = _descriptor.Descriptor( - name='LRNParameter', - full_name='caffe.LRNParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='local_size', - full_name='caffe.LRNParameter.local_size', - index=0, - number=1, - type=13, - cpp_type=3, - label=1, - has_default_value=True, - default_value=5, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='alpha', - full_name='caffe.LRNParameter.alpha', - index=1, - number=2, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(1), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='beta', - full_name='caffe.LRNParameter.beta', - index=2, - number=3, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(0.75), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='norm_region', - full_name='caffe.LRNParameter.norm_region', - index=3, - number=4, - type=14, - cpp_type=8, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='k', - full_name='caffe.LRNParameter.k', - index=4, - number=5, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(1), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='engine', - full_name='caffe.LRNParameter.engine', - index=5, - number=6, - type=14, - cpp_type=8, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[ - _LRNPARAMETER_NORMREGION, - _LRNPARAMETER_ENGINE, - ], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=13355, - serialized_end=13667, -) - -_MEMORYDATAPARAMETER = _descriptor.Descriptor( - name='MemoryDataParameter', - full_name='caffe.MemoryDataParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ _descriptor.FieldDescriptor( - name='batch_size', - full_name='caffe.MemoryDataParameter.batch_size', - index=0, - number=1, - type=13, - cpp_type=3, - label=1, + name='mean_value', + full_name='caffe.TransformationParameter.mean_value', + index=6, + number=5, + type=2, + cpp_type=6, + label=3, has_default_value=False, - default_value=0, + default_value=[], message_type=None, enum_type=None, containing_type=None, @@ -8824,15 +5068,15 @@ _MEMORYDATAPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='channels', - full_name='caffe.MemoryDataParameter.channels', - index=1, - number=2, - type=13, - cpp_type=3, + name='force_color', + full_name='caffe.TransformationParameter.force_color', + index=7, + number=6, + type=8, + cpp_type=7, label=1, - has_default_value=False, - default_value=0, + has_default_value=True, + default_value=False, message_type=None, enum_type=None, containing_type=None, @@ -8841,15 +5085,15 @@ _MEMORYDATAPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='height', - full_name='caffe.MemoryDataParameter.height', - index=2, - number=3, - type=13, - cpp_type=3, + name='force_gray', + full_name='caffe.TransformationParameter.force_gray', + index=8, + number=7, + type=8, + cpp_type=7, label=1, - has_default_value=False, - default_value=0, + has_default_value=True, + default_value=False, message_type=None, enum_type=None, containing_type=None, @@ -8857,52 +5101,16 @@ _MEMORYDATAPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='width', - full_name='caffe.MemoryDataParameter.width', - index=3, - number=4, - type=13, - cpp_type=3, - label=1, - has_default_value=False, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=13669, - serialized_end=13759, -) - -_MULTIBOXLOSSPARAMETER = _descriptor.Descriptor( - name='MultiBoxLossParameter', - full_name='caffe.MultiBoxLossParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ _descriptor.FieldDescriptor( - name='loc_loss_type', - full_name='caffe.MultiBoxLossParameter.loc_loss_type', - index=0, - number=1, - type=14, - cpp_type=8, + name='resize_param', + full_name='caffe.TransformationParameter.resize_param', + index=9, + number=8, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=1, + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -8911,15 +5119,15 @@ _MULTIBOXLOSSPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='conf_loss_type', - full_name='caffe.MultiBoxLossParameter.conf_loss_type', - index=1, - number=2, - type=14, - cpp_type=8, + name='noise_param', + full_name='caffe.TransformationParameter.noise_param', + index=10, + number=9, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=0, + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -8928,15 +5136,15 @@ _MULTIBOXLOSSPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='loc_weight', - full_name='caffe.MultiBoxLossParameter.loc_weight', - index=2, - number=3, - type=2, - cpp_type=6, + name='distort_param', + full_name='caffe.TransformationParameter.distort_param', + index=11, + number=13, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=float(1), + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -8945,15 +5153,15 @@ _MULTIBOXLOSSPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='num_classes', - full_name='caffe.MultiBoxLossParameter.num_classes', - index=3, - number=4, - type=13, - cpp_type=3, + name='expand_param', + full_name='caffe.TransformationParameter.expand_param', + index=12, + number=14, + type=11, + cpp_type=10, label=1, has_default_value=False, - default_value=0, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -8962,15 +5170,15 @@ _MULTIBOXLOSSPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='share_location', - full_name='caffe.MultiBoxLossParameter.share_location', - index=4, - number=5, - type=8, - cpp_type=7, + name='emit_constraint', + full_name='caffe.TransformationParameter.emit_constraint', + index=13, + number=10, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=True, + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -8978,16 +5186,35 @@ _MULTIBOXLOSSPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=7266, + serialized_end=7724, ) + +_RESIZEPARAMETER = _descriptor.Descriptor( + name='ResizeParameter', + full_name='caffe.ResizeParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ _descriptor.FieldDescriptor( - name='match_type', - full_name='caffe.MultiBoxLossParameter.match_type', - index=5, - number=6, - type=14, - cpp_type=8, + name='prob', + full_name='caffe.ResizeParameter.prob', + index=0, + number=1, + type=2, + cpp_type=6, label=1, has_default_value=True, - default_value=1, + default_value=float(1), message_type=None, enum_type=None, containing_type=None, @@ -8996,15 +5223,15 @@ _MULTIBOXLOSSPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='overlap_threshold', - full_name='caffe.MultiBoxLossParameter.overlap_threshold', - index=6, - number=7, - type=2, - cpp_type=6, + name='resize_mode', + full_name='caffe.ResizeParameter.resize_mode', + index=1, + number=2, + type=14, + cpp_type=8, label=1, has_default_value=True, - default_value=float(0.5), + default_value=1, message_type=None, enum_type=None, containing_type=None, @@ -9013,15 +5240,15 @@ _MULTIBOXLOSSPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='use_prior_for_matching', - full_name='caffe.MultiBoxLossParameter.use_prior_for_matching', - index=7, - number=8, - type=8, - cpp_type=7, + name='height', + full_name='caffe.ResizeParameter.height', + index=2, + number=3, + type=13, + cpp_type=3, label=1, has_default_value=True, - default_value=True, + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -9030,10 +5257,10 @@ _MULTIBOXLOSSPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='background_label_id', - full_name='caffe.MultiBoxLossParameter.background_label_id', - index=8, - number=9, + name='width', + full_name='caffe.ResizeParameter.width', + index=3, + number=4, type=13, cpp_type=3, label=1, @@ -9047,15 +5274,15 @@ _MULTIBOXLOSSPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='use_difficult_gt', - full_name='caffe.MultiBoxLossParameter.use_difficult_gt', - index=9, - number=10, - type=8, - cpp_type=7, + name='height_scale', + full_name='caffe.ResizeParameter.height_scale', + index=4, + number=8, + type=13, + cpp_type=3, label=1, has_default_value=True, - default_value=True, + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -9064,15 +5291,15 @@ _MULTIBOXLOSSPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='do_neg_mining', - full_name='caffe.MultiBoxLossParameter.do_neg_mining', - index=10, - number=11, - type=8, - cpp_type=7, + name='width_scale', + full_name='caffe.ResizeParameter.width_scale', + index=5, + number=9, + type=13, + cpp_type=3, label=1, - has_default_value=False, - default_value=False, + has_default_value=True, + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -9081,15 +5308,15 @@ _MULTIBOXLOSSPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='neg_pos_ratio', - full_name='caffe.MultiBoxLossParameter.neg_pos_ratio', - index=11, - number=12, - type=2, - cpp_type=6, + name='pad_mode', + full_name='caffe.ResizeParameter.pad_mode', + index=6, + number=5, + type=14, + cpp_type=8, label=1, has_default_value=True, - default_value=float(3), + default_value=1, message_type=None, enum_type=None, containing_type=None, @@ -9098,15 +5325,15 @@ _MULTIBOXLOSSPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='neg_overlap', - full_name='caffe.MultiBoxLossParameter.neg_overlap', - index=12, - number=13, + name='pad_value', + full_name='caffe.ResizeParameter.pad_value', + index=7, + number=6, type=2, cpp_type=6, - label=1, - has_default_value=True, - default_value=float(0.5), + label=3, + has_default_value=False, + default_value=[], message_type=None, enum_type=None, containing_type=None, @@ -9115,32 +5342,15 @@ _MULTIBOXLOSSPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='code_type', - full_name='caffe.MultiBoxLossParameter.code_type', - index=13, - number=14, + name='interp_mode', + full_name='caffe.ResizeParameter.interp_mode', + index=8, + number=7, type=14, cpp_type=8, - label=1, - has_default_value=True, - default_value=1, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor( - name='encode_variance_in_target', - full_name='caffe.MultiBoxLossParameter.encode_variance_in_target', - index=14, - number=16, - type=8, - cpp_type=7, - label=1, - has_default_value=True, - default_value=False, + label=3, + has_default_value=False, + default_value=[], message_type=None, enum_type=None, containing_type=None, @@ -9148,16 +5358,39 @@ _MULTIBOXLOSSPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor( - name='map_object_to_agnostic', - full_name='caffe.MultiBoxLossParameter.map_object_to_agnostic', - index=15, - number=17, - type=8, - cpp_type=7, + ], + extensions=[], + nested_types=[], + enum_types=[ + _RESIZEPARAMETER_RESIZE_MODE, + _RESIZEPARAMETER_PAD_MODE, + _RESIZEPARAMETER_INTERP_MODE, + ], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=7727, + serialized_end=8255, ) + +_SALTPEPPERPARAMETER = _descriptor.Descriptor( + name='SaltPepperParameter', + full_name='caffe.SaltPepperParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='fraction', + full_name='caffe.SaltPepperParameter.fraction', + index=0, + number=1, + type=2, + cpp_type=6, label=1, has_default_value=True, - default_value=False, + default_value=float(0), message_type=None, enum_type=None, containing_type=None, @@ -9166,10 +5399,63 @@ _MULTIBOXLOSSPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='ignore_cross_boundary_bbox', - full_name='caffe.MultiBoxLossParameter.ignore_cross_boundary_bbox', - index=16, - number=18, + name='value', + full_name='caffe.SaltPepperParameter.value', + index=1, + number=2, + type=2, + cpp_type=6, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=8257, + serialized_end=8314, ) + +_NOISEPARAMETER = _descriptor.Descriptor( + name='NoiseParameter', + full_name='caffe.NoiseParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='prob', + full_name='caffe.NoiseParameter.prob', + index=0, + number=1, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(0), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='hist_eq', + full_name='caffe.NoiseParameter.hist_eq', + index=1, + number=2, type=8, cpp_type=7, label=1, @@ -9183,10 +5469,10 @@ _MULTIBOXLOSSPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='bp_inside', - full_name='caffe.MultiBoxLossParameter.bp_inside', - index=17, - number=19, + name='inverse', + full_name='caffe.NoiseParameter.inverse', + index=2, + number=3, type=8, cpp_type=7, label=1, @@ -9200,15 +5486,15 @@ _MULTIBOXLOSSPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='mining_type', - full_name='caffe.MultiBoxLossParameter.mining_type', - index=18, - number=20, - type=14, - cpp_type=8, + name='decolorize', + full_name='caffe.NoiseParameter.decolorize', + index=3, + number=4, + type=8, + cpp_type=7, label=1, has_default_value=True, - default_value=1, + default_value=False, message_type=None, enum_type=None, containing_type=None, @@ -9217,15 +5503,15 @@ _MULTIBOXLOSSPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='nms_param', - full_name='caffe.MultiBoxLossParameter.nms_param', - index=19, - number=21, - type=11, - cpp_type=10, + name='gauss_blur', + full_name='caffe.NoiseParameter.gauss_blur', + index=4, + number=5, + type=8, + cpp_type=7, label=1, - has_default_value=False, - default_value=None, + has_default_value=True, + default_value=False, message_type=None, enum_type=None, containing_type=None, @@ -9234,15 +5520,15 @@ _MULTIBOXLOSSPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='sample_size', - full_name='caffe.MultiBoxLossParameter.sample_size', - index=20, - number=22, - type=5, - cpp_type=1, + name='jpeg', + full_name='caffe.NoiseParameter.jpeg', + index=5, + number=6, + type=2, + cpp_type=6, label=1, has_default_value=True, - default_value=64, + default_value=float(-1), message_type=None, enum_type=None, containing_type=None, @@ -9251,10 +5537,10 @@ _MULTIBOXLOSSPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='use_prior_for_nms', - full_name='caffe.MultiBoxLossParameter.use_prior_for_nms', - index=21, - number=23, + name='posterize', + full_name='caffe.NoiseParameter.posterize', + index=6, + number=7, type=8, cpp_type=7, label=1, @@ -9267,41 +5553,16 @@ _MULTIBOXLOSSPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[ - _MULTIBOXLOSSPARAMETER_LOCLOSSTYPE, - _MULTIBOXLOSSPARAMETER_CONFLOSSTYPE, - _MULTIBOXLOSSPARAMETER_MATCHTYPE, - _MULTIBOXLOSSPARAMETER_MININGTYPE, - ], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=13762, - serialized_end=14890, -) - -_MVNPARAMETER = _descriptor.Descriptor( - name='MVNParameter', - full_name='caffe.MVNParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ _descriptor.FieldDescriptor( - name='normalize_variance', - full_name='caffe.MVNParameter.normalize_variance', - index=0, - number=1, + name='erode', + full_name='caffe.NoiseParameter.erode', + index=7, + number=8, type=8, cpp_type=7, label=1, has_default_value=True, - default_value=True, + default_value=False, message_type=None, enum_type=None, containing_type=None, @@ -9310,10 +5571,10 @@ _MVNPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='across_channels', - full_name='caffe.MVNParameter.across_channels', - index=1, - number=2, + name='saltpepper', + full_name='caffe.NoiseParameter.saltpepper', + index=8, + number=9, type=8, cpp_type=7, label=1, @@ -9326,52 +5587,33 @@ _MVNPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='eps', - full_name='caffe.MVNParameter.eps', - index=2, - number=3, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(1e-09), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=14892, - serialized_end=14992, -) - -_NORMALIZEPARAMETER = _descriptor.Descriptor( - name='NormalizeParameter', - full_name='caffe.NormalizeParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ _descriptor.FieldDescriptor( - name='across_spatial', - full_name='caffe.NormalizeParameter.across_spatial', - index=0, - number=1, + name='saltpepper_param', + full_name='caffe.NoiseParameter.saltpepper_param', + index=9, + number=10, + type=11, + cpp_type=10, + label=1, + has_default_value=False, + default_value=None, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='clahe', + full_name='caffe.NoiseParameter.clahe', + index=10, + number=11, type=8, cpp_type=7, label=1, has_default_value=True, - default_value=True, + default_value=False, message_type=None, enum_type=None, containing_type=None, @@ -9380,15 +5622,15 @@ _NORMALIZEPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='scale_filler', - full_name='caffe.NormalizeParameter.scale_filler', - index=1, - number=2, - type=11, - cpp_type=10, + name='convert_to_hsv', + full_name='caffe.NoiseParameter.convert_to_hsv', + index=11, + number=12, + type=8, + cpp_type=7, label=1, - has_default_value=False, - default_value=None, + has_default_value=True, + default_value=False, message_type=None, enum_type=None, containing_type=None, @@ -9397,15 +5639,15 @@ _NORMALIZEPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='channel_shared', - full_name='caffe.NormalizeParameter.channel_shared', - index=2, - number=3, + name='convert_to_lab', + full_name='caffe.NoiseParameter.convert_to_lab', + index=12, + number=13, type=8, cpp_type=7, label=1, has_default_value=True, - default_value=True, + default_value=False, message_type=None, enum_type=None, containing_type=None, @@ -9413,22 +5655,6 @@ _NORMALIZEPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='eps', - full_name='caffe.NormalizeParameter.eps', - index=3, - number=4, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(1e-10), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), ], extensions=[], nested_types=[], @@ -9438,163 +5664,26 @@ _NORMALIZEPARAMETER = _descriptor.Descriptor( syntax='proto2', extension_ranges=[], oneofs=[], - serialized_start=14995, - serialized_end=15141, -) + serialized_start=8317, + serialized_end=8683, ) -_PARAMETERPARAMETER = _descriptor.Descriptor( - name='ParameterParameter', - full_name='caffe.ParameterParameter', +_DISTORTIONPARAMETER = _descriptor.Descriptor( + name='DistortionParameter', + full_name='caffe.DistortionParameter', filename=None, file=DESCRIPTOR, containing_type=None, fields=[ - _descriptor.FieldDescriptor(name='shape', - full_name='caffe.ParameterParameter.shape', - index=0, - number=1, - type=11, - cpp_type=10, - label=1, - has_default_value=False, - default_value=None, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=15143, - serialized_end=15196, -) - -_PERMUTEPARAMETER = _descriptor.Descriptor( - name='PermuteParameter', - full_name='caffe.PermuteParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='order', - full_name='caffe.PermuteParameter.order', - index=0, - number=1, - type=13, - cpp_type=3, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=15198, - serialized_end=15231, -) - -_POOLINGPARAMETER = _descriptor.Descriptor( - name='PoolingParameter', - full_name='caffe.PoolingParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='pool', - full_name='caffe.PoolingParameter.pool', - index=0, - number=1, - type=14, - cpp_type=8, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='pad', - full_name='caffe.PoolingParameter.pad', - index=1, - number=4, - type=13, - cpp_type=3, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='pad_h', - full_name='caffe.PoolingParameter.pad_h', - index=2, - number=9, - type=13, - cpp_type=3, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='pad_w', - full_name='caffe.PoolingParameter.pad_w', - index=3, - number=10, - type=13, - cpp_type=3, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='kernel_size', - full_name='caffe.PoolingParameter.kernel_size', - index=4, - number=2, - type=13, - cpp_type=3, + name='brightness_prob', + full_name='caffe.DistortionParameter.brightness_prob', + index=0, + number=1, + type=2, + cpp_type=6, label=1, - has_default_value=False, - default_value=0, + has_default_value=True, + default_value=float(0), message_type=None, enum_type=None, containing_type=None, @@ -9602,112 +5691,16 @@ _POOLINGPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='kernel_h', - full_name='caffe.PoolingParameter.kernel_h', - index=5, - number=5, - type=13, - cpp_type=3, - label=1, - has_default_value=False, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='kernel_w', - full_name='caffe.PoolingParameter.kernel_w', - index=6, - number=6, - type=13, - cpp_type=3, - label=1, - has_default_value=False, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='stride', - full_name='caffe.PoolingParameter.stride', - index=7, - number=3, - type=13, - cpp_type=3, - label=1, - has_default_value=True, - default_value=1, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='stride_h', - full_name='caffe.PoolingParameter.stride_h', - index=8, - number=7, - type=13, - cpp_type=3, - label=1, - has_default_value=False, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='stride_w', - full_name='caffe.PoolingParameter.stride_w', - index=9, - number=8, - type=13, - cpp_type=3, - label=1, - has_default_value=False, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='engine', - full_name='caffe.PoolingParameter.engine', - index=10, - number=11, - type=14, - cpp_type=8, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='global_pooling', - full_name='caffe.PoolingParameter.global_pooling', - index=11, - number=12, - type=8, - cpp_type=7, + name='brightness_delta', + full_name='caffe.DistortionParameter.brightness_delta', + index=1, + number=2, + type=2, + cpp_type=6, label=1, has_default_value=True, - default_value=False, + default_value=float(0), message_type=None, enum_type=None, containing_type=None, @@ -9715,107 +5708,16 @@ _POOLINGPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[ - _POOLINGPARAMETER_POOLMETHOD, - _POOLINGPARAMETER_ENGINE, - ], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=15234, - serialized_end=15652, -) - -_POWERPARAMETER = _descriptor.Descriptor( - name='PowerParameter', - full_name='caffe.PowerParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='power', - full_name='caffe.PowerParameter.power', - index=0, - number=1, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(1), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='scale', - full_name='caffe.PowerParameter.scale', - index=1, - number=2, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(1), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='shift', - full_name='caffe.PowerParameter.shift', - index=2, - number=3, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(0), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=15654, - serialized_end=15724, -) - -_PRIORBOXPARAMETER = _descriptor.Descriptor( - name='PriorBoxParameter', - full_name='caffe.PriorBoxParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ _descriptor.FieldDescriptor( - name='min_size', - full_name='caffe.PriorBoxParameter.min_size', - index=0, - number=1, + name='contrast_prob', + full_name='caffe.DistortionParameter.contrast_prob', + index=2, + number=3, type=2, cpp_type=6, - label=3, - has_default_value=False, - default_value=[], + label=1, + has_default_value=True, + default_value=float(0), message_type=None, enum_type=None, containing_type=None, @@ -9824,15 +5726,15 @@ _PRIORBOXPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='max_size', - full_name='caffe.PriorBoxParameter.max_size', - index=1, - number=2, + name='contrast_lower', + full_name='caffe.DistortionParameter.contrast_lower', + index=3, + number=4, type=2, cpp_type=6, - label=3, - has_default_value=False, - default_value=[], + label=1, + has_default_value=True, + default_value=float(0), message_type=None, enum_type=None, containing_type=None, @@ -9841,15 +5743,15 @@ _PRIORBOXPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='aspect_ratio', - full_name='caffe.PriorBoxParameter.aspect_ratio', - index=2, - number=3, + name='contrast_upper', + full_name='caffe.DistortionParameter.contrast_upper', + index=4, + number=5, type=2, cpp_type=6, - label=3, - has_default_value=False, - default_value=[], + label=1, + has_default_value=True, + default_value=float(0), message_type=None, enum_type=None, containing_type=None, @@ -9857,48 +5759,16 @@ _PRIORBOXPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='flip', - full_name='caffe.PriorBoxParameter.flip', - index=3, - number=4, - type=8, - cpp_type=7, - label=1, - has_default_value=True, - default_value=True, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='clip', - full_name='caffe.PriorBoxParameter.clip', - index=4, - number=5, - type=8, - cpp_type=7, - label=1, - has_default_value=True, - default_value=False, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='variance', - full_name='caffe.PriorBoxParameter.variance', + name='hue_prob', + full_name='caffe.DistortionParameter.hue_prob', index=5, number=6, type=2, cpp_type=6, - label=3, - has_default_value=False, - default_value=[], + label=1, + has_default_value=True, + default_value=float(0), message_type=None, enum_type=None, containing_type=None, @@ -9907,15 +5777,15 @@ _PRIORBOXPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='img_size', - full_name='caffe.PriorBoxParameter.img_size', + name='hue_delta', + full_name='caffe.DistortionParameter.hue_delta', index=6, number=7, - type=13, - cpp_type=3, + type=2, + cpp_type=6, label=1, - has_default_value=False, - default_value=0, + has_default_value=True, + default_value=float(0), message_type=None, enum_type=None, containing_type=None, @@ -9923,182 +5793,16 @@ _PRIORBOXPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='img_h', - full_name='caffe.PriorBoxParameter.img_h', - index=7, - number=8, - type=13, - cpp_type=3, - label=1, - has_default_value=False, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='img_w', - full_name='caffe.PriorBoxParameter.img_w', - index=8, - number=9, - type=13, - cpp_type=3, - label=1, - has_default_value=False, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='step', - full_name='caffe.PriorBoxParameter.step', - index=9, - number=10, - type=2, - cpp_type=6, - label=1, - has_default_value=False, - default_value=float(0), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='step_h', - full_name='caffe.PriorBoxParameter.step_h', - index=10, - number=11, - type=2, - cpp_type=6, - label=1, - has_default_value=False, - default_value=float(0), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='step_w', - full_name='caffe.PriorBoxParameter.step_w', - index=11, - number=12, - type=2, - cpp_type=6, - label=1, - has_default_value=False, - default_value=float(0), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='offset', - full_name='caffe.PriorBoxParameter.offset', - index=12, - number=13, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(0.5), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[ - _PRIORBOXPARAMETER_CODETYPE, - ], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=15727, - serialized_end=16036, -) - -_PYTHONPARAMETER = _descriptor.Descriptor( - name='PythonParameter', - full_name='caffe.PythonParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='module', - full_name='caffe.PythonParameter.module', - index=0, - number=1, - type=9, - cpp_type=9, - label=1, - has_default_value=False, - default_value=_b("").decode('utf-8'), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='layer', - full_name='caffe.PythonParameter.layer', - index=1, - number=2, - type=9, - cpp_type=9, - label=1, - has_default_value=False, - default_value=_b("").decode('utf-8'), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='param_str', - full_name='caffe.PythonParameter.param_str', - index=2, - number=3, - type=9, - cpp_type=9, - label=1, - has_default_value=True, - default_value=_b("").decode('utf-8'), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='share_in_parallel', - full_name='caffe.PythonParameter.share_in_parallel', - index=3, - number=4, - type=8, - cpp_type=7, + name='saturation_prob', + full_name='caffe.DistortionParameter.saturation_prob', + index=7, + number=8, + type=2, + cpp_type=6, label=1, has_default_value=True, - default_value=False, + default_value=float(0), message_type=None, enum_type=None, containing_type=None, @@ -10106,36 +5810,16 @@ _PYTHONPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=16038, - serialized_end=16141, -) - -_RECURRENTPARAMETER = _descriptor.Descriptor( - name='RecurrentParameter', - full_name='caffe.RecurrentParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ _descriptor.FieldDescriptor( - name='num_output', - full_name='caffe.RecurrentParameter.num_output', - index=0, - number=1, - type=13, - cpp_type=3, + name='saturation_lower', + full_name='caffe.DistortionParameter.saturation_lower', + index=8, + number=9, + type=2, + cpp_type=6, label=1, has_default_value=True, - default_value=0, + default_value=float(0), message_type=None, enum_type=None, containing_type=None, @@ -10144,15 +5828,15 @@ _RECURRENTPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='weight_filler', - full_name='caffe.RecurrentParameter.weight_filler', - index=1, - number=2, - type=11, - cpp_type=10, + name='saturation_upper', + full_name='caffe.DistortionParameter.saturation_upper', + index=9, + number=10, + type=2, + cpp_type=6, label=1, - has_default_value=False, - default_value=None, + has_default_value=True, + default_value=float(0), message_type=None, enum_type=None, containing_type=None, @@ -10161,15 +5845,15 @@ _RECURRENTPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='bias_filler', - full_name='caffe.RecurrentParameter.bias_filler', - index=2, - number=3, - type=11, - cpp_type=10, - label=1, - has_default_value=False, - default_value=None, + name='random_order_prob', + full_name='caffe.DistortionParameter.random_order_prob', + index=10, + number=11, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(0), message_type=None, enum_type=None, containing_type=None, @@ -10177,16 +5861,35 @@ _RECURRENTPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=8686, + serialized_end=9003, ) + +_EXPANSIONPARAMETER = _descriptor.Descriptor( + name='ExpansionParameter', + full_name='caffe.ExpansionParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ _descriptor.FieldDescriptor( - name='debug_info', - full_name='caffe.RecurrentParameter.debug_info', - index=3, - number=4, - type=8, - cpp_type=7, + name='prob', + full_name='caffe.ExpansionParameter.prob', + index=0, + number=1, + type=2, + cpp_type=6, label=1, has_default_value=True, - default_value=False, + default_value=float(1), message_type=None, enum_type=None, containing_type=None, @@ -10195,15 +5898,15 @@ _RECURRENTPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='expose_hidden', - full_name='caffe.RecurrentParameter.expose_hidden', - index=4, - number=5, - type=8, - cpp_type=7, + name='max_expand_ratio', + full_name='caffe.ExpansionParameter.max_expand_ratio', + index=1, + number=2, + type=2, + cpp_type=6, label=1, has_default_value=True, - default_value=False, + default_value=float(1), message_type=None, enum_type=None, containing_type=None, @@ -10220,22 +5923,38 @@ _RECURRENTPARAMETER = _descriptor.Descriptor( syntax='proto2', extension_ranges=[], oneofs=[], - serialized_start=16144, - serialized_end=16336, -) + serialized_start=9005, + serialized_end=9071, ) -_REDUCTIONPARAMETER = _descriptor.Descriptor( - name='ReductionParameter', - full_name='caffe.ReductionParameter', +_LOSSPARAMETER = _descriptor.Descriptor( + name='LossParameter', + full_name='caffe.LossParameter', filename=None, file=DESCRIPTOR, containing_type=None, fields=[ _descriptor.FieldDescriptor( - name='operation', - full_name='caffe.ReductionParameter.operation', + name='ignore_label', + full_name='caffe.LossParameter.ignore_label', index=0, number=1, + type=5, + cpp_type=1, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='normalization', + full_name='caffe.LossParameter.normalization', + index=1, + number=3, type=14, cpp_type=8, label=1, @@ -10248,70 +5967,86 @@ _REDUCTIONPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='axis', - full_name='caffe.ReductionParameter.axis', - index=1, - number=2, - type=5, - cpp_type=1, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='coeff', - full_name='caffe.ReductionParameter.coeff', - index=2, - number=3, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(1), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='normalize', + full_name='caffe.LossParameter.normalize', + index=2, + number=2, + type=8, + cpp_type=7, + label=1, + has_default_value=False, + default_value=False, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), ], extensions=[], nested_types=[], - enum_types=[ - _REDUCTIONPARAMETER_REDUCTIONOP, - ], + enum_types=[_LOSSPARAMETER_NORMALIZATIONMODE, ], serialized_options=None, is_extendable=False, syntax='proto2', extension_ranges=[], oneofs=[], - serialized_start=16339, - serialized_end=16512, -) + serialized_start=9074, + serialized_end=9268, ) -_RELUPARAMETER = _descriptor.Descriptor( - name='ReLUParameter', - full_name='caffe.ReLUParameter', +_ACCURACYPARAMETER = _descriptor.Descriptor( + name='AccuracyParameter', + full_name='caffe.AccuracyParameter', filename=None, file=DESCRIPTOR, containing_type=None, fields=[ _descriptor.FieldDescriptor( - name='negative_slope', - full_name='caffe.ReLUParameter.negative_slope', + name='top_k', + full_name='caffe.AccuracyParameter.top_k', index=0, number=1, - type=2, - cpp_type=6, + type=13, + cpp_type=3, label=1, has_default_value=True, - default_value=float(0), + default_value=1, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='axis', + full_name='caffe.AccuracyParameter.axis', + index=1, + number=2, + type=5, + cpp_type=1, + label=1, + has_default_value=True, + default_value=1, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='ignore_label', + full_name='caffe.AccuracyParameter.ignore_label', + index=2, + number=3, + type=5, + cpp_type=1, + label=1, + has_default_value=False, + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -10319,92 +6054,6 @@ _RELUPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='engine', - full_name='caffe.ReLUParameter.engine', - index=1, - number=2, - type=14, - cpp_type=8, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[ - _RELUPARAMETER_ENGINE, - ], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=16515, - serialized_end=16656, -) - -_RESHAPEPARAMETER = _descriptor.Descriptor( - name='ReshapeParameter', - full_name='caffe.ReshapeParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='shape', - full_name='caffe.ReshapeParameter.shape', - index=0, - number=1, - type=11, - cpp_type=10, - label=1, - has_default_value=False, - default_value=None, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='axis', - full_name='caffe.ReshapeParameter.axis', - index=1, - number=2, - type=5, - cpp_type=1, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='num_axes', - full_name='caffe.ReshapeParameter.num_axes', - index=2, - number=3, - type=5, - cpp_type=1, - label=1, - has_default_value=True, - default_value=-1, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), ], extensions=[], nested_types=[], @@ -10414,91 +6063,26 @@ _RESHAPEPARAMETER = _descriptor.Descriptor( syntax='proto2', extension_ranges=[], oneofs=[], - serialized_start=16658, - serialized_end=16748, -) + serialized_start=9270, + serialized_end=9346, ) -_SCALEPARAMETER = _descriptor.Descriptor( - name='ScaleParameter', - full_name='caffe.ScaleParameter', +_ANNOTATEDDATAPARAMETER = _descriptor.Descriptor( + name='AnnotatedDataParameter', + full_name='caffe.AnnotatedDataParameter', filename=None, file=DESCRIPTOR, containing_type=None, fields=[ - _descriptor.FieldDescriptor(name='axis', - full_name='caffe.ScaleParameter.axis', - index=0, - number=1, - type=5, - cpp_type=1, - label=1, - has_default_value=True, - default_value=1, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='num_axes', - full_name='caffe.ScaleParameter.num_axes', - index=1, - number=2, - type=5, - cpp_type=1, - label=1, - has_default_value=True, - default_value=1, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='filler', - full_name='caffe.ScaleParameter.filler', - index=2, - number=3, - type=11, - cpp_type=10, - label=1, - has_default_value=False, - default_value=None, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='bias_term', - full_name='caffe.ScaleParameter.bias_term', - index=3, - number=4, - type=8, - cpp_type=7, - label=1, - has_default_value=True, - default_value=False, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='bias_filler', - full_name='caffe.ScaleParameter.bias_filler', - index=4, - number=5, + name='batch_sampler', + full_name='caffe.AnnotatedDataParameter.batch_sampler', + index=0, + number=1, type=11, cpp_type=10, - label=1, + label=3, has_default_value=False, - default_value=None, + default_value=[], message_type=None, enum_type=None, containing_type=None, @@ -10506,90 +6090,33 @@ _SCALEPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=16751, - serialized_end=16916, -) - -_SIGMOIDPARAMETER = _descriptor.Descriptor( - name='SigmoidParameter', - full_name='caffe.SigmoidParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='engine', - full_name='caffe.SigmoidParameter.engine', - index=0, - number=1, - type=14, - cpp_type=8, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[ - _SIGMOIDPARAMETER_ENGINE, - ], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=16918, - serialized_end=17038, -) - -_SLICEPARAMETER = _descriptor.Descriptor( - name='SliceParameter', - full_name='caffe.SliceParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='axis', - full_name='caffe.SliceParameter.axis', - index=0, - number=3, - type=5, - cpp_type=1, - label=1, - has_default_value=True, - default_value=1, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='slice_point', - full_name='caffe.SliceParameter.slice_point', + name='label_map_file', + full_name='caffe.AnnotatedDataParameter.label_map_file', index=1, number=2, - type=13, - cpp_type=3, - label=3, + type=9, + cpp_type=9, + label=1, has_default_value=False, - default_value=[], + default_value=_b("").decode('utf-8'), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='anno_type', + full_name='caffe.AnnotatedDataParameter.anno_type', + index=2, + number=3, + type=14, + cpp_type=8, + label=1, + has_default_value=False, + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -10597,22 +6124,6 @@ _SLICEPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='slice_dim', - full_name='caffe.SliceParameter.slice_dim', - index=2, - number=1, - type=13, - cpp_type=3, - label=1, - has_default_value=True, - default_value=1, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), ], extensions=[], nested_types=[], @@ -10622,141 +6133,120 @@ _SLICEPARAMETER = _descriptor.Descriptor( syntax='proto2', extension_ranges=[], oneofs=[], - serialized_start=17040, - serialized_end=17116, -) - -_SOFTMAXPARAMETER = _descriptor.Descriptor( - name='SoftmaxParameter', - full_name='caffe.SoftmaxParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='engine', - full_name='caffe.SoftmaxParameter.engine', - index=0, - number=1, - type=14, - cpp_type=8, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='axis', - full_name='caffe.SoftmaxParameter.axis', - index=1, - number=2, - type=5, - cpp_type=1, - label=1, - has_default_value=True, - default_value=1, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[ - _SOFTMAXPARAMETER_ENGINE, - ], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=17119, - serialized_end=17256, -) + serialized_start=9349, + serialized_end=9498, ) -_TANHPARAMETER = _descriptor.Descriptor( - name='TanHParameter', - full_name='caffe.TanHParameter', +_ARGMAXPARAMETER = _descriptor.Descriptor( + name='ArgMaxParameter', + full_name='caffe.ArgMaxParameter', filename=None, file=DESCRIPTOR, containing_type=None, fields=[ - _descriptor.FieldDescriptor(name='engine', - full_name='caffe.TanHParameter.engine', - index=0, - number=1, - type=14, - cpp_type=8, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='out_max_val', + full_name='caffe.ArgMaxParameter.out_max_val', + index=0, + number=1, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=False, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='top_k', + full_name='caffe.ArgMaxParameter.top_k', + index=1, + number=2, + type=13, + cpp_type=3, + label=1, + has_default_value=True, + default_value=1, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='axis', + full_name='caffe.ArgMaxParameter.axis', + index=2, + number=3, + type=5, + cpp_type=1, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), ], extensions=[], nested_types=[], - enum_types=[ - _TANHPARAMETER_ENGINE, - ], + enum_types=[], serialized_options=None, is_extendable=False, syntax='proto2', extension_ranges=[], oneofs=[], - serialized_start=17258, - serialized_end=17372, -) + serialized_start=9500, + serialized_end=9577, ) -_TILEPARAMETER = _descriptor.Descriptor( - name='TileParameter', - full_name='caffe.TileParameter', +_CONCATPARAMETER = _descriptor.Descriptor( + name='ConcatParameter', + full_name='caffe.ConcatParameter', filename=None, file=DESCRIPTOR, containing_type=None, fields=[ - _descriptor.FieldDescriptor(name='axis', - full_name='caffe.TileParameter.axis', - index=0, - number=1, - type=5, - cpp_type=1, - label=1, - has_default_value=True, - default_value=1, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='tiles', - full_name='caffe.TileParameter.tiles', - index=1, - number=2, - type=5, - cpp_type=1, - label=1, - has_default_value=False, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='axis', + full_name='caffe.ConcatParameter.axis', + index=0, + number=2, + type=5, + cpp_type=1, + label=1, + has_default_value=True, + default_value=1, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='concat_dim', + full_name='caffe.ConcatParameter.concat_dim', + index=1, + number=1, + type=13, + cpp_type=3, + label=1, + has_default_value=True, + default_value=1, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), ], extensions=[], nested_types=[], @@ -10766,27 +6256,60 @@ _TILEPARAMETER = _descriptor.Descriptor( syntax='proto2', extension_ranges=[], oneofs=[], - serialized_start=17374, - serialized_end=17421, -) + serialized_start=9579, + serialized_end=9636, ) -_THRESHOLDPARAMETER = _descriptor.Descriptor( - name='ThresholdParameter', - full_name='caffe.ThresholdParameter', +_BATCHNORMPARAMETER = _descriptor.Descriptor( + name='BatchNormParameter', + full_name='caffe.BatchNormParameter', filename=None, file=DESCRIPTOR, containing_type=None, fields=[ _descriptor.FieldDescriptor( - name='threshold', - full_name='caffe.ThresholdParameter.threshold', + name='use_global_stats', + full_name='caffe.BatchNormParameter.use_global_stats', index=0, number=1, + type=8, + cpp_type=7, + label=1, + has_default_value=False, + default_value=False, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='moving_average_fraction', + full_name='caffe.BatchNormParameter.moving_average_fraction', + index=1, + number=2, type=2, cpp_type=6, label=1, has_default_value=True, - default_value=float(0), + default_value=float(0.999), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='eps', + full_name='caffe.BatchNormParameter.eps', + index=2, + number=3, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(1e-05), message_type=None, enum_type=None, containing_type=None, @@ -10803,27 +6326,26 @@ _THRESHOLDPARAMETER = _descriptor.Descriptor( syntax='proto2', extension_ranges=[], oneofs=[], - serialized_start=17423, - serialized_end=17465, -) + serialized_start=9638, + serialized_end=9744, ) -_VIDEODATAPARAMETER = _descriptor.Descriptor( - name='VideoDataParameter', - full_name='caffe.VideoDataParameter', +_BIASPARAMETER = _descriptor.Descriptor( + name='BiasParameter', + full_name='caffe.BiasParameter', filename=None, file=DESCRIPTOR, containing_type=None, fields=[ _descriptor.FieldDescriptor( - name='video_type', - full_name='caffe.VideoDataParameter.video_type', + name='axis', + full_name='caffe.BiasParameter.axis', index=0, number=1, - type=14, - cpp_type=8, + type=5, + cpp_type=1, label=1, has_default_value=True, - default_value=0, + default_value=1, message_type=None, enum_type=None, containing_type=None, @@ -10832,15 +6354,15 @@ _VIDEODATAPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='device_id', - full_name='caffe.VideoDataParameter.device_id', + name='num_axes', + full_name='caffe.BiasParameter.num_axes', index=1, number=2, type=5, cpp_type=1, label=1, has_default_value=True, - default_value=0, + default_value=1, message_type=None, enum_type=None, containing_type=None, @@ -10849,15 +6371,15 @@ _VIDEODATAPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='video_file', - full_name='caffe.VideoDataParameter.video_file', + name='filler', + full_name='caffe.BiasParameter.filler', index=2, number=3, - type=9, - cpp_type=9, + type=11, + cpp_type=10, label=1, has_default_value=False, - default_value=_b("").decode('utf-8'), + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -10865,16 +6387,52 @@ _VIDEODATAPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=9746, + serialized_end=9839, ) + +_CONTRASTIVELOSSPARAMETER = _descriptor.Descriptor( + name='ContrastiveLossParameter', + full_name='caffe.ContrastiveLossParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ _descriptor.FieldDescriptor( - name='skip_frames', - full_name='caffe.VideoDataParameter.skip_frames', - index=3, - number=4, - type=13, - cpp_type=3, + name='margin', + full_name='caffe.ContrastiveLossParameter.margin', + index=0, + number=1, + type=2, + cpp_type=6, label=1, has_default_value=True, - default_value=0, + default_value=float(1), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='legacy_version', + full_name='caffe.ContrastiveLossParameter.legacy_version', + index=1, + number=2, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=False, message_type=None, enum_type=None, containing_type=None, @@ -10885,35 +6443,5053 @@ _VIDEODATAPARAMETER = _descriptor.Descriptor( ], extensions=[], nested_types=[], - enum_types=[ - _VIDEODATAPARAMETER_VIDEOTYPE, - ], + enum_types=[], serialized_options=None, is_extendable=False, syntax='proto2', extension_ranges=[], oneofs=[], - serialized_start=17468, - serialized_end=17655, -) + serialized_start=9841, + serialized_end=9917, ) -_WINDOWDATAPARAMETER = _descriptor.Descriptor( - name='WindowDataParameter', - full_name='caffe.WindowDataParameter', +_CONVOLUTIONPARAMETER = _descriptor.Descriptor( + name='ConvolutionParameter', + full_name='caffe.ConvolutionParameter', filename=None, file=DESCRIPTOR, containing_type=None, fields=[ _descriptor.FieldDescriptor( - name='source', - full_name='caffe.WindowDataParameter.source', - index=0, + name='num_output', + full_name='caffe.ConvolutionParameter.num_output', + index=0, + number=1, + type=13, + cpp_type=3, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='bias_term', + full_name='caffe.ConvolutionParameter.bias_term', + index=1, + number=2, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=True, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='pad', + full_name='caffe.ConvolutionParameter.pad', + index=2, + number=3, + type=13, + cpp_type=3, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='kernel_size', + full_name='caffe.ConvolutionParameter.kernel_size', + index=3, + number=4, + type=13, + cpp_type=3, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='stride', + full_name='caffe.ConvolutionParameter.stride', + index=4, + number=6, + type=13, + cpp_type=3, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='dilation', + full_name='caffe.ConvolutionParameter.dilation', + index=5, + number=18, + type=13, + cpp_type=3, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='pad_h', + full_name='caffe.ConvolutionParameter.pad_h', + index=6, + number=9, + type=13, + cpp_type=3, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='pad_w', + full_name='caffe.ConvolutionParameter.pad_w', + index=7, + number=10, + type=13, + cpp_type=3, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='kernel_h', + full_name='caffe.ConvolutionParameter.kernel_h', + index=8, + number=11, + type=13, + cpp_type=3, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='kernel_w', + full_name='caffe.ConvolutionParameter.kernel_w', + index=9, + number=12, + type=13, + cpp_type=3, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='stride_h', + full_name='caffe.ConvolutionParameter.stride_h', + index=10, + number=13, + type=13, + cpp_type=3, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='stride_w', + full_name='caffe.ConvolutionParameter.stride_w', + index=11, + number=14, + type=13, + cpp_type=3, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='group', + full_name='caffe.ConvolutionParameter.group', + index=12, + number=5, + type=13, + cpp_type=3, + label=1, + has_default_value=True, + default_value=1, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='weight_filler', + full_name='caffe.ConvolutionParameter.weight_filler', + index=13, + number=7, + type=11, + cpp_type=10, + label=1, + has_default_value=False, + default_value=None, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='bias_filler', + full_name='caffe.ConvolutionParameter.bias_filler', + index=14, + number=8, + type=11, + cpp_type=10, + label=1, + has_default_value=False, + default_value=None, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='engine', + full_name='caffe.ConvolutionParameter.engine', + index=15, + number=15, + type=14, + cpp_type=8, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='axis', + full_name='caffe.ConvolutionParameter.axis', + index=16, + number=16, + type=5, + cpp_type=1, + label=1, + has_default_value=True, + default_value=1, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='force_nd_im2col', + full_name='caffe.ConvolutionParameter.force_nd_im2col', + index=17, + number=17, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=False, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[_CONVOLUTIONPARAMETER_ENGINE, ], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=9920, + serialized_end=10428, ) + +_CROPPARAMETER = _descriptor.Descriptor( + name='CropParameter', + full_name='caffe.CropParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='axis', + full_name='caffe.CropParameter.axis', + index=0, + number=1, + type=5, + cpp_type=1, + label=1, + has_default_value=True, + default_value=2, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='offset', + full_name='caffe.CropParameter.offset', + index=1, + number=2, + type=13, + cpp_type=3, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=10430, + serialized_end=10478, ) + +_DATAPARAMETER = _descriptor.Descriptor( + name='DataParameter', + full_name='caffe.DataParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='source', + full_name='caffe.DataParameter.source', + index=0, + number=1, + type=9, + cpp_type=9, + label=1, + has_default_value=False, + default_value=_b("").decode('utf-8'), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='batch_size', + full_name='caffe.DataParameter.batch_size', + index=1, + number=4, + type=13, + cpp_type=3, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='rand_skip', + full_name='caffe.DataParameter.rand_skip', + index=2, + number=7, + type=13, + cpp_type=3, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='backend', + full_name='caffe.DataParameter.backend', + index=3, + number=8, + type=14, + cpp_type=8, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='scale', + full_name='caffe.DataParameter.scale', + index=4, + number=2, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(1), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='mean_file', + full_name='caffe.DataParameter.mean_file', + index=5, + number=3, + type=9, + cpp_type=9, + label=1, + has_default_value=False, + default_value=_b("").decode('utf-8'), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='crop_size', + full_name='caffe.DataParameter.crop_size', + index=6, + number=5, + type=13, + cpp_type=3, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='mirror', + full_name='caffe.DataParameter.mirror', + index=7, + number=6, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=False, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='force_encoded_color', + full_name='caffe.DataParameter.force_encoded_color', + index=8, + number=9, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=False, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='prefetch', + full_name='caffe.DataParameter.prefetch', + index=9, + number=10, + type=13, + cpp_type=3, + label=1, + has_default_value=True, + default_value=4, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[_DATAPARAMETER_DB, ], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=10481, + serialized_end=10773, ) + +_DETECTIONEVALUATEPARAMETER = _descriptor.Descriptor( + name='DetectionEvaluateParameter', + full_name='caffe.DetectionEvaluateParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='num_classes', + full_name='caffe.DetectionEvaluateParameter.num_classes', + index=0, + number=1, + type=13, + cpp_type=3, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='background_label_id', + full_name='caffe.DetectionEvaluateParameter.background_label_id', + index=1, + number=2, + type=13, + cpp_type=3, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='overlap_threshold', + full_name='caffe.DetectionEvaluateParameter.overlap_threshold', + index=2, + number=3, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(0.5), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='evaluate_difficult_gt', + full_name='caffe.DetectionEvaluateParameter.evaluate_difficult_gt', + index=3, + number=4, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=True, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='name_size_file', + full_name='caffe.DetectionEvaluateParameter.name_size_file', + index=4, + number=5, + type=9, + cpp_type=9, + label=1, + has_default_value=False, + default_value=_b("").decode('utf-8'), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='resize_param', + full_name='caffe.DetectionEvaluateParameter.resize_param', + index=5, + number=6, + type=11, + cpp_type=10, + label=1, + has_default_value=False, + default_value=None, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=10776, + serialized_end=10996, ) + +_NONMAXIMUMSUPPRESSIONPARAMETER = _descriptor.Descriptor( + name='NonMaximumSuppressionParameter', + full_name='caffe.NonMaximumSuppressionParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='nms_threshold', + full_name='caffe.NonMaximumSuppressionParameter.nms_threshold', + index=0, + number=1, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(0.3), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='top_k', + full_name='caffe.NonMaximumSuppressionParameter.top_k', + index=1, + number=2, + type=5, + cpp_type=1, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='eta', + full_name='caffe.NonMaximumSuppressionParameter.eta', + index=2, + number=3, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(1), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=10998, + serialized_end=11089, ) + +_SAVEOUTPUTPARAMETER = _descriptor.Descriptor( + name='SaveOutputParameter', + full_name='caffe.SaveOutputParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='output_directory', + full_name='caffe.SaveOutputParameter.output_directory', + index=0, + number=1, + type=9, + cpp_type=9, + label=1, + has_default_value=False, + default_value=_b("").decode('utf-8'), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='output_name_prefix', + full_name='caffe.SaveOutputParameter.output_name_prefix', + index=1, + number=2, + type=9, + cpp_type=9, + label=1, + has_default_value=False, + default_value=_b("").decode('utf-8'), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='output_format', + full_name='caffe.SaveOutputParameter.output_format', + index=2, + number=3, + type=9, + cpp_type=9, + label=1, + has_default_value=False, + default_value=_b("").decode('utf-8'), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='label_map_file', + full_name='caffe.SaveOutputParameter.label_map_file', + index=3, + number=4, + type=9, + cpp_type=9, + label=1, + has_default_value=False, + default_value=_b("").decode('utf-8'), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='name_size_file', + full_name='caffe.SaveOutputParameter.name_size_file', + index=4, + number=5, + type=9, + cpp_type=9, + label=1, + has_default_value=False, + default_value=_b("").decode('utf-8'), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='num_test_image', + full_name='caffe.SaveOutputParameter.num_test_image', + index=5, + number=6, + type=13, + cpp_type=3, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='resize_param', + full_name='caffe.SaveOutputParameter.resize_param', + index=6, + number=7, + type=11, + cpp_type=10, + label=1, + has_default_value=False, + default_value=None, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=11092, + serialized_end=11308, ) + +_DETECTIONOUTPUTPARAMETER = _descriptor.Descriptor( + name='DetectionOutputParameter', + full_name='caffe.DetectionOutputParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='num_classes', + full_name='caffe.DetectionOutputParameter.num_classes', + index=0, + number=1, + type=13, + cpp_type=3, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='share_location', + full_name='caffe.DetectionOutputParameter.share_location', + index=1, + number=2, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=True, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='background_label_id', + full_name='caffe.DetectionOutputParameter.background_label_id', + index=2, + number=3, + type=5, + cpp_type=1, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='nms_param', + full_name='caffe.DetectionOutputParameter.nms_param', + index=3, + number=4, + type=11, + cpp_type=10, + label=1, + has_default_value=False, + default_value=None, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='save_output_param', + full_name='caffe.DetectionOutputParameter.save_output_param', + index=4, + number=5, + type=11, + cpp_type=10, + label=1, + has_default_value=False, + default_value=None, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='code_type', + full_name='caffe.DetectionOutputParameter.code_type', + index=5, + number=6, + type=14, + cpp_type=8, + label=1, + has_default_value=True, + default_value=1, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='variance_encoded_in_target', + full_name='caffe.DetectionOutputParameter.variance_encoded_in_target', + index=6, + number=8, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=False, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='keep_top_k', + full_name='caffe.DetectionOutputParameter.keep_top_k', + index=7, + number=7, + type=5, + cpp_type=1, + label=1, + has_default_value=True, + default_value=-1, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='confidence_threshold', + full_name='caffe.DetectionOutputParameter.confidence_threshold', + index=8, + number=9, + type=2, + cpp_type=6, + label=1, + has_default_value=False, + default_value=float(0), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='visualize', + full_name='caffe.DetectionOutputParameter.visualize', + index=9, + number=10, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=False, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='visualize_threshold', + full_name='caffe.DetectionOutputParameter.visualize_threshold', + index=10, + number=11, + type=2, + cpp_type=6, + label=1, + has_default_value=False, + default_value=float(0), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='save_file', + full_name='caffe.DetectionOutputParameter.save_file', + index=11, + number=12, + type=9, + cpp_type=9, + label=1, + has_default_value=False, + default_value=_b("").decode('utf-8'), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=11311, + serialized_end=11766, ) + +_DROPOUTPARAMETER = _descriptor.Descriptor( + name='DropoutParameter', + full_name='caffe.DropoutParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='dropout_ratio', + full_name='caffe.DropoutParameter.dropout_ratio', + index=0, + number=1, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(0.5), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=11768, + serialized_end=11814, ) + +_DUMMYDATAPARAMETER = _descriptor.Descriptor( + name='DummyDataParameter', + full_name='caffe.DummyDataParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='data_filler', + full_name='caffe.DummyDataParameter.data_filler', + index=0, + number=1, + type=11, + cpp_type=10, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='shape', + full_name='caffe.DummyDataParameter.shape', + index=1, + number=6, + type=11, + cpp_type=10, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='num', + full_name='caffe.DummyDataParameter.num', + index=2, + number=2, + type=13, + cpp_type=3, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='channels', + full_name='caffe.DummyDataParameter.channels', + index=3, + number=3, + type=13, + cpp_type=3, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='height', + full_name='caffe.DummyDataParameter.height', + index=4, + number=4, + type=13, + cpp_type=3, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='width', + full_name='caffe.DummyDataParameter.width', + index=5, + number=5, + type=13, + cpp_type=3, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=11817, + serialized_end=11977, ) + +_ELTWISEPARAMETER = _descriptor.Descriptor( + name='EltwiseParameter', + full_name='caffe.EltwiseParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='operation', + full_name='caffe.EltwiseParameter.operation', + index=0, + number=1, + type=14, + cpp_type=8, + label=1, + has_default_value=True, + default_value=1, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='coeff', + full_name='caffe.EltwiseParameter.coeff', + index=1, + number=2, + type=2, + cpp_type=6, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='stable_prod_grad', + full_name='caffe.EltwiseParameter.stable_prod_grad', + index=2, + number=3, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=True, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[_ELTWISEPARAMETER_ELTWISEOP, ], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=11980, + serialized_end=12145, ) + +_ELUPARAMETER = _descriptor.Descriptor( + name='ELUParameter', + full_name='caffe.ELUParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='alpha', + full_name='caffe.ELUParameter.alpha', + index=0, + number=1, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(1), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=12147, + serialized_end=12179, ) + +_EMBEDPARAMETER = _descriptor.Descriptor( + name='EmbedParameter', + full_name='caffe.EmbedParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='num_output', + full_name='caffe.EmbedParameter.num_output', + index=0, + number=1, + type=13, + cpp_type=3, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='input_dim', + full_name='caffe.EmbedParameter.input_dim', + index=1, + number=2, + type=13, + cpp_type=3, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='bias_term', + full_name='caffe.EmbedParameter.bias_term', + index=2, + number=3, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=True, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='weight_filler', + full_name='caffe.EmbedParameter.weight_filler', + index=3, + number=4, + type=11, + cpp_type=10, + label=1, + has_default_value=False, + default_value=None, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='bias_filler', + full_name='caffe.EmbedParameter.bias_filler', + index=4, + number=5, + type=11, + cpp_type=10, + label=1, + has_default_value=False, + default_value=None, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=12182, + serialized_end=12354, ) + +_EXPPARAMETER = _descriptor.Descriptor( + name='ExpParameter', + full_name='caffe.ExpParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='base', + full_name='caffe.ExpParameter.base', + index=0, + number=1, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(-1), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='scale', + full_name='caffe.ExpParameter.scale', + index=1, + number=2, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(1), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='shift', + full_name='caffe.ExpParameter.shift', + index=2, + number=3, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(0), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=12356, + serialized_end=12424, ) + +_FLATTENPARAMETER = _descriptor.Descriptor( + name='FlattenParameter', + full_name='caffe.FlattenParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='axis', + full_name='caffe.FlattenParameter.axis', + index=0, + number=1, + type=5, + cpp_type=1, + label=1, + has_default_value=True, + default_value=1, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='end_axis', + full_name='caffe.FlattenParameter.end_axis', + index=1, + number=2, + type=5, + cpp_type=1, + label=1, + has_default_value=True, + default_value=-1, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=12426, + serialized_end=12483, ) + +_HDF5DATAPARAMETER = _descriptor.Descriptor( + name='HDF5DataParameter', + full_name='caffe.HDF5DataParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='source', + full_name='caffe.HDF5DataParameter.source', + index=0, + number=1, + type=9, + cpp_type=9, + label=1, + has_default_value=False, + default_value=_b("").decode('utf-8'), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='batch_size', + full_name='caffe.HDF5DataParameter.batch_size', + index=1, + number=2, + type=13, + cpp_type=3, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='shuffle', + full_name='caffe.HDF5DataParameter.shuffle', + index=2, + number=3, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=False, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=12485, + serialized_end=12564, ) + +_HDF5OUTPUTPARAMETER = _descriptor.Descriptor( + name='HDF5OutputParameter', + full_name='caffe.HDF5OutputParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='file_name', + full_name='caffe.HDF5OutputParameter.file_name', + index=0, + number=1, + type=9, + cpp_type=9, + label=1, + has_default_value=False, + default_value=_b("").decode('utf-8'), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=12566, + serialized_end=12606, ) + +_HINGELOSSPARAMETER = _descriptor.Descriptor( + name='HingeLossParameter', + full_name='caffe.HingeLossParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='norm', + full_name='caffe.HingeLossParameter.norm', + index=0, + number=1, + type=14, + cpp_type=8, + label=1, + has_default_value=True, + default_value=1, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[_HINGELOSSPARAMETER_NORM, ], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=12608, + serialized_end=12702, ) + +_IMAGEDATAPARAMETER = _descriptor.Descriptor( + name='ImageDataParameter', + full_name='caffe.ImageDataParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='source', + full_name='caffe.ImageDataParameter.source', + index=0, + number=1, + type=9, + cpp_type=9, + label=1, + has_default_value=False, + default_value=_b("").decode('utf-8'), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='batch_size', + full_name='caffe.ImageDataParameter.batch_size', + index=1, + number=4, + type=13, + cpp_type=3, + label=1, + has_default_value=True, + default_value=1, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='rand_skip', + full_name='caffe.ImageDataParameter.rand_skip', + index=2, + number=7, + type=13, + cpp_type=3, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='shuffle', + full_name='caffe.ImageDataParameter.shuffle', + index=3, + number=8, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=False, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='new_height', + full_name='caffe.ImageDataParameter.new_height', + index=4, + number=9, + type=13, + cpp_type=3, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='new_width', + full_name='caffe.ImageDataParameter.new_width', + index=5, + number=10, + type=13, + cpp_type=3, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='is_color', + full_name='caffe.ImageDataParameter.is_color', + index=6, + number=11, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=True, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='scale', + full_name='caffe.ImageDataParameter.scale', + index=7, + number=2, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(1), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='mean_file', + full_name='caffe.ImageDataParameter.mean_file', + index=8, + number=3, + type=9, + cpp_type=9, + label=1, + has_default_value=False, + default_value=_b("").decode('utf-8'), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='crop_size', + full_name='caffe.ImageDataParameter.crop_size', + index=9, + number=5, + type=13, + cpp_type=3, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='mirror', + full_name='caffe.ImageDataParameter.mirror', + index=10, + number=6, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=False, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='root_folder', + full_name='caffe.ImageDataParameter.root_folder', + index=11, + number=12, + type=9, + cpp_type=9, + label=1, + has_default_value=True, + default_value=_b("").decode('utf-8'), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=12705, + serialized_end=12984, ) + +_INFOGAINLOSSPARAMETER = _descriptor.Descriptor( + name='InfogainLossParameter', + full_name='caffe.InfogainLossParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='source', + full_name='caffe.InfogainLossParameter.source', + index=0, + number=1, + type=9, + cpp_type=9, + label=1, + has_default_value=False, + default_value=_b("").decode('utf-8'), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=12986, + serialized_end=13025, ) + +_INNERPRODUCTPARAMETER = _descriptor.Descriptor( + name='InnerProductParameter', + full_name='caffe.InnerProductParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='num_output', + full_name='caffe.InnerProductParameter.num_output', + index=0, + number=1, + type=13, + cpp_type=3, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='bias_term', + full_name='caffe.InnerProductParameter.bias_term', + index=1, + number=2, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=True, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='weight_filler', + full_name='caffe.InnerProductParameter.weight_filler', + index=2, + number=3, + type=11, + cpp_type=10, + label=1, + has_default_value=False, + default_value=None, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='bias_filler', + full_name='caffe.InnerProductParameter.bias_filler', + index=3, + number=4, + type=11, + cpp_type=10, + label=1, + has_default_value=False, + default_value=None, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='axis', + full_name='caffe.InnerProductParameter.axis', + index=4, + number=5, + type=5, + cpp_type=1, + label=1, + has_default_value=True, + default_value=1, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='transpose', + full_name='caffe.InnerProductParameter.transpose', + index=5, + number=6, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=False, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=13028, + serialized_end=13231, ) + +_INPUTPARAMETER = _descriptor.Descriptor( + name='InputParameter', + full_name='caffe.InputParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='shape', + full_name='caffe.InputParameter.shape', + index=0, + number=1, + type=11, + cpp_type=10, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=13233, + serialized_end=13282, ) + +_LOGPARAMETER = _descriptor.Descriptor( + name='LogParameter', + full_name='caffe.LogParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='base', + full_name='caffe.LogParameter.base', + index=0, + number=1, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(-1), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='scale', + full_name='caffe.LogParameter.scale', + index=1, + number=2, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(1), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='shift', + full_name='caffe.LogParameter.shift', + index=2, + number=3, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(0), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=13284, + serialized_end=13352, ) + +_LRNPARAMETER = _descriptor.Descriptor( + name='LRNParameter', + full_name='caffe.LRNParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='local_size', + full_name='caffe.LRNParameter.local_size', + index=0, + number=1, + type=13, + cpp_type=3, + label=1, + has_default_value=True, + default_value=5, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='alpha', + full_name='caffe.LRNParameter.alpha', + index=1, + number=2, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(1), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='beta', + full_name='caffe.LRNParameter.beta', + index=2, + number=3, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(0.75), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='norm_region', + full_name='caffe.LRNParameter.norm_region', + index=3, + number=4, + type=14, + cpp_type=8, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='k', + full_name='caffe.LRNParameter.k', + index=4, + number=5, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(1), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='engine', + full_name='caffe.LRNParameter.engine', + index=5, + number=6, + type=14, + cpp_type=8, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[ + _LRNPARAMETER_NORMREGION, + _LRNPARAMETER_ENGINE, + ], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=13355, + serialized_end=13667, ) + +_MEMORYDATAPARAMETER = _descriptor.Descriptor( + name='MemoryDataParameter', + full_name='caffe.MemoryDataParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='batch_size', + full_name='caffe.MemoryDataParameter.batch_size', + index=0, + number=1, + type=13, + cpp_type=3, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='channels', + full_name='caffe.MemoryDataParameter.channels', + index=1, + number=2, + type=13, + cpp_type=3, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='height', + full_name='caffe.MemoryDataParameter.height', + index=2, + number=3, + type=13, + cpp_type=3, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='width', + full_name='caffe.MemoryDataParameter.width', + index=3, + number=4, + type=13, + cpp_type=3, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=13669, + serialized_end=13759, ) + +_MULTIBOXLOSSPARAMETER = _descriptor.Descriptor( + name='MultiBoxLossParameter', + full_name='caffe.MultiBoxLossParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='loc_loss_type', + full_name='caffe.MultiBoxLossParameter.loc_loss_type', + index=0, + number=1, + type=14, + cpp_type=8, + label=1, + has_default_value=True, + default_value=1, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='conf_loss_type', + full_name='caffe.MultiBoxLossParameter.conf_loss_type', + index=1, + number=2, + type=14, + cpp_type=8, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='loc_weight', + full_name='caffe.MultiBoxLossParameter.loc_weight', + index=2, + number=3, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(1), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='num_classes', + full_name='caffe.MultiBoxLossParameter.num_classes', + index=3, + number=4, + type=13, + cpp_type=3, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='share_location', + full_name='caffe.MultiBoxLossParameter.share_location', + index=4, + number=5, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=True, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='match_type', + full_name='caffe.MultiBoxLossParameter.match_type', + index=5, + number=6, + type=14, + cpp_type=8, + label=1, + has_default_value=True, + default_value=1, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='overlap_threshold', + full_name='caffe.MultiBoxLossParameter.overlap_threshold', + index=6, + number=7, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(0.5), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='use_prior_for_matching', + full_name='caffe.MultiBoxLossParameter.use_prior_for_matching', + index=7, + number=8, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=True, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='background_label_id', + full_name='caffe.MultiBoxLossParameter.background_label_id', + index=8, + number=9, + type=13, + cpp_type=3, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='use_difficult_gt', + full_name='caffe.MultiBoxLossParameter.use_difficult_gt', + index=9, + number=10, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=True, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='do_neg_mining', + full_name='caffe.MultiBoxLossParameter.do_neg_mining', + index=10, + number=11, + type=8, + cpp_type=7, + label=1, + has_default_value=False, + default_value=False, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='neg_pos_ratio', + full_name='caffe.MultiBoxLossParameter.neg_pos_ratio', + index=11, + number=12, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(3), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='neg_overlap', + full_name='caffe.MultiBoxLossParameter.neg_overlap', + index=12, + number=13, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(0.5), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='code_type', + full_name='caffe.MultiBoxLossParameter.code_type', + index=13, + number=14, + type=14, + cpp_type=8, + label=1, + has_default_value=True, + default_value=1, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='encode_variance_in_target', + full_name='caffe.MultiBoxLossParameter.encode_variance_in_target', + index=14, + number=16, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=False, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='map_object_to_agnostic', + full_name='caffe.MultiBoxLossParameter.map_object_to_agnostic', + index=15, + number=17, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=False, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='ignore_cross_boundary_bbox', + full_name='caffe.MultiBoxLossParameter.ignore_cross_boundary_bbox', + index=16, + number=18, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=False, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='bp_inside', + full_name='caffe.MultiBoxLossParameter.bp_inside', + index=17, + number=19, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=False, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='mining_type', + full_name='caffe.MultiBoxLossParameter.mining_type', + index=18, + number=20, + type=14, + cpp_type=8, + label=1, + has_default_value=True, + default_value=1, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='nms_param', + full_name='caffe.MultiBoxLossParameter.nms_param', + index=19, + number=21, + type=11, + cpp_type=10, + label=1, + has_default_value=False, + default_value=None, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='sample_size', + full_name='caffe.MultiBoxLossParameter.sample_size', + index=20, + number=22, + type=5, + cpp_type=1, + label=1, + has_default_value=True, + default_value=64, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='use_prior_for_nms', + full_name='caffe.MultiBoxLossParameter.use_prior_for_nms', + index=21, + number=23, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=False, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[ + _MULTIBOXLOSSPARAMETER_LOCLOSSTYPE, + _MULTIBOXLOSSPARAMETER_CONFLOSSTYPE, + _MULTIBOXLOSSPARAMETER_MATCHTYPE, + _MULTIBOXLOSSPARAMETER_MININGTYPE, + ], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=13762, + serialized_end=14890, ) + +_MVNPARAMETER = _descriptor.Descriptor( + name='MVNParameter', + full_name='caffe.MVNParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='normalize_variance', + full_name='caffe.MVNParameter.normalize_variance', + index=0, + number=1, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=True, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='across_channels', + full_name='caffe.MVNParameter.across_channels', + index=1, + number=2, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=False, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='eps', + full_name='caffe.MVNParameter.eps', + index=2, + number=3, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(1e-09), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=14892, + serialized_end=14992, ) + +_NORMALIZEPARAMETER = _descriptor.Descriptor( + name='NormalizeParameter', + full_name='caffe.NormalizeParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='across_spatial', + full_name='caffe.NormalizeParameter.across_spatial', + index=0, + number=1, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=True, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='scale_filler', + full_name='caffe.NormalizeParameter.scale_filler', + index=1, + number=2, + type=11, + cpp_type=10, + label=1, + has_default_value=False, + default_value=None, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='channel_shared', + full_name='caffe.NormalizeParameter.channel_shared', + index=2, + number=3, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=True, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='eps', + full_name='caffe.NormalizeParameter.eps', + index=3, + number=4, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(1e-10), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=14995, + serialized_end=15141, ) + +_PARAMETERPARAMETER = _descriptor.Descriptor( + name='ParameterParameter', + full_name='caffe.ParameterParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='shape', + full_name='caffe.ParameterParameter.shape', + index=0, + number=1, + type=11, + cpp_type=10, + label=1, + has_default_value=False, + default_value=None, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=15143, + serialized_end=15196, ) + +_PERMUTEPARAMETER = _descriptor.Descriptor( + name='PermuteParameter', + full_name='caffe.PermuteParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='order', + full_name='caffe.PermuteParameter.order', + index=0, + number=1, + type=13, + cpp_type=3, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=15198, + serialized_end=15231, ) + +_POOLINGPARAMETER = _descriptor.Descriptor( + name='PoolingParameter', + full_name='caffe.PoolingParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='pool', + full_name='caffe.PoolingParameter.pool', + index=0, + number=1, + type=14, + cpp_type=8, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='pad', + full_name='caffe.PoolingParameter.pad', + index=1, + number=4, + type=13, + cpp_type=3, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='pad_h', + full_name='caffe.PoolingParameter.pad_h', + index=2, + number=9, + type=13, + cpp_type=3, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='pad_w', + full_name='caffe.PoolingParameter.pad_w', + index=3, + number=10, + type=13, + cpp_type=3, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='kernel_size', + full_name='caffe.PoolingParameter.kernel_size', + index=4, + number=2, + type=13, + cpp_type=3, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='kernel_h', + full_name='caffe.PoolingParameter.kernel_h', + index=5, + number=5, + type=13, + cpp_type=3, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='kernel_w', + full_name='caffe.PoolingParameter.kernel_w', + index=6, + number=6, + type=13, + cpp_type=3, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='stride', + full_name='caffe.PoolingParameter.stride', + index=7, + number=3, + type=13, + cpp_type=3, + label=1, + has_default_value=True, + default_value=1, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='stride_h', + full_name='caffe.PoolingParameter.stride_h', + index=8, + number=7, + type=13, + cpp_type=3, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='stride_w', + full_name='caffe.PoolingParameter.stride_w', + index=9, + number=8, + type=13, + cpp_type=3, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='engine', + full_name='caffe.PoolingParameter.engine', + index=10, + number=11, + type=14, + cpp_type=8, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='global_pooling', + full_name='caffe.PoolingParameter.global_pooling', + index=11, + number=12, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=False, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[ + _POOLINGPARAMETER_POOLMETHOD, + _POOLINGPARAMETER_ENGINE, + ], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=15234, + serialized_end=15652, ) + +_POWERPARAMETER = _descriptor.Descriptor( + name='PowerParameter', + full_name='caffe.PowerParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='power', + full_name='caffe.PowerParameter.power', + index=0, + number=1, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(1), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='scale', + full_name='caffe.PowerParameter.scale', + index=1, + number=2, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(1), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='shift', + full_name='caffe.PowerParameter.shift', + index=2, + number=3, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(0), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=15654, + serialized_end=15724, ) + +_PRIORBOXPARAMETER = _descriptor.Descriptor( + name='PriorBoxParameter', + full_name='caffe.PriorBoxParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='min_size', + full_name='caffe.PriorBoxParameter.min_size', + index=0, + number=1, + type=2, + cpp_type=6, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_size', + full_name='caffe.PriorBoxParameter.max_size', + index=1, + number=2, + type=2, + cpp_type=6, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='aspect_ratio', + full_name='caffe.PriorBoxParameter.aspect_ratio', + index=2, + number=3, + type=2, + cpp_type=6, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='flip', + full_name='caffe.PriorBoxParameter.flip', + index=3, + number=4, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=True, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='clip', + full_name='caffe.PriorBoxParameter.clip', + index=4, + number=5, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=False, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='variance', + full_name='caffe.PriorBoxParameter.variance', + index=5, + number=6, + type=2, + cpp_type=6, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='img_size', + full_name='caffe.PriorBoxParameter.img_size', + index=6, + number=7, + type=13, + cpp_type=3, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='img_h', + full_name='caffe.PriorBoxParameter.img_h', + index=7, + number=8, + type=13, + cpp_type=3, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='img_w', + full_name='caffe.PriorBoxParameter.img_w', + index=8, + number=9, + type=13, + cpp_type=3, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='step', + full_name='caffe.PriorBoxParameter.step', + index=9, + number=10, + type=2, + cpp_type=6, + label=1, + has_default_value=False, + default_value=float(0), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='step_h', + full_name='caffe.PriorBoxParameter.step_h', + index=10, + number=11, + type=2, + cpp_type=6, + label=1, + has_default_value=False, + default_value=float(0), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='step_w', + full_name='caffe.PriorBoxParameter.step_w', + index=11, + number=12, + type=2, + cpp_type=6, + label=1, + has_default_value=False, + default_value=float(0), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='offset', + full_name='caffe.PriorBoxParameter.offset', + index=12, + number=13, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(0.5), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[_PRIORBOXPARAMETER_CODETYPE, ], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=15727, + serialized_end=16036, ) + +_PYTHONPARAMETER = _descriptor.Descriptor( + name='PythonParameter', + full_name='caffe.PythonParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='module', + full_name='caffe.PythonParameter.module', + index=0, + number=1, + type=9, + cpp_type=9, + label=1, + has_default_value=False, + default_value=_b("").decode('utf-8'), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='layer', + full_name='caffe.PythonParameter.layer', + index=1, + number=2, + type=9, + cpp_type=9, + label=1, + has_default_value=False, + default_value=_b("").decode('utf-8'), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='param_str', + full_name='caffe.PythonParameter.param_str', + index=2, + number=3, + type=9, + cpp_type=9, + label=1, + has_default_value=True, + default_value=_b("").decode('utf-8'), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='share_in_parallel', + full_name='caffe.PythonParameter.share_in_parallel', + index=3, + number=4, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=False, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=16038, + serialized_end=16141, ) + +_RECURRENTPARAMETER = _descriptor.Descriptor( + name='RecurrentParameter', + full_name='caffe.RecurrentParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='num_output', + full_name='caffe.RecurrentParameter.num_output', + index=0, + number=1, + type=13, + cpp_type=3, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='weight_filler', + full_name='caffe.RecurrentParameter.weight_filler', + index=1, + number=2, + type=11, + cpp_type=10, + label=1, + has_default_value=False, + default_value=None, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='bias_filler', + full_name='caffe.RecurrentParameter.bias_filler', + index=2, + number=3, + type=11, + cpp_type=10, + label=1, + has_default_value=False, + default_value=None, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='debug_info', + full_name='caffe.RecurrentParameter.debug_info', + index=3, + number=4, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=False, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='expose_hidden', + full_name='caffe.RecurrentParameter.expose_hidden', + index=4, + number=5, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=False, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=16144, + serialized_end=16336, ) + +_REDUCTIONPARAMETER = _descriptor.Descriptor( + name='ReductionParameter', + full_name='caffe.ReductionParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='operation', + full_name='caffe.ReductionParameter.operation', + index=0, + number=1, + type=14, + cpp_type=8, + label=1, + has_default_value=True, + default_value=1, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='axis', + full_name='caffe.ReductionParameter.axis', + index=1, + number=2, + type=5, + cpp_type=1, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='coeff', + full_name='caffe.ReductionParameter.coeff', + index=2, + number=3, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(1), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[_REDUCTIONPARAMETER_REDUCTIONOP, ], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=16339, + serialized_end=16512, ) + +_RELUPARAMETER = _descriptor.Descriptor( + name='ReLUParameter', + full_name='caffe.ReLUParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='negative_slope', + full_name='caffe.ReLUParameter.negative_slope', + index=0, + number=1, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(0), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='engine', + full_name='caffe.ReLUParameter.engine', + index=1, + number=2, + type=14, + cpp_type=8, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[_RELUPARAMETER_ENGINE, ], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=16515, + serialized_end=16656, ) + +_RESHAPEPARAMETER = _descriptor.Descriptor( + name='ReshapeParameter', + full_name='caffe.ReshapeParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='shape', + full_name='caffe.ReshapeParameter.shape', + index=0, + number=1, + type=11, + cpp_type=10, + label=1, + has_default_value=False, + default_value=None, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='axis', + full_name='caffe.ReshapeParameter.axis', + index=1, + number=2, + type=5, + cpp_type=1, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='num_axes', + full_name='caffe.ReshapeParameter.num_axes', + index=2, + number=3, + type=5, + cpp_type=1, + label=1, + has_default_value=True, + default_value=-1, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=16658, + serialized_end=16748, ) + +_SCALEPARAMETER = _descriptor.Descriptor( + name='ScaleParameter', + full_name='caffe.ScaleParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='axis', + full_name='caffe.ScaleParameter.axis', + index=0, + number=1, + type=5, + cpp_type=1, + label=1, + has_default_value=True, + default_value=1, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='num_axes', + full_name='caffe.ScaleParameter.num_axes', + index=1, + number=2, + type=5, + cpp_type=1, + label=1, + has_default_value=True, + default_value=1, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='filler', + full_name='caffe.ScaleParameter.filler', + index=2, + number=3, + type=11, + cpp_type=10, + label=1, + has_default_value=False, + default_value=None, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='bias_term', + full_name='caffe.ScaleParameter.bias_term', + index=3, + number=4, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=False, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='bias_filler', + full_name='caffe.ScaleParameter.bias_filler', + index=4, + number=5, + type=11, + cpp_type=10, + label=1, + has_default_value=False, + default_value=None, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=16751, + serialized_end=16916, ) + +_SIGMOIDPARAMETER = _descriptor.Descriptor( + name='SigmoidParameter', + full_name='caffe.SigmoidParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='engine', + full_name='caffe.SigmoidParameter.engine', + index=0, + number=1, + type=14, + cpp_type=8, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[_SIGMOIDPARAMETER_ENGINE, ], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=16918, + serialized_end=17038, ) + +_SLICEPARAMETER = _descriptor.Descriptor( + name='SliceParameter', + full_name='caffe.SliceParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='axis', + full_name='caffe.SliceParameter.axis', + index=0, + number=3, + type=5, + cpp_type=1, + label=1, + has_default_value=True, + default_value=1, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='slice_point', + full_name='caffe.SliceParameter.slice_point', + index=1, + number=2, + type=13, + cpp_type=3, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='slice_dim', + full_name='caffe.SliceParameter.slice_dim', + index=2, + number=1, + type=13, + cpp_type=3, + label=1, + has_default_value=True, + default_value=1, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=17040, + serialized_end=17116, ) + +_SOFTMAXPARAMETER = _descriptor.Descriptor( + name='SoftmaxParameter', + full_name='caffe.SoftmaxParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='engine', + full_name='caffe.SoftmaxParameter.engine', + index=0, + number=1, + type=14, + cpp_type=8, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='axis', + full_name='caffe.SoftmaxParameter.axis', + index=1, + number=2, + type=5, + cpp_type=1, + label=1, + has_default_value=True, + default_value=1, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[_SOFTMAXPARAMETER_ENGINE, ], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=17119, + serialized_end=17256, ) + +_TANHPARAMETER = _descriptor.Descriptor( + name='TanHParameter', + full_name='caffe.TanHParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='engine', + full_name='caffe.TanHParameter.engine', + index=0, + number=1, + type=14, + cpp_type=8, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[_TANHPARAMETER_ENGINE, ], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=17258, + serialized_end=17372, ) + +_TILEPARAMETER = _descriptor.Descriptor( + name='TileParameter', + full_name='caffe.TileParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='axis', + full_name='caffe.TileParameter.axis', + index=0, + number=1, + type=5, + cpp_type=1, + label=1, + has_default_value=True, + default_value=1, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='tiles', + full_name='caffe.TileParameter.tiles', + index=1, + number=2, + type=5, + cpp_type=1, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=17374, + serialized_end=17421, ) + +_THRESHOLDPARAMETER = _descriptor.Descriptor( + name='ThresholdParameter', + full_name='caffe.ThresholdParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='threshold', + full_name='caffe.ThresholdParameter.threshold', + index=0, + number=1, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(0), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=17423, + serialized_end=17465, ) + +_VIDEODATAPARAMETER = _descriptor.Descriptor( + name='VideoDataParameter', + full_name='caffe.VideoDataParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='video_type', + full_name='caffe.VideoDataParameter.video_type', + index=0, + number=1, + type=14, + cpp_type=8, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='device_id', + full_name='caffe.VideoDataParameter.device_id', + index=1, + number=2, + type=5, + cpp_type=1, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='video_file', + full_name='caffe.VideoDataParameter.video_file', + index=2, + number=3, + type=9, + cpp_type=9, + label=1, + has_default_value=False, + default_value=_b("").decode('utf-8'), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='skip_frames', + full_name='caffe.VideoDataParameter.skip_frames', + index=3, + number=4, + type=13, + cpp_type=3, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[_VIDEODATAPARAMETER_VIDEOTYPE, ], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=17468, + serialized_end=17655, ) + +_WINDOWDATAPARAMETER = _descriptor.Descriptor( + name='WindowDataParameter', + full_name='caffe.WindowDataParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='source', + full_name='caffe.WindowDataParameter.source', + index=0, + number=1, + type=9, + cpp_type=9, + label=1, + has_default_value=False, + default_value=_b("").decode('utf-8'), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='scale', + full_name='caffe.WindowDataParameter.scale', + index=1, + number=2, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(1), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='mean_file', + full_name='caffe.WindowDataParameter.mean_file', + index=2, + number=3, + type=9, + cpp_type=9, + label=1, + has_default_value=False, + default_value=_b("").decode('utf-8'), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='batch_size', + full_name='caffe.WindowDataParameter.batch_size', + index=3, + number=4, + type=13, + cpp_type=3, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='crop_size', + full_name='caffe.WindowDataParameter.crop_size', + index=4, + number=5, + type=13, + cpp_type=3, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='mirror', + full_name='caffe.WindowDataParameter.mirror', + index=5, + number=6, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=False, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='fg_threshold', + full_name='caffe.WindowDataParameter.fg_threshold', + index=6, + number=7, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(0.5), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='bg_threshold', + full_name='caffe.WindowDataParameter.bg_threshold', + index=7, + number=8, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(0.5), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='fg_fraction', + full_name='caffe.WindowDataParameter.fg_fraction', + index=8, + number=9, + type=2, + cpp_type=6, + label=1, + has_default_value=True, + default_value=float(0.25), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='context_pad', + full_name='caffe.WindowDataParameter.context_pad', + index=9, + number=10, + type=13, + cpp_type=3, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='crop_mode', + full_name='caffe.WindowDataParameter.crop_mode', + index=10, + number=11, + type=9, + cpp_type=9, + label=1, + has_default_value=True, + default_value=_b("warp").decode('utf-8'), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='cache_images', + full_name='caffe.WindowDataParameter.cache_images', + index=11, + number=12, + type=8, + cpp_type=7, + label=1, + has_default_value=True, + default_value=False, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='root_folder', + full_name='caffe.WindowDataParameter.root_folder', + index=12, + number=13, + type=9, + cpp_type=9, + label=1, + has_default_value=True, + default_value=_b("").decode('utf-8'), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=17658, + serialized_end=17979, ) + +_SPPPARAMETER = _descriptor.Descriptor( + name='SPPParameter', + full_name='caffe.SPPParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='pyramid_height', + full_name='caffe.SPPParameter.pyramid_height', + index=0, number=1, + type=13, + cpp_type=3, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='pool', + full_name='caffe.SPPParameter.pool', + index=1, + number=2, + type=14, + cpp_type=8, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='engine', + full_name='caffe.SPPParameter.engine', + index=2, + number=6, + type=14, + cpp_type=8, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[ + _SPPPARAMETER_POOLMETHOD, + _SPPPARAMETER_ENGINE, + ], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=17982, + serialized_end=18217, ) + +_V1LAYERPARAMETER = _descriptor.Descriptor( + name='V1LayerParameter', + full_name='caffe.V1LayerParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='bottom', + full_name='caffe.V1LayerParameter.bottom', + index=0, + number=2, + type=9, + cpp_type=9, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='top', + full_name='caffe.V1LayerParameter.top', + index=1, + number=3, + type=9, + cpp_type=9, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='name', + full_name='caffe.V1LayerParameter.name', + index=2, + number=4, + type=9, + cpp_type=9, + label=1, + has_default_value=False, + default_value=_b("").decode('utf-8'), + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='include', + full_name='caffe.V1LayerParameter.include', + index=3, + number=32, + type=11, + cpp_type=10, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='exclude', + full_name='caffe.V1LayerParameter.exclude', + index=4, + number=33, + type=11, + cpp_type=10, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='type', + full_name='caffe.V1LayerParameter.type', + index=5, + number=5, + type=14, + cpp_type=8, + label=1, + has_default_value=False, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='blobs', + full_name='caffe.V1LayerParameter.blobs', + index=6, + number=6, + type=11, + cpp_type=10, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='param', + full_name='caffe.V1LayerParameter.param', + index=7, + number=1001, type=9, cpp_type=9, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='blob_share_mode', + full_name='caffe.V1LayerParameter.blob_share_mode', + index=8, + number=1002, + type=14, + cpp_type=8, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='blobs_lr', + full_name='caffe.V1LayerParameter.blobs_lr', + index=9, + number=7, + type=2, + cpp_type=6, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='weight_decay', + full_name='caffe.V1LayerParameter.weight_decay', + index=10, + number=8, + type=2, + cpp_type=6, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='loss_weight', + full_name='caffe.V1LayerParameter.loss_weight', + index=11, + number=35, + type=2, + cpp_type=6, + label=3, + has_default_value=False, + default_value=[], + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='accuracy_param', + full_name='caffe.V1LayerParameter.accuracy_param', + index=12, + number=27, + type=11, + cpp_type=10, + label=1, + has_default_value=False, + default_value=None, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='argmax_param', + full_name='caffe.V1LayerParameter.argmax_param', + index=13, + number=23, + type=11, + cpp_type=10, + label=1, + has_default_value=False, + default_value=None, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='concat_param', + full_name='caffe.V1LayerParameter.concat_param', + index=14, + number=9, + type=11, + cpp_type=10, label=1, has_default_value=False, - default_value=_b("").decode('utf-8'), + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -10921,32 +11497,16 @@ _WINDOWDATAPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='scale', - full_name='caffe.WindowDataParameter.scale', - index=1, - number=2, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(1), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='mean_file', - full_name='caffe.WindowDataParameter.mean_file', - index=2, - number=3, - type=9, - cpp_type=9, + name='contrastive_loss_param', + full_name='caffe.V1LayerParameter.contrastive_loss_param', + index=15, + number=40, + type=11, + cpp_type=10, label=1, has_default_value=False, - default_value=_b("").decode('utf-8'), + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -10955,15 +11515,15 @@ _WINDOWDATAPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='batch_size', - full_name='caffe.WindowDataParameter.batch_size', - index=3, - number=4, - type=13, - cpp_type=3, + name='convolution_param', + full_name='caffe.V1LayerParameter.convolution_param', + index=16, + number=10, + type=11, + cpp_type=10, label=1, has_default_value=False, - default_value=0, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -10972,15 +11532,15 @@ _WINDOWDATAPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='crop_size', - full_name='caffe.WindowDataParameter.crop_size', - index=4, - number=5, - type=13, - cpp_type=3, + name='data_param', + full_name='caffe.V1LayerParameter.data_param', + index=17, + number=11, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=0, + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -10989,15 +11549,15 @@ _WINDOWDATAPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='mirror', - full_name='caffe.WindowDataParameter.mirror', - index=5, - number=6, - type=8, - cpp_type=7, + name='dropout_param', + full_name='caffe.V1LayerParameter.dropout_param', + index=18, + number=12, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=False, + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -11006,15 +11566,15 @@ _WINDOWDATAPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='fg_threshold', - full_name='caffe.WindowDataParameter.fg_threshold', - index=6, - number=7, - type=2, - cpp_type=6, + name='dummy_data_param', + full_name='caffe.V1LayerParameter.dummy_data_param', + index=19, + number=26, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=float(0.5), + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -11023,15 +11583,15 @@ _WINDOWDATAPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='bg_threshold', - full_name='caffe.WindowDataParameter.bg_threshold', - index=7, - number=8, - type=2, - cpp_type=6, + name='eltwise_param', + full_name='caffe.V1LayerParameter.eltwise_param', + index=20, + number=24, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=float(0.5), + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -11040,15 +11600,15 @@ _WINDOWDATAPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='fg_fraction', - full_name='caffe.WindowDataParameter.fg_fraction', - index=8, - number=9, - type=2, - cpp_type=6, + name='exp_param', + full_name='caffe.V1LayerParameter.exp_param', + index=21, + number=41, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=float(0.25), + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -11057,15 +11617,15 @@ _WINDOWDATAPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='context_pad', - full_name='caffe.WindowDataParameter.context_pad', - index=9, - number=10, - type=13, - cpp_type=3, + name='hdf5_data_param', + full_name='caffe.V1LayerParameter.hdf5_data_param', + index=22, + number=13, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=0, + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -11074,15 +11634,15 @@ _WINDOWDATAPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='crop_mode', - full_name='caffe.WindowDataParameter.crop_mode', - index=10, - number=11, - type=9, - cpp_type=9, + name='hdf5_output_param', + full_name='caffe.V1LayerParameter.hdf5_output_param', + index=23, + number=14, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=_b("warp").decode('utf-8'), + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -11091,15 +11651,15 @@ _WINDOWDATAPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='cache_images', - full_name='caffe.WindowDataParameter.cache_images', - index=11, - number=12, - type=8, - cpp_type=7, + name='hinge_loss_param', + full_name='caffe.V1LayerParameter.hinge_loss_param', + index=24, + number=29, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=False, + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -11108,15 +11668,15 @@ _WINDOWDATAPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='root_folder', - full_name='caffe.WindowDataParameter.root_folder', - index=12, - number=13, - type=9, - cpp_type=9, + name='image_data_param', + full_name='caffe.V1LayerParameter.image_data_param', + index=25, + number=15, + type=11, + cpp_type=10, label=1, - has_default_value=True, - default_value=_b("").decode('utf-8'), + has_default_value=False, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -11124,36 +11684,16 @@ _WINDOWDATAPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=17658, - serialized_end=17979, -) - -_SPPPARAMETER = _descriptor.Descriptor( - name='SPPParameter', - full_name='caffe.SPPParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ _descriptor.FieldDescriptor( - name='pyramid_height', - full_name='caffe.SPPParameter.pyramid_height', - index=0, - number=1, - type=13, - cpp_type=3, + name='infogain_loss_param', + full_name='caffe.V1LayerParameter.infogain_loss_param', + index=26, + number=16, + type=11, + cpp_type=10, label=1, has_default_value=False, - default_value=0, + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -11161,199 +11701,16 @@ _SPPPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='pool', - full_name='caffe.SPPParameter.pool', - index=1, - number=2, - type=14, - cpp_type=8, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='engine', - full_name='caffe.SPPParameter.engine', - index=2, - number=6, - type=14, - cpp_type=8, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[ - _SPPPARAMETER_POOLMETHOD, - _SPPPARAMETER_ENGINE, - ], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=17982, - serialized_end=18217, -) - -_V1LAYERPARAMETER = _descriptor.Descriptor( - name='V1LayerParameter', - full_name='caffe.V1LayerParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='bottom', - full_name='caffe.V1LayerParameter.bottom', - index=0, - number=2, - type=9, - cpp_type=9, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='top', - full_name='caffe.V1LayerParameter.top', - index=1, - number=3, - type=9, - cpp_type=9, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='name', - full_name='caffe.V1LayerParameter.name', - index=2, - number=4, - type=9, - cpp_type=9, - label=1, - has_default_value=False, - default_value=_b("").decode('utf-8'), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='include', - full_name='caffe.V1LayerParameter.include', - index=3, - number=32, - type=11, - cpp_type=10, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='exclude', - full_name='caffe.V1LayerParameter.exclude', - index=4, - number=33, - type=11, - cpp_type=10, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='type', - full_name='caffe.V1LayerParameter.type', - index=5, - number=5, - type=14, - cpp_type=8, - label=1, - has_default_value=False, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='blobs', - full_name='caffe.V1LayerParameter.blobs', - index=6, - number=6, - type=11, - cpp_type=10, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='param', - full_name='caffe.V1LayerParameter.param', - index=7, - number=1001, - type=9, - cpp_type=9, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='blob_share_mode', - full_name='caffe.V1LayerParameter.blob_share_mode', - index=8, - number=1002, - type=14, - cpp_type=8, - label=3, + name='inner_product_param', + full_name='caffe.V1LayerParameter.inner_product_param', + index=27, + number=17, + type=11, + cpp_type=10, + label=1, has_default_value=False, - default_value=[], + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -11361,32 +11718,16 @@ _V1LAYERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='blobs_lr', - full_name='caffe.V1LayerParameter.blobs_lr', - index=9, - number=7, - type=2, - cpp_type=6, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='weight_decay', - full_name='caffe.V1LayerParameter.weight_decay', - index=10, - number=8, - type=2, - cpp_type=6, - label=3, + name='lrn_param', + full_name='caffe.V1LayerParameter.lrn_param', + index=28, + number=18, + type=11, + cpp_type=10, + label=1, has_default_value=False, - default_value=[], + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -11395,15 +11736,15 @@ _V1LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='loss_weight', - full_name='caffe.V1LayerParameter.loss_weight', - index=11, - number=35, - type=2, - cpp_type=6, - label=3, + name='memory_data_param', + full_name='caffe.V1LayerParameter.memory_data_param', + index=29, + number=22, + type=11, + cpp_type=10, + label=1, has_default_value=False, - default_value=[], + default_value=None, message_type=None, enum_type=None, containing_type=None, @@ -11412,10 +11753,10 @@ _V1LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='accuracy_param', - full_name='caffe.V1LayerParameter.accuracy_param', - index=12, - number=27, + name='mvn_param', + full_name='caffe.V1LayerParameter.mvn_param', + index=30, + number=34, type=11, cpp_type=10, label=1, @@ -11429,10 +11770,10 @@ _V1LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='argmax_param', - full_name='caffe.V1LayerParameter.argmax_param', - index=13, - number=23, + name='pooling_param', + full_name='caffe.V1LayerParameter.pooling_param', + index=31, + number=19, type=11, cpp_type=10, label=1, @@ -11446,10 +11787,10 @@ _V1LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='concat_param', - full_name='caffe.V1LayerParameter.concat_param', - index=14, - number=9, + name='power_param', + full_name='caffe.V1LayerParameter.power_param', + index=32, + number=21, type=11, cpp_type=10, label=1, @@ -11463,10 +11804,10 @@ _V1LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='contrastive_loss_param', - full_name='caffe.V1LayerParameter.contrastive_loss_param', - index=15, - number=40, + name='relu_param', + full_name='caffe.V1LayerParameter.relu_param', + index=33, + number=30, type=11, cpp_type=10, label=1, @@ -11480,10 +11821,10 @@ _V1LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='convolution_param', - full_name='caffe.V1LayerParameter.convolution_param', - index=16, - number=10, + name='sigmoid_param', + full_name='caffe.V1LayerParameter.sigmoid_param', + index=34, + number=38, type=11, cpp_type=10, label=1, @@ -11497,10 +11838,10 @@ _V1LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='data_param', - full_name='caffe.V1LayerParameter.data_param', - index=17, - number=11, + name='softmax_param', + full_name='caffe.V1LayerParameter.softmax_param', + index=35, + number=39, type=11, cpp_type=10, label=1, @@ -11514,10 +11855,10 @@ _V1LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='dropout_param', - full_name='caffe.V1LayerParameter.dropout_param', - index=18, - number=12, + name='slice_param', + full_name='caffe.V1LayerParameter.slice_param', + index=36, + number=31, type=11, cpp_type=10, label=1, @@ -11531,10 +11872,10 @@ _V1LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='dummy_data_param', - full_name='caffe.V1LayerParameter.dummy_data_param', - index=19, - number=26, + name='tanh_param', + full_name='caffe.V1LayerParameter.tanh_param', + index=37, + number=37, type=11, cpp_type=10, label=1, @@ -11548,10 +11889,10 @@ _V1LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='eltwise_param', - full_name='caffe.V1LayerParameter.eltwise_param', - index=20, - number=24, + name='threshold_param', + full_name='caffe.V1LayerParameter.threshold_param', + index=38, + number=25, type=11, cpp_type=10, label=1, @@ -11565,10 +11906,10 @@ _V1LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='exp_param', - full_name='caffe.V1LayerParameter.exp_param', - index=21, - number=41, + name='window_data_param', + full_name='caffe.V1LayerParameter.window_data_param', + index=39, + number=20, type=11, cpp_type=10, label=1, @@ -11582,10 +11923,10 @@ _V1LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='hdf5_data_param', - full_name='caffe.V1LayerParameter.hdf5_data_param', - index=22, - number=13, + name='transform_param', + full_name='caffe.V1LayerParameter.transform_param', + index=40, + number=36, type=11, cpp_type=10, label=1, @@ -11599,10 +11940,10 @@ _V1LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='hdf5_output_param', - full_name='caffe.V1LayerParameter.hdf5_output_param', - index=23, - number=14, + name='loss_param', + full_name='caffe.V1LayerParameter.loss_param', + index=41, + number=42, type=11, cpp_type=10, label=1, @@ -11616,10 +11957,10 @@ _V1LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='hinge_loss_param', - full_name='caffe.V1LayerParameter.hinge_loss_param', - index=24, - number=29, + name='layer', + full_name='caffe.V1LayerParameter.layer', + index=42, + number=1, type=11, cpp_type=10, label=1, @@ -11632,16 +11973,38 @@ _V1LAYERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), + ], + extensions=[], + nested_types=[], + enum_types=[ + _V1LAYERPARAMETER_LAYERTYPE, + _V1LAYERPARAMETER_DIMCHECKMODE, + ], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[], + serialized_start=18220, + serialized_end=20748, ) + +_V0LAYERPARAMETER = _descriptor.Descriptor( + name='V0LayerParameter', + full_name='caffe.V0LayerParameter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ _descriptor.FieldDescriptor( - name='image_data_param', - full_name='caffe.V1LayerParameter.image_data_param', - index=25, - number=15, - type=11, - cpp_type=10, + name='name', + full_name='caffe.V0LayerParameter.name', + index=0, + number=1, + type=9, + cpp_type=9, label=1, has_default_value=False, - default_value=None, + default_value=_b("").decode('utf-8'), message_type=None, enum_type=None, containing_type=None, @@ -11650,15 +12013,15 @@ _V1LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='infogain_loss_param', - full_name='caffe.V1LayerParameter.infogain_loss_param', - index=26, - number=16, - type=11, - cpp_type=10, + name='type', + full_name='caffe.V0LayerParameter.type', + index=1, + number=2, + type=9, + cpp_type=9, label=1, has_default_value=False, - default_value=None, + default_value=_b("").decode('utf-8'), message_type=None, enum_type=None, containing_type=None, @@ -11667,15 +12030,15 @@ _V1LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='inner_product_param', - full_name='caffe.V1LayerParameter.inner_product_param', - index=27, - number=17, - type=11, - cpp_type=10, + name='num_output', + full_name='caffe.V0LayerParameter.num_output', + index=2, + number=3, + type=13, + cpp_type=3, label=1, has_default_value=False, - default_value=None, + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -11684,15 +12047,15 @@ _V1LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='lrn_param', - full_name='caffe.V1LayerParameter.lrn_param', - index=28, - number=18, - type=11, - cpp_type=10, + name='biasterm', + full_name='caffe.V0LayerParameter.biasterm', + index=3, + number=4, + type=8, + cpp_type=7, label=1, - has_default_value=False, - default_value=None, + has_default_value=True, + default_value=True, message_type=None, enum_type=None, containing_type=None, @@ -11701,10 +12064,10 @@ _V1LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='memory_data_param', - full_name='caffe.V1LayerParameter.memory_data_param', - index=29, - number=22, + name='weight_filler', + full_name='caffe.V0LayerParameter.weight_filler', + index=4, + number=5, type=11, cpp_type=10, label=1, @@ -11718,10 +12081,10 @@ _V1LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='mvn_param', - full_name='caffe.V1LayerParameter.mvn_param', - index=30, - number=34, + name='bias_filler', + full_name='caffe.V0LayerParameter.bias_filler', + index=5, + number=6, type=11, cpp_type=10, label=1, @@ -11735,15 +12098,15 @@ _V1LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='pooling_param', - full_name='caffe.V1LayerParameter.pooling_param', - index=31, - number=19, - type=11, - cpp_type=10, + name='pad', + full_name='caffe.V0LayerParameter.pad', + index=6, + number=7, + type=13, + cpp_type=3, label=1, - has_default_value=False, - default_value=None, + has_default_value=True, + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -11752,15 +12115,15 @@ _V1LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='power_param', - full_name='caffe.V1LayerParameter.power_param', - index=32, - number=21, - type=11, - cpp_type=10, + name='kernelsize', + full_name='caffe.V0LayerParameter.kernelsize', + index=7, + number=8, + type=13, + cpp_type=3, label=1, has_default_value=False, - default_value=None, + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -11769,15 +12132,15 @@ _V1LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='relu_param', - full_name='caffe.V1LayerParameter.relu_param', - index=33, - number=30, - type=11, - cpp_type=10, + name='group', + full_name='caffe.V0LayerParameter.group', + index=8, + number=9, + type=13, + cpp_type=3, label=1, - has_default_value=False, - default_value=None, + has_default_value=True, + default_value=1, message_type=None, enum_type=None, containing_type=None, @@ -11786,15 +12149,15 @@ _V1LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='sigmoid_param', - full_name='caffe.V1LayerParameter.sigmoid_param', - index=34, - number=38, - type=11, - cpp_type=10, + name='stride', + full_name='caffe.V0LayerParameter.stride', + index=9, + number=10, + type=13, + cpp_type=3, label=1, - has_default_value=False, - default_value=None, + has_default_value=True, + default_value=1, message_type=None, enum_type=None, containing_type=None, @@ -11803,15 +12166,15 @@ _V1LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='softmax_param', - full_name='caffe.V1LayerParameter.softmax_param', - index=35, - number=39, - type=11, - cpp_type=10, + name='pool', + full_name='caffe.V0LayerParameter.pool', + index=10, + number=11, + type=14, + cpp_type=8, label=1, - has_default_value=False, - default_value=None, + has_default_value=True, + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -11820,15 +12183,15 @@ _V1LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='slice_param', - full_name='caffe.V1LayerParameter.slice_param', - index=36, - number=31, - type=11, - cpp_type=10, + name='dropout_ratio', + full_name='caffe.V0LayerParameter.dropout_ratio', + index=11, + number=12, + type=2, + cpp_type=6, label=1, - has_default_value=False, - default_value=None, + has_default_value=True, + default_value=float(0.5), message_type=None, enum_type=None, containing_type=None, @@ -11837,15 +12200,15 @@ _V1LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='tanh_param', - full_name='caffe.V1LayerParameter.tanh_param', - index=37, - number=37, - type=11, - cpp_type=10, + name='local_size', + full_name='caffe.V0LayerParameter.local_size', + index=12, + number=13, + type=13, + cpp_type=3, label=1, - has_default_value=False, - default_value=None, + has_default_value=True, + default_value=5, message_type=None, enum_type=None, containing_type=None, @@ -11854,15 +12217,15 @@ _V1LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='threshold_param', - full_name='caffe.V1LayerParameter.threshold_param', - index=38, - number=25, - type=11, - cpp_type=10, + name='alpha', + full_name='caffe.V0LayerParameter.alpha', + index=13, + number=14, + type=2, + cpp_type=6, label=1, - has_default_value=False, - default_value=None, + has_default_value=True, + default_value=float(1), message_type=None, enum_type=None, containing_type=None, @@ -11871,15 +12234,15 @@ _V1LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='window_data_param', - full_name='caffe.V1LayerParameter.window_data_param', - index=39, - number=20, - type=11, - cpp_type=10, + name='beta', + full_name='caffe.V0LayerParameter.beta', + index=14, + number=15, + type=2, + cpp_type=6, label=1, - has_default_value=False, - default_value=None, + has_default_value=True, + default_value=float(0.75), message_type=None, enum_type=None, containing_type=None, @@ -11888,15 +12251,15 @@ _V1LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='transform_param', - full_name='caffe.V1LayerParameter.transform_param', - index=40, - number=36, - type=11, - cpp_type=10, + name='k', + full_name='caffe.V0LayerParameter.k', + index=15, + number=22, + type=2, + cpp_type=6, label=1, - has_default_value=False, - default_value=None, + has_default_value=True, + default_value=float(1), message_type=None, enum_type=None, containing_type=None, @@ -11905,15 +12268,15 @@ _V1LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='loss_param', - full_name='caffe.V1LayerParameter.loss_param', - index=41, - number=42, - type=11, - cpp_type=10, + name='source', + full_name='caffe.V0LayerParameter.source', + index=16, + number=16, + type=9, + cpp_type=9, label=1, has_default_value=False, - default_value=None, + default_value=_b("").decode('utf-8'), message_type=None, enum_type=None, containing_type=None, @@ -11921,87 +12284,16 @@ _V1LAYERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='layer', - full_name='caffe.V1LayerParameter.layer', - index=42, - number=1, - type=11, - cpp_type=10, - label=1, - has_default_value=False, - default_value=None, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - ], - extensions=[], - nested_types=[], - enum_types=[ - _V1LAYERPARAMETER_LAYERTYPE, - _V1LAYERPARAMETER_DIMCHECKMODE, - ], - serialized_options=None, - is_extendable=False, - syntax='proto2', - extension_ranges=[], - oneofs=[], - serialized_start=18220, - serialized_end=20748, -) - -_V0LAYERPARAMETER = _descriptor.Descriptor( - name='V0LayerParameter', - full_name='caffe.V0LayerParameter', - filename=None, - file=DESCRIPTOR, - containing_type=None, - fields=[ - _descriptor.FieldDescriptor(name='name', - full_name='caffe.V0LayerParameter.name', - index=0, - number=1, - type=9, - cpp_type=9, - label=1, - has_default_value=False, - default_value=_b("").decode('utf-8'), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='type', - full_name='caffe.V0LayerParameter.type', - index=1, - number=2, - type=9, - cpp_type=9, - label=1, - has_default_value=False, - default_value=_b("").decode('utf-8'), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='num_output', - full_name='caffe.V0LayerParameter.num_output', - index=2, - number=3, - type=13, - cpp_type=3, + name='scale', + full_name='caffe.V0LayerParameter.scale', + index=17, + number=17, + type=2, + cpp_type=6, label=1, - has_default_value=False, - default_value=0, + has_default_value=True, + default_value=float(1), message_type=None, enum_type=None, containing_type=None, @@ -12009,32 +12301,16 @@ _V0LAYERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='biasterm', - full_name='caffe.V0LayerParameter.biasterm', - index=3, - number=4, - type=8, - cpp_type=7, - label=1, - has_default_value=True, - default_value=True, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='weight_filler', - full_name='caffe.V0LayerParameter.weight_filler', - index=4, - number=5, - type=11, - cpp_type=10, + name='meanfile', + full_name='caffe.V0LayerParameter.meanfile', + index=18, + number=18, + type=9, + cpp_type=9, label=1, has_default_value=False, - default_value=None, + default_value=_b("").decode('utf-8'), message_type=None, enum_type=None, containing_type=None, @@ -12043,15 +12319,15 @@ _V0LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='bias_filler', - full_name='caffe.V0LayerParameter.bias_filler', - index=5, - number=6, - type=11, - cpp_type=10, + name='batchsize', + full_name='caffe.V0LayerParameter.batchsize', + index=19, + number=19, + type=13, + cpp_type=3, label=1, has_default_value=False, - default_value=None, + default_value=0, message_type=None, enum_type=None, containing_type=None, @@ -12059,31 +12335,15 @@ _V0LAYERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='pad', - full_name='caffe.V0LayerParameter.pad', - index=6, - number=7, - type=13, - cpp_type=3, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='kernelsize', - full_name='caffe.V0LayerParameter.kernelsize', - index=7, - number=8, + name='cropsize', + full_name='caffe.V0LayerParameter.cropsize', + index=20, + number=20, type=13, cpp_type=3, label=1, - has_default_value=False, + has_default_value=True, default_value=0, message_type=None, enum_type=None, @@ -12092,64 +12352,16 @@ _V0LAYERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='group', - full_name='caffe.V0LayerParameter.group', - index=8, - number=9, - type=13, - cpp_type=3, - label=1, - has_default_value=True, - default_value=1, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='stride', - full_name='caffe.V0LayerParameter.stride', - index=9, - number=10, - type=13, - cpp_type=3, - label=1, - has_default_value=True, - default_value=1, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='pool', - full_name='caffe.V0LayerParameter.pool', - index=10, - number=11, - type=14, - cpp_type=8, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='dropout_ratio', - full_name='caffe.V0LayerParameter.dropout_ratio', - index=11, - number=12, - type=2, - cpp_type=6, + name='mirror', + full_name='caffe.V0LayerParameter.mirror', + index=21, + number=21, + type=8, + cpp_type=7, label=1, has_default_value=True, - default_value=float(0.5), + default_value=False, message_type=None, enum_type=None, containing_type=None, @@ -12158,15 +12370,15 @@ _V0LAYERPARAMETER = _descriptor.Descriptor( serialized_options=None, file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='local_size', - full_name='caffe.V0LayerParameter.local_size', - index=12, - number=13, - type=13, - cpp_type=3, - label=1, - has_default_value=True, - default_value=5, + name='blobs', + full_name='caffe.V0LayerParameter.blobs', + index=22, + number=50, + type=11, + cpp_type=10, + label=3, + has_default_value=False, + default_value=[], message_type=None, enum_type=None, containing_type=None, @@ -12174,112 +12386,16 @@ _V0LAYERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='alpha', - full_name='caffe.V0LayerParameter.alpha', - index=13, - number=14, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(1), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='beta', - full_name='caffe.V0LayerParameter.beta', - index=14, - number=15, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(0.75), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='k', - full_name='caffe.V0LayerParameter.k', - index=15, - number=22, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(1), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='source', - full_name='caffe.V0LayerParameter.source', - index=16, - number=16, - type=9, - cpp_type=9, - label=1, - has_default_value=False, - default_value=_b("").decode('utf-8'), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='scale', - full_name='caffe.V0LayerParameter.scale', - index=17, - number=17, - type=2, - cpp_type=6, - label=1, - has_default_value=True, - default_value=float(1), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='meanfile', - full_name='caffe.V0LayerParameter.meanfile', - index=18, - number=18, - type=9, - cpp_type=9, - label=1, - has_default_value=False, - default_value=_b("").decode('utf-8'), - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( - name='batchsize', - full_name='caffe.V0LayerParameter.batchsize', - index=19, - number=19, - type=13, - cpp_type=3, - label=1, + name='blobs_lr', + full_name='caffe.V0LayerParameter.blobs_lr', + index=23, + number=51, + type=2, + cpp_type=6, + label=3, has_default_value=False, - default_value=0, + default_value=[], message_type=None, enum_type=None, containing_type=None, @@ -12287,70 +12403,6 @@ _V0LAYERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='cropsize', - full_name='caffe.V0LayerParameter.cropsize', - index=20, - number=20, - type=13, - cpp_type=3, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='mirror', - full_name='caffe.V0LayerParameter.mirror', - index=21, - number=21, - type=8, - cpp_type=7, - label=1, - has_default_value=True, - default_value=False, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='blobs', - full_name='caffe.V0LayerParameter.blobs', - index=22, - number=50, - type=11, - cpp_type=10, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='blobs_lr', - full_name='caffe.V0LayerParameter.blobs_lr', - index=23, - number=51, - type=2, - cpp_type=6, - label=3, - has_default_value=False, - default_value=[], - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), _descriptor.FieldDescriptor( name='weight_decay', full_name='caffe.V0LayerParameter.weight_decay', @@ -12470,22 +12522,23 @@ _V0LAYERPARAMETER = _descriptor.Descriptor( extension_scope=None, serialized_options=None, file=DESCRIPTOR), - _descriptor.FieldDescriptor(name='new_num', - full_name='caffe.V0LayerParameter.new_num', - index=31, - number=60, - type=5, - cpp_type=1, - label=1, - has_default_value=True, - default_value=0, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='new_num', + full_name='caffe.V0LayerParameter.new_num', + index=31, + number=60, + type=5, + cpp_type=1, + label=1, + has_default_value=True, + default_value=0, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), _descriptor.FieldDescriptor( name='new_channels', full_name='caffe.V0LayerParameter.new_channels', @@ -12591,17 +12644,14 @@ _V0LAYERPARAMETER = _descriptor.Descriptor( ], extensions=[], nested_types=[], - enum_types=[ - _V0LAYERPARAMETER_POOLMETHOD, - ], + enum_types=[_V0LAYERPARAMETER_POOLMETHOD, ], serialized_options=None, is_extendable=False, syntax='proto2', extension_ranges=[], oneofs=[], serialized_start=20751, - serialized_end=21772, -) + serialized_end=21772, ) _PRELUPARAMETER = _descriptor.Descriptor( name='PReLUParameter', @@ -12610,22 +12660,23 @@ _PRELUPARAMETER = _descriptor.Descriptor( file=DESCRIPTOR, containing_type=None, fields=[ - _descriptor.FieldDescriptor(name='filler', - full_name='caffe.PReLUParameter.filler', - index=0, - number=1, - type=11, - cpp_type=10, - label=1, - has_default_value=False, - default_value=None, - message_type=None, - enum_type=None, - containing_type=None, - is_extension=False, - extension_scope=None, - serialized_options=None, - file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='filler', + full_name='caffe.PReLUParameter.filler', + index=0, + number=1, + type=11, + cpp_type=10, + label=1, + has_default_value=False, + default_value=None, + message_type=None, + enum_type=None, + containing_type=None, + is_extension=False, + extension_scope=None, + serialized_options=None, + file=DESCRIPTOR), _descriptor.FieldDescriptor( name='channel_shared', full_name='caffe.PReLUParameter.channel_shared', @@ -12653,8 +12704,7 @@ _PRELUPARAMETER = _descriptor.Descriptor( extension_ranges=[], oneofs=[], serialized_start=21774, - serialized_end=21861, -) + serialized_end=21861, ) _SHUFFLECHANNELPARAMETER = _descriptor.Descriptor( name='ShuffleChannelParameter', @@ -12690,8 +12740,7 @@ _SHUFFLECHANNELPARAMETER = _descriptor.Descriptor( extension_ranges=[], oneofs=[], serialized_start=21863, - serialized_end=21906, -) + serialized_end=21906, ) _BLOBPROTO.fields_by_name['shape'].message_type = _BLOBSHAPE _BLOBPROTOVECTOR.fields_by_name['blobs'].message_type = _BLOBPROTO @@ -13109,352 +13158,391 @@ _sym_db.RegisterFileDescriptor(DESCRIPTOR) BlobShape = _reflection.GeneratedProtocolMessageType( 'BlobShape', (_message.Message, ), - dict(DESCRIPTOR=_BLOBSHAPE, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.BlobShape) - )) + dict( + DESCRIPTOR=_BLOBSHAPE, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.BlobShape) + )) _sym_db.RegisterMessage(BlobShape) BlobProto = _reflection.GeneratedProtocolMessageType( 'BlobProto', (_message.Message, ), - dict(DESCRIPTOR=_BLOBPROTO, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.BlobProto) - )) + dict( + DESCRIPTOR=_BLOBPROTO, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.BlobProto) + )) _sym_db.RegisterMessage(BlobProto) BlobProtoVector = _reflection.GeneratedProtocolMessageType( 'BlobProtoVector', (_message.Message, ), - dict(DESCRIPTOR=_BLOBPROTOVECTOR, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.BlobProtoVector) - )) + dict( + DESCRIPTOR=_BLOBPROTOVECTOR, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.BlobProtoVector) + )) _sym_db.RegisterMessage(BlobProtoVector) Datum = _reflection.GeneratedProtocolMessageType( 'Datum', (_message.Message, ), - dict(DESCRIPTOR=_DATUM, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.Datum) - )) + dict( + DESCRIPTOR=_DATUM, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.Datum) + )) _sym_db.RegisterMessage(Datum) LabelMapItem = _reflection.GeneratedProtocolMessageType( 'LabelMapItem', (_message.Message, ), - dict(DESCRIPTOR=_LABELMAPITEM, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.LabelMapItem) - )) + dict( + DESCRIPTOR=_LABELMAPITEM, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.LabelMapItem) + )) _sym_db.RegisterMessage(LabelMapItem) LabelMap = _reflection.GeneratedProtocolMessageType( 'LabelMap', (_message.Message, ), - dict(DESCRIPTOR=_LABELMAP, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.LabelMap) - )) + dict( + DESCRIPTOR=_LABELMAP, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.LabelMap) + )) _sym_db.RegisterMessage(LabelMap) Sampler = _reflection.GeneratedProtocolMessageType( 'Sampler', (_message.Message, ), - dict(DESCRIPTOR=_SAMPLER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.Sampler) - )) + dict( + DESCRIPTOR=_SAMPLER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.Sampler) + )) _sym_db.RegisterMessage(Sampler) SampleConstraint = _reflection.GeneratedProtocolMessageType( 'SampleConstraint', (_message.Message, ), - dict(DESCRIPTOR=_SAMPLECONSTRAINT, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.SampleConstraint) - )) + dict( + DESCRIPTOR=_SAMPLECONSTRAINT, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.SampleConstraint) + )) _sym_db.RegisterMessage(SampleConstraint) BatchSampler = _reflection.GeneratedProtocolMessageType( 'BatchSampler', (_message.Message, ), - dict(DESCRIPTOR=_BATCHSAMPLER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.BatchSampler) - )) + dict( + DESCRIPTOR=_BATCHSAMPLER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.BatchSampler) + )) _sym_db.RegisterMessage(BatchSampler) EmitConstraint = _reflection.GeneratedProtocolMessageType( 'EmitConstraint', (_message.Message, ), - dict(DESCRIPTOR=_EMITCONSTRAINT, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.EmitConstraint) - )) + dict( + DESCRIPTOR=_EMITCONSTRAINT, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.EmitConstraint) + )) _sym_db.RegisterMessage(EmitConstraint) NormalizedBBox = _reflection.GeneratedProtocolMessageType( 'NormalizedBBox', (_message.Message, ), - dict(DESCRIPTOR=_NORMALIZEDBBOX, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.NormalizedBBox) - )) + dict( + DESCRIPTOR=_NORMALIZEDBBOX, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.NormalizedBBox) + )) _sym_db.RegisterMessage(NormalizedBBox) Annotation = _reflection.GeneratedProtocolMessageType( 'Annotation', (_message.Message, ), - dict(DESCRIPTOR=_ANNOTATION, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.Annotation) - )) + dict( + DESCRIPTOR=_ANNOTATION, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.Annotation) + )) _sym_db.RegisterMessage(Annotation) AnnotationGroup = _reflection.GeneratedProtocolMessageType( 'AnnotationGroup', (_message.Message, ), - dict(DESCRIPTOR=_ANNOTATIONGROUP, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.AnnotationGroup) - )) + dict( + DESCRIPTOR=_ANNOTATIONGROUP, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.AnnotationGroup) + )) _sym_db.RegisterMessage(AnnotationGroup) AnnotatedDatum = _reflection.GeneratedProtocolMessageType( 'AnnotatedDatum', (_message.Message, ), - dict(DESCRIPTOR=_ANNOTATEDDATUM, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.AnnotatedDatum) - )) + dict( + DESCRIPTOR=_ANNOTATEDDATUM, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.AnnotatedDatum) + )) _sym_db.RegisterMessage(AnnotatedDatum) FillerParameter = _reflection.GeneratedProtocolMessageType( 'FillerParameter', (_message.Message, ), - dict(DESCRIPTOR=_FILLERPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.FillerParameter) - )) + dict( + DESCRIPTOR=_FILLERPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.FillerParameter) + )) _sym_db.RegisterMessage(FillerParameter) NetParameter = _reflection.GeneratedProtocolMessageType( 'NetParameter', (_message.Message, ), - dict(DESCRIPTOR=_NETPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.NetParameter) - )) + dict( + DESCRIPTOR=_NETPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.NetParameter) + )) _sym_db.RegisterMessage(NetParameter) SolverParameter = _reflection.GeneratedProtocolMessageType( 'SolverParameter', (_message.Message, ), - dict(DESCRIPTOR=_SOLVERPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.SolverParameter) - )) + dict( + DESCRIPTOR=_SOLVERPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.SolverParameter) + )) _sym_db.RegisterMessage(SolverParameter) SolverState = _reflection.GeneratedProtocolMessageType( 'SolverState', (_message.Message, ), - dict(DESCRIPTOR=_SOLVERSTATE, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.SolverState) - )) + dict( + DESCRIPTOR=_SOLVERSTATE, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.SolverState) + )) _sym_db.RegisterMessage(SolverState) NetState = _reflection.GeneratedProtocolMessageType( 'NetState', (_message.Message, ), - dict(DESCRIPTOR=_NETSTATE, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.NetState) - )) + dict( + DESCRIPTOR=_NETSTATE, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.NetState) + )) _sym_db.RegisterMessage(NetState) NetStateRule = _reflection.GeneratedProtocolMessageType( 'NetStateRule', (_message.Message, ), - dict(DESCRIPTOR=_NETSTATERULE, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.NetStateRule) - )) + dict( + DESCRIPTOR=_NETSTATERULE, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.NetStateRule) + )) _sym_db.RegisterMessage(NetStateRule) ParamSpec = _reflection.GeneratedProtocolMessageType( 'ParamSpec', (_message.Message, ), - dict(DESCRIPTOR=_PARAMSPEC, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.ParamSpec) - )) + dict( + DESCRIPTOR=_PARAMSPEC, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.ParamSpec) + )) _sym_db.RegisterMessage(ParamSpec) LayerParameter = _reflection.GeneratedProtocolMessageType( 'LayerParameter', (_message.Message, ), - dict(DESCRIPTOR=_LAYERPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.LayerParameter) - )) + dict( + DESCRIPTOR=_LAYERPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.LayerParameter) + )) _sym_db.RegisterMessage(LayerParameter) TransformationParameter = _reflection.GeneratedProtocolMessageType( 'TransformationParameter', (_message.Message, ), - dict(DESCRIPTOR=_TRANSFORMATIONPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.TransformationParameter) - )) + dict( + DESCRIPTOR=_TRANSFORMATIONPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.TransformationParameter) + )) _sym_db.RegisterMessage(TransformationParameter) ResizeParameter = _reflection.GeneratedProtocolMessageType( 'ResizeParameter', (_message.Message, ), - dict(DESCRIPTOR=_RESIZEPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.ResizeParameter) - )) + dict( + DESCRIPTOR=_RESIZEPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.ResizeParameter) + )) _sym_db.RegisterMessage(ResizeParameter) SaltPepperParameter = _reflection.GeneratedProtocolMessageType( 'SaltPepperParameter', (_message.Message, ), - dict(DESCRIPTOR=_SALTPEPPERPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.SaltPepperParameter) - )) + dict( + DESCRIPTOR=_SALTPEPPERPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.SaltPepperParameter) + )) _sym_db.RegisterMessage(SaltPepperParameter) NoiseParameter = _reflection.GeneratedProtocolMessageType( 'NoiseParameter', (_message.Message, ), - dict(DESCRIPTOR=_NOISEPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.NoiseParameter) - )) + dict( + DESCRIPTOR=_NOISEPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.NoiseParameter) + )) _sym_db.RegisterMessage(NoiseParameter) DistortionParameter = _reflection.GeneratedProtocolMessageType( 'DistortionParameter', (_message.Message, ), - dict(DESCRIPTOR=_DISTORTIONPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.DistortionParameter) - )) + dict( + DESCRIPTOR=_DISTORTIONPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.DistortionParameter) + )) _sym_db.RegisterMessage(DistortionParameter) ExpansionParameter = _reflection.GeneratedProtocolMessageType( 'ExpansionParameter', (_message.Message, ), - dict(DESCRIPTOR=_EXPANSIONPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.ExpansionParameter) - )) + dict( + DESCRIPTOR=_EXPANSIONPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.ExpansionParameter) + )) _sym_db.RegisterMessage(ExpansionParameter) LossParameter = _reflection.GeneratedProtocolMessageType( 'LossParameter', (_message.Message, ), - dict(DESCRIPTOR=_LOSSPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.LossParameter) - )) + dict( + DESCRIPTOR=_LOSSPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.LossParameter) + )) _sym_db.RegisterMessage(LossParameter) AccuracyParameter = _reflection.GeneratedProtocolMessageType( 'AccuracyParameter', (_message.Message, ), - dict(DESCRIPTOR=_ACCURACYPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.AccuracyParameter) - )) + dict( + DESCRIPTOR=_ACCURACYPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.AccuracyParameter) + )) _sym_db.RegisterMessage(AccuracyParameter) AnnotatedDataParameter = _reflection.GeneratedProtocolMessageType( 'AnnotatedDataParameter', (_message.Message, ), - dict(DESCRIPTOR=_ANNOTATEDDATAPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.AnnotatedDataParameter) - )) + dict( + DESCRIPTOR=_ANNOTATEDDATAPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.AnnotatedDataParameter) + )) _sym_db.RegisterMessage(AnnotatedDataParameter) ArgMaxParameter = _reflection.GeneratedProtocolMessageType( 'ArgMaxParameter', (_message.Message, ), - dict(DESCRIPTOR=_ARGMAXPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.ArgMaxParameter) - )) + dict( + DESCRIPTOR=_ARGMAXPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.ArgMaxParameter) + )) _sym_db.RegisterMessage(ArgMaxParameter) ConcatParameter = _reflection.GeneratedProtocolMessageType( 'ConcatParameter', (_message.Message, ), - dict(DESCRIPTOR=_CONCATPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.ConcatParameter) - )) + dict( + DESCRIPTOR=_CONCATPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.ConcatParameter) + )) _sym_db.RegisterMessage(ConcatParameter) BatchNormParameter = _reflection.GeneratedProtocolMessageType( 'BatchNormParameter', (_message.Message, ), - dict(DESCRIPTOR=_BATCHNORMPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.BatchNormParameter) - )) + dict( + DESCRIPTOR=_BATCHNORMPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.BatchNormParameter) + )) _sym_db.RegisterMessage(BatchNormParameter) BiasParameter = _reflection.GeneratedProtocolMessageType( 'BiasParameter', (_message.Message, ), - dict(DESCRIPTOR=_BIASPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.BiasParameter) - )) + dict( + DESCRIPTOR=_BIASPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.BiasParameter) + )) _sym_db.RegisterMessage(BiasParameter) ContrastiveLossParameter = _reflection.GeneratedProtocolMessageType( 'ContrastiveLossParameter', (_message.Message, ), - dict(DESCRIPTOR=_CONTRASTIVELOSSPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.ContrastiveLossParameter) - )) + dict( + DESCRIPTOR=_CONTRASTIVELOSSPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.ContrastiveLossParameter) + )) _sym_db.RegisterMessage(ContrastiveLossParameter) ConvolutionParameter = _reflection.GeneratedProtocolMessageType( 'ConvolutionParameter', (_message.Message, ), - dict(DESCRIPTOR=_CONVOLUTIONPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.ConvolutionParameter) - )) + dict( + DESCRIPTOR=_CONVOLUTIONPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.ConvolutionParameter) + )) _sym_db.RegisterMessage(ConvolutionParameter) CropParameter = _reflection.GeneratedProtocolMessageType( 'CropParameter', (_message.Message, ), - dict(DESCRIPTOR=_CROPPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.CropParameter) - )) + dict( + DESCRIPTOR=_CROPPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.CropParameter) + )) _sym_db.RegisterMessage(CropParameter) DataParameter = _reflection.GeneratedProtocolMessageType( 'DataParameter', (_message.Message, ), - dict(DESCRIPTOR=_DATAPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.DataParameter) - )) + dict( + DESCRIPTOR=_DATAPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.DataParameter) + )) _sym_db.RegisterMessage(DataParameter) DetectionEvaluateParameter = _reflection.GeneratedProtocolMessageType( @@ -13480,415 +13568,461 @@ _sym_db.RegisterMessage(NonMaximumSuppressionParameter) SaveOutputParameter = _reflection.GeneratedProtocolMessageType( 'SaveOutputParameter', (_message.Message, ), - dict(DESCRIPTOR=_SAVEOUTPUTPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.SaveOutputParameter) - )) + dict( + DESCRIPTOR=_SAVEOUTPUTPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.SaveOutputParameter) + )) _sym_db.RegisterMessage(SaveOutputParameter) DetectionOutputParameter = _reflection.GeneratedProtocolMessageType( 'DetectionOutputParameter', (_message.Message, ), - dict(DESCRIPTOR=_DETECTIONOUTPUTPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.DetectionOutputParameter) - )) + dict( + DESCRIPTOR=_DETECTIONOUTPUTPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.DetectionOutputParameter) + )) _sym_db.RegisterMessage(DetectionOutputParameter) DropoutParameter = _reflection.GeneratedProtocolMessageType( 'DropoutParameter', (_message.Message, ), - dict(DESCRIPTOR=_DROPOUTPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.DropoutParameter) - )) + dict( + DESCRIPTOR=_DROPOUTPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.DropoutParameter) + )) _sym_db.RegisterMessage(DropoutParameter) DummyDataParameter = _reflection.GeneratedProtocolMessageType( 'DummyDataParameter', (_message.Message, ), - dict(DESCRIPTOR=_DUMMYDATAPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.DummyDataParameter) - )) + dict( + DESCRIPTOR=_DUMMYDATAPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.DummyDataParameter) + )) _sym_db.RegisterMessage(DummyDataParameter) EltwiseParameter = _reflection.GeneratedProtocolMessageType( 'EltwiseParameter', (_message.Message, ), - dict(DESCRIPTOR=_ELTWISEPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.EltwiseParameter) - )) + dict( + DESCRIPTOR=_ELTWISEPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.EltwiseParameter) + )) _sym_db.RegisterMessage(EltwiseParameter) ELUParameter = _reflection.GeneratedProtocolMessageType( 'ELUParameter', (_message.Message, ), - dict(DESCRIPTOR=_ELUPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.ELUParameter) - )) + dict( + DESCRIPTOR=_ELUPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.ELUParameter) + )) _sym_db.RegisterMessage(ELUParameter) EmbedParameter = _reflection.GeneratedProtocolMessageType( 'EmbedParameter', (_message.Message, ), - dict(DESCRIPTOR=_EMBEDPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.EmbedParameter) - )) + dict( + DESCRIPTOR=_EMBEDPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.EmbedParameter) + )) _sym_db.RegisterMessage(EmbedParameter) ExpParameter = _reflection.GeneratedProtocolMessageType( 'ExpParameter', (_message.Message, ), - dict(DESCRIPTOR=_EXPPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.ExpParameter) - )) + dict( + DESCRIPTOR=_EXPPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.ExpParameter) + )) _sym_db.RegisterMessage(ExpParameter) FlattenParameter = _reflection.GeneratedProtocolMessageType( 'FlattenParameter', (_message.Message, ), - dict(DESCRIPTOR=_FLATTENPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.FlattenParameter) - )) + dict( + DESCRIPTOR=_FLATTENPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.FlattenParameter) + )) _sym_db.RegisterMessage(FlattenParameter) HDF5DataParameter = _reflection.GeneratedProtocolMessageType( 'HDF5DataParameter', (_message.Message, ), - dict(DESCRIPTOR=_HDF5DATAPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.HDF5DataParameter) - )) + dict( + DESCRIPTOR=_HDF5DATAPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.HDF5DataParameter) + )) _sym_db.RegisterMessage(HDF5DataParameter) HDF5OutputParameter = _reflection.GeneratedProtocolMessageType( 'HDF5OutputParameter', (_message.Message, ), - dict(DESCRIPTOR=_HDF5OUTPUTPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.HDF5OutputParameter) - )) + dict( + DESCRIPTOR=_HDF5OUTPUTPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.HDF5OutputParameter) + )) _sym_db.RegisterMessage(HDF5OutputParameter) HingeLossParameter = _reflection.GeneratedProtocolMessageType( 'HingeLossParameter', (_message.Message, ), - dict(DESCRIPTOR=_HINGELOSSPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.HingeLossParameter) - )) + dict( + DESCRIPTOR=_HINGELOSSPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.HingeLossParameter) + )) _sym_db.RegisterMessage(HingeLossParameter) ImageDataParameter = _reflection.GeneratedProtocolMessageType( 'ImageDataParameter', (_message.Message, ), - dict(DESCRIPTOR=_IMAGEDATAPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.ImageDataParameter) - )) + dict( + DESCRIPTOR=_IMAGEDATAPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.ImageDataParameter) + )) _sym_db.RegisterMessage(ImageDataParameter) InfogainLossParameter = _reflection.GeneratedProtocolMessageType( 'InfogainLossParameter', (_message.Message, ), - dict(DESCRIPTOR=_INFOGAINLOSSPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.InfogainLossParameter) - )) + dict( + DESCRIPTOR=_INFOGAINLOSSPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.InfogainLossParameter) + )) _sym_db.RegisterMessage(InfogainLossParameter) InnerProductParameter = _reflection.GeneratedProtocolMessageType( 'InnerProductParameter', (_message.Message, ), - dict(DESCRIPTOR=_INNERPRODUCTPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.InnerProductParameter) - )) + dict( + DESCRIPTOR=_INNERPRODUCTPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.InnerProductParameter) + )) _sym_db.RegisterMessage(InnerProductParameter) InputParameter = _reflection.GeneratedProtocolMessageType( 'InputParameter', (_message.Message, ), - dict(DESCRIPTOR=_INPUTPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.InputParameter) - )) + dict( + DESCRIPTOR=_INPUTPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.InputParameter) + )) _sym_db.RegisterMessage(InputParameter) LogParameter = _reflection.GeneratedProtocolMessageType( 'LogParameter', (_message.Message, ), - dict(DESCRIPTOR=_LOGPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.LogParameter) - )) + dict( + DESCRIPTOR=_LOGPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.LogParameter) + )) _sym_db.RegisterMessage(LogParameter) LRNParameter = _reflection.GeneratedProtocolMessageType( 'LRNParameter', (_message.Message, ), - dict(DESCRIPTOR=_LRNPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.LRNParameter) - )) + dict( + DESCRIPTOR=_LRNPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.LRNParameter) + )) _sym_db.RegisterMessage(LRNParameter) MemoryDataParameter = _reflection.GeneratedProtocolMessageType( 'MemoryDataParameter', (_message.Message, ), - dict(DESCRIPTOR=_MEMORYDATAPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.MemoryDataParameter) - )) + dict( + DESCRIPTOR=_MEMORYDATAPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.MemoryDataParameter) + )) _sym_db.RegisterMessage(MemoryDataParameter) MultiBoxLossParameter = _reflection.GeneratedProtocolMessageType( 'MultiBoxLossParameter', (_message.Message, ), - dict(DESCRIPTOR=_MULTIBOXLOSSPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.MultiBoxLossParameter) - )) + dict( + DESCRIPTOR=_MULTIBOXLOSSPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.MultiBoxLossParameter) + )) _sym_db.RegisterMessage(MultiBoxLossParameter) MVNParameter = _reflection.GeneratedProtocolMessageType( 'MVNParameter', (_message.Message, ), - dict(DESCRIPTOR=_MVNPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.MVNParameter) - )) + dict( + DESCRIPTOR=_MVNPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.MVNParameter) + )) _sym_db.RegisterMessage(MVNParameter) NormalizeParameter = _reflection.GeneratedProtocolMessageType( 'NormalizeParameter', (_message.Message, ), - dict(DESCRIPTOR=_NORMALIZEPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.NormalizeParameter) - )) + dict( + DESCRIPTOR=_NORMALIZEPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.NormalizeParameter) + )) _sym_db.RegisterMessage(NormalizeParameter) ParameterParameter = _reflection.GeneratedProtocolMessageType( 'ParameterParameter', (_message.Message, ), - dict(DESCRIPTOR=_PARAMETERPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.ParameterParameter) - )) + dict( + DESCRIPTOR=_PARAMETERPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.ParameterParameter) + )) _sym_db.RegisterMessage(ParameterParameter) PermuteParameter = _reflection.GeneratedProtocolMessageType( 'PermuteParameter', (_message.Message, ), - dict(DESCRIPTOR=_PERMUTEPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.PermuteParameter) - )) + dict( + DESCRIPTOR=_PERMUTEPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.PermuteParameter) + )) _sym_db.RegisterMessage(PermuteParameter) PoolingParameter = _reflection.GeneratedProtocolMessageType( 'PoolingParameter', (_message.Message, ), - dict(DESCRIPTOR=_POOLINGPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.PoolingParameter) - )) + dict( + DESCRIPTOR=_POOLINGPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.PoolingParameter) + )) _sym_db.RegisterMessage(PoolingParameter) PowerParameter = _reflection.GeneratedProtocolMessageType( 'PowerParameter', (_message.Message, ), - dict(DESCRIPTOR=_POWERPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.PowerParameter) - )) + dict( + DESCRIPTOR=_POWERPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.PowerParameter) + )) _sym_db.RegisterMessage(PowerParameter) PriorBoxParameter = _reflection.GeneratedProtocolMessageType( 'PriorBoxParameter', (_message.Message, ), - dict(DESCRIPTOR=_PRIORBOXPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.PriorBoxParameter) - )) + dict( + DESCRIPTOR=_PRIORBOXPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.PriorBoxParameter) + )) _sym_db.RegisterMessage(PriorBoxParameter) PythonParameter = _reflection.GeneratedProtocolMessageType( 'PythonParameter', (_message.Message, ), - dict(DESCRIPTOR=_PYTHONPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.PythonParameter) - )) + dict( + DESCRIPTOR=_PYTHONPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.PythonParameter) + )) _sym_db.RegisterMessage(PythonParameter) RecurrentParameter = _reflection.GeneratedProtocolMessageType( 'RecurrentParameter', (_message.Message, ), - dict(DESCRIPTOR=_RECURRENTPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.RecurrentParameter) - )) + dict( + DESCRIPTOR=_RECURRENTPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.RecurrentParameter) + )) _sym_db.RegisterMessage(RecurrentParameter) ReductionParameter = _reflection.GeneratedProtocolMessageType( 'ReductionParameter', (_message.Message, ), - dict(DESCRIPTOR=_REDUCTIONPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.ReductionParameter) - )) + dict( + DESCRIPTOR=_REDUCTIONPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.ReductionParameter) + )) _sym_db.RegisterMessage(ReductionParameter) ReLUParameter = _reflection.GeneratedProtocolMessageType( 'ReLUParameter', (_message.Message, ), - dict(DESCRIPTOR=_RELUPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.ReLUParameter) - )) + dict( + DESCRIPTOR=_RELUPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.ReLUParameter) + )) _sym_db.RegisterMessage(ReLUParameter) ReshapeParameter = _reflection.GeneratedProtocolMessageType( 'ReshapeParameter', (_message.Message, ), - dict(DESCRIPTOR=_RESHAPEPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.ReshapeParameter) - )) + dict( + DESCRIPTOR=_RESHAPEPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.ReshapeParameter) + )) _sym_db.RegisterMessage(ReshapeParameter) ScaleParameter = _reflection.GeneratedProtocolMessageType( 'ScaleParameter', (_message.Message, ), - dict(DESCRIPTOR=_SCALEPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.ScaleParameter) - )) + dict( + DESCRIPTOR=_SCALEPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.ScaleParameter) + )) _sym_db.RegisterMessage(ScaleParameter) SigmoidParameter = _reflection.GeneratedProtocolMessageType( 'SigmoidParameter', (_message.Message, ), - dict(DESCRIPTOR=_SIGMOIDPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.SigmoidParameter) - )) + dict( + DESCRIPTOR=_SIGMOIDPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.SigmoidParameter) + )) _sym_db.RegisterMessage(SigmoidParameter) SliceParameter = _reflection.GeneratedProtocolMessageType( 'SliceParameter', (_message.Message, ), - dict(DESCRIPTOR=_SLICEPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.SliceParameter) - )) + dict( + DESCRIPTOR=_SLICEPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.SliceParameter) + )) _sym_db.RegisterMessage(SliceParameter) SoftmaxParameter = _reflection.GeneratedProtocolMessageType( 'SoftmaxParameter', (_message.Message, ), - dict(DESCRIPTOR=_SOFTMAXPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.SoftmaxParameter) - )) + dict( + DESCRIPTOR=_SOFTMAXPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.SoftmaxParameter) + )) _sym_db.RegisterMessage(SoftmaxParameter) TanHParameter = _reflection.GeneratedProtocolMessageType( 'TanHParameter', (_message.Message, ), - dict(DESCRIPTOR=_TANHPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.TanHParameter) - )) + dict( + DESCRIPTOR=_TANHPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.TanHParameter) + )) _sym_db.RegisterMessage(TanHParameter) TileParameter = _reflection.GeneratedProtocolMessageType( 'TileParameter', (_message.Message, ), - dict(DESCRIPTOR=_TILEPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.TileParameter) - )) + dict( + DESCRIPTOR=_TILEPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.TileParameter) + )) _sym_db.RegisterMessage(TileParameter) ThresholdParameter = _reflection.GeneratedProtocolMessageType( 'ThresholdParameter', (_message.Message, ), - dict(DESCRIPTOR=_THRESHOLDPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.ThresholdParameter) - )) + dict( + DESCRIPTOR=_THRESHOLDPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.ThresholdParameter) + )) _sym_db.RegisterMessage(ThresholdParameter) VideoDataParameter = _reflection.GeneratedProtocolMessageType( 'VideoDataParameter', (_message.Message, ), - dict(DESCRIPTOR=_VIDEODATAPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.VideoDataParameter) - )) + dict( + DESCRIPTOR=_VIDEODATAPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.VideoDataParameter) + )) _sym_db.RegisterMessage(VideoDataParameter) WindowDataParameter = _reflection.GeneratedProtocolMessageType( 'WindowDataParameter', (_message.Message, ), - dict(DESCRIPTOR=_WINDOWDATAPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.WindowDataParameter) - )) + dict( + DESCRIPTOR=_WINDOWDATAPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.WindowDataParameter) + )) _sym_db.RegisterMessage(WindowDataParameter) SPPParameter = _reflection.GeneratedProtocolMessageType( 'SPPParameter', (_message.Message, ), - dict(DESCRIPTOR=_SPPPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.SPPParameter) - )) + dict( + DESCRIPTOR=_SPPPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.SPPParameter) + )) _sym_db.RegisterMessage(SPPParameter) V1LayerParameter = _reflection.GeneratedProtocolMessageType( 'V1LayerParameter', (_message.Message, ), - dict(DESCRIPTOR=_V1LAYERPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.V1LayerParameter) - )) + dict( + DESCRIPTOR=_V1LAYERPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.V1LayerParameter) + )) _sym_db.RegisterMessage(V1LayerParameter) V0LayerParameter = _reflection.GeneratedProtocolMessageType( 'V0LayerParameter', (_message.Message, ), - dict(DESCRIPTOR=_V0LAYERPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.V0LayerParameter) - )) + dict( + DESCRIPTOR=_V0LAYERPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.V0LayerParameter) + )) _sym_db.RegisterMessage(V0LayerParameter) PReLUParameter = _reflection.GeneratedProtocolMessageType( 'PReLUParameter', (_message.Message, ), - dict(DESCRIPTOR=_PRELUPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.PReLUParameter) - )) + dict( + DESCRIPTOR=_PRELUPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.PReLUParameter) + )) _sym_db.RegisterMessage(PReLUParameter) ShuffleChannelParameter = _reflection.GeneratedProtocolMessageType( 'ShuffleChannelParameter', (_message.Message, ), - dict(DESCRIPTOR=_SHUFFLECHANNELPARAMETER, - __module__='caffe_pb2' - # @@protoc_insertion_point(class_scope:caffe.ShuffleChannelParameter) - )) + dict( + DESCRIPTOR=_SHUFFLECHANNELPARAMETER, + __module__='caffe_pb2' + # @@protoc_insertion_point(class_scope:caffe.ShuffleChannelParameter) + )) _sym_db.RegisterMessage(ShuffleChannelParameter) _BLOBSHAPE.fields_by_name['dim']._options = None diff --git a/x2paddle/decoder/onnx_decoder.py b/x2paddle/decoder/onnx_decoder.py index 0a385003eb63d45ce43a3b6fd36f12f2f240cdd7..922c6629582ab62843359c8f9219f9e83cb2aa13 100644 --- a/x2paddle/decoder/onnx_decoder.py +++ b/x2paddle/decoder/onnx_decoder.py @@ -71,9 +71,8 @@ class ONNXGraphNode(GraphNode): if attr.type == onnx.AttributeProto.TENSOR: dtype = np.dtype(TENSOR_TYPE_TO_NP_TYPE[attr.t.data_type]) data = attr.t.raw_data - value = np.frombuffer(data, - dtype=dtype, - count=(len(data) // dtype.itemsize)) + value = np.frombuffer( + data, dtype=dtype, count=(len(data) // dtype.itemsize)) elif attr.type == onnx.AttributeProto.STRING: value = attr.s value = value.decode() if isinstance(value, bytes) else value @@ -205,9 +204,8 @@ class ONNXGraph(Graph): self.node_map[name].weight = weight self.node_map[name].embeded_as = [] else: - self.node_map[name] = ONNXGraphDataNode(initializer, - layer_name=name, - is_global_input=False) + self.node_map[name] = ONNXGraphDataNode( + initializer, layer_name=name, is_global_input=False) self.node_map[name].weight = weight self.node_map[name].embeded_as = [] @@ -494,8 +492,8 @@ class ONNXDecoder(object): sess = rt.InferenceSession(model_path) for ipt in sess.get_inputs(): datatype = datatype_map[ipt.type] - input_dict[ipt.name] = np.random.random( - ipt.shape).astype(datatype) + input_dict[ipt.name] = np.random.random(ipt.shape).astype( + datatype) res = sess.run(None, input_feed=input_dict) except: diff --git a/x2paddle/decoder/tf_decoder.py b/x2paddle/decoder/tf_decoder.py index 089be82db65747aefd108fec40de083f89aaa27c..dcc534241a06b1d70e462bfe16d3a7da5b82b2fd 100644 --- a/x2paddle/decoder/tf_decoder.py +++ b/x2paddle/decoder/tf_decoder.py @@ -120,13 +120,13 @@ class TFGraph(Graph): def build(self): for layer in self.model.node: self.node_map[layer.name.replace('/', '_').replace( - '-', '_')] = TFGraphNode(layer, data_format=self.tf_data_format) + '-', '_')] = TFGraphNode( + layer, data_format=self.tf_data_format) for layer_name, node in self.node_map.items(): for in_node in node.layer.input: - in_node = in_node.replace('/', - '_').replace('-', - '_').replace('^', '') + in_node = in_node.replace('/', '_').replace('-', '_').replace( + '^', '') if in_node not in self.node_map: if in_node.strip().split(':')[0] in self.node_map: self.connect(in_node.strip().split(':')[0], layer_name) @@ -390,10 +390,10 @@ class TFDecoder(object): shape=shape, name="x2paddle_{}".format(layer.name)) except: - x2paddle_input = tf.placeholder(dtype=dtype, - shape=shape, - name="x2paddle_{}".format( - layer.name)) + x2paddle_input = tf.placeholder( + dtype=dtype, + shape=shape, + name="x2paddle_{}".format(layer.name)) input_map["{}:0".format(layer.name)] = x2paddle_input if shape.count(None) > 0: diff --git a/x2paddle/op_mapper/caffe_custom_layer/convolutiondepthwise.py b/x2paddle/op_mapper/caffe_custom_layer/convolutiondepthwise.py index 7011100ba6cb701f896646a11b81e64f2c51eae4..670d5436af6af54bdc74dd1a79e4ef4e30f42205 100644 --- a/x2paddle/op_mapper/caffe_custom_layer/convolutiondepthwise.py +++ b/x2paddle/op_mapper/caffe_custom_layer/convolutiondepthwise.py @@ -122,16 +122,17 @@ def convolutiondepthwise_layer(inputs, c_out = num_output if num_output is not None else input_shape[0][1] group = int(c_in / (c_in / c_out)) if c_in > c_out else int(c_in / (c_out / c_in)) - out = fluid.layers.conv2d(input, - dilation=[dila_h, dila_w], - filter_size=[k_h, k_w], - stride=[s_h, s_w], - padding=[p_h, p_w], - groups=group, - num_filters=c_out, - param_attr=name + '_weights', - bias_attr=name + '_bias', - name=name) + out = fluid.layers.conv2d( + input, + dilation=[dila_h, dila_w], + filter_size=[k_h, k_w], + stride=[s_h, s_w], + padding=[p_h, p_w], + groups=group, + num_filters=c_out, + param_attr=name + '_weights', + bias_attr=name + '_bias', + name=name) return out @@ -142,7 +143,8 @@ def convolutiondepthwise_weights(name, data=None): return weights_name -register(kind='ConvolutionDepthwise', - shape=convolutiondepthwise_shape, - layer=convolutiondepthwise_layer, - weights=convolutiondepthwise_weights) +register( + kind='ConvolutionDepthwise', + shape=convolutiondepthwise_shape, + layer=convolutiondepthwise_layer, + weights=convolutiondepthwise_weights) diff --git a/x2paddle/op_mapper/caffe_custom_layer/detectionoutput.py b/x2paddle/op_mapper/caffe_custom_layer/detectionoutput.py index 173f5f31d5f26545a112d11b0994d73097ebb16b..e8463d52df19c191d3793bc8ceae57a6ff3a56a1 100644 --- a/x2paddle/op_mapper/caffe_custom_layer/detectionoutput.py +++ b/x2paddle/op_mapper/caffe_custom_layer/detectionoutput.py @@ -37,8 +37,8 @@ def detectionoutput_layer(inputs, pbv = fluid.layers.reshape(x=pbv, shape=[-1, 4]) mbox_loc = inputs[0] mbox_loc = fluid.layers.reshape(x=mbox_loc, shape=[-1, pb.shape[0], 4]) - mbox_conf_flatten = fluid.layers.reshape(x=mbox_conf_flatten, - shape=[0, pb.shape[0], -1]) + mbox_conf_flatten = fluid.layers.reshape( + x=mbox_conf_flatten, shape=[0, pb.shape[0], -1]) default = {"nms_threshold": 0.3, "top_k": 10, "eta": 1.0} fields = ['eta', 'top_k', 'nms_threshold'] @@ -64,7 +64,8 @@ def detectionoutput_weights(name, data=None): return weights_name -register(kind='DetectionOutput', - shape=detectionoutput_shape, - layer=detectionoutput_layer, - weights=detectionoutput_weights) +register( + kind='DetectionOutput', + shape=detectionoutput_shape, + layer=detectionoutput_layer, + weights=detectionoutput_weights) diff --git a/x2paddle/op_mapper/caffe_custom_layer/normalize.py b/x2paddle/op_mapper/caffe_custom_layer/normalize.py index 3cef4e6551a2656a0da656bf5330deffa111a061..15bbb2043a1e385d987c577d099c8db902de3f26 100644 --- a/x2paddle/op_mapper/caffe_custom_layer/normalize.py +++ b/x2paddle/op_mapper/caffe_custom_layer/normalize.py @@ -20,9 +20,8 @@ def normalize_layer(inputs, attr=name + '_scale') scale_param = fluid.layers.reshape(x=scale_param, \ shape=[1] if channel_shared else [input_shape[0][1]]) - out = fluid.layers.elementwise_mul(x=l2_norm, - y=scale_param, - axis=-1 if channel_shared else 1) + out = fluid.layers.elementwise_mul( + x=l2_norm, y=scale_param, axis=-1 if channel_shared else 1) return out @@ -31,7 +30,8 @@ def normalize_weights(name, data=None): return weights_name -register(kind='Normalize', - shape=normalize_shape, - layer=normalize_layer, - weights=normalize_weights) +register( + kind='Normalize', + shape=normalize_shape, + layer=normalize_layer, + weights=normalize_weights) diff --git a/x2paddle/op_mapper/caffe_custom_layer/permute.py b/x2paddle/op_mapper/caffe_custom_layer/permute.py index f8dac3a3a4832c92e17a504a93a17741d811e07e..617af1ff7fbc68bda84556deb6a86d96a01a0cf4 100644 --- a/x2paddle/op_mapper/caffe_custom_layer/permute.py +++ b/x2paddle/op_mapper/caffe_custom_layer/permute.py @@ -23,7 +23,8 @@ def permute_weights(name, data=None): return weights_name -register(kind='Permute', - shape=permute_shape, - layer=permute_layer, - weights=permute_weights) +register( + kind='Permute', + shape=permute_shape, + layer=permute_layer, + weights=permute_weights) diff --git a/x2paddle/op_mapper/caffe_custom_layer/priorbox.py b/x2paddle/op_mapper/caffe_custom_layer/priorbox.py index 829c3e365fefbfc81b9e4efb382726f1d473ecab..ba41335ecab6293f40f43bb349f80376c71be7ca 100644 --- a/x2paddle/op_mapper/caffe_custom_layer/priorbox.py +++ b/x2paddle/op_mapper/caffe_custom_layer/priorbox.py @@ -30,18 +30,19 @@ def priorbox_layer(inputs, steps = tuple(step) if type(step) is list or type(step) is tuple else (step, step) - box, variance_ = fluid.layers.prior_box(input, - image, - min_sizes=min_size, - max_sizes=max_size, - aspect_ratios=aspect_ratio, - variance=variance, - flip=flip, - clip=clip, - steps=steps, - offset=offset, - name=name, - min_max_aspect_ratios_order=True) + box, variance_ = fluid.layers.prior_box( + input, + image, + min_sizes=min_size, + max_sizes=max_size, + aspect_ratios=aspect_ratio, + variance=variance, + flip=flip, + clip=clip, + steps=steps, + offset=offset, + name=name, + min_max_aspect_ratios_order=True) box = fluid.layers.reshape(box, [1, 1, -1]) variance_ = fluid.layers.reshape(variance_, [1, 1, -1]) out = fluid.layers.concat([box, variance_], axis=1) @@ -53,7 +54,8 @@ def priorbox_weights(name, data=None): return weights_name -register(kind='PriorBox', - shape=priorbox_shape, - layer=priorbox_layer, - weights=priorbox_weights) +register( + kind='PriorBox', + shape=priorbox_shape, + layer=priorbox_layer, + weights=priorbox_weights) diff --git a/x2paddle/op_mapper/caffe_custom_layer/register.py b/x2paddle/op_mapper/caffe_custom_layer/register.py index 183ed4404e1e48d2e9d6c3324e1ebdfd513adc01..1d68d7fbb7f058451c4a27304943ebf85d341a21 100644 --- a/x2paddle/op_mapper/caffe_custom_layer/register.py +++ b/x2paddle/op_mapper/caffe_custom_layer/register.py @@ -23,8 +23,7 @@ def register(kind, shape, layer, weights): kind = [kind] else: assert type( - kind - ) is list, 'invalid param "kind" for register, not a list or str' + kind) is list, 'invalid param "kind" for register, not a list or str' for k in kind: assert type( diff --git a/x2paddle/op_mapper/caffe_custom_layer/roipooling.py b/x2paddle/op_mapper/caffe_custom_layer/roipooling.py index 1eece0d1b62f754ed2f3cc35f18096e383f1d8e8..5e0d459f1302bb94e99cc922379e739ce087a18e 100644 --- a/x2paddle/op_mapper/caffe_custom_layer/roipooling.py +++ b/x2paddle/op_mapper/caffe_custom_layer/roipooling.py @@ -21,11 +21,12 @@ def roipooling_layer(inputs, input = inputs[0] roi = inputs[1] roi = fluid.layers.slice(roi, axes=[1], starts=[1], ends=[5]) - out = fluid.layers.roi_pool(input, - roi, - pooled_height=pooled_h, - pooled_width=pooled_w, - spatial_scale=spatial_scale) + out = fluid.layers.roi_pool( + input, + roi, + pooled_height=pooled_h, + pooled_width=pooled_w, + spatial_scale=spatial_scale) return out @@ -34,7 +35,8 @@ def roipooling_weights(name, data=None): return weights_name -register(kind='ROIPooling', - shape=roipooling_shape, - layer=roipooling_layer, - weights=roipooling_weights) +register( + kind='ROIPooling', + shape=roipooling_shape, + layer=roipooling_layer, + weights=roipooling_weights) diff --git a/x2paddle/op_mapper/caffe_custom_layer/select.py b/x2paddle/op_mapper/caffe_custom_layer/select.py index e4a635a4434f4b87785a17a28a752104352a3e6b..497a3c24212b66d6686ff0bc8b2977195c165e28 100644 --- a/x2paddle/op_mapper/caffe_custom_layer/select.py +++ b/x2paddle/op_mapper/caffe_custom_layer/select.py @@ -30,11 +30,12 @@ def select_layer(inputs, out = [] for i in range(len(slice_point)): out.append( - fluid.layers.slice(input, - axes=[axis], - starts=[slice_point[i]], - ends=[slice_point[i + 1]], - name=name + '_' + str(i))) + fluid.layers.slice( + input, + axes=[axis], + starts=[slice_point[i]], + ends=[slice_point[i + 1]], + name=name + '_' + str(i))) if i == len(slice_point) - 2: break return out @@ -45,7 +46,8 @@ def select_weights(name, data=None): return weights_name -register(kind='Select', - shape=select_shape, - layer=select_layer, - weights=select_weights) +register( + kind='Select', + shape=select_shape, + layer=select_layer, + weights=select_weights) diff --git a/x2paddle/op_mapper/caffe_custom_layer/shufflechannel.py b/x2paddle/op_mapper/caffe_custom_layer/shufflechannel.py index 191f23c459066ccc731e2afa9770e2046a197591..d71404689d032ac4884c9b944999cf5e12e95e40 100644 --- a/x2paddle/op_mapper/caffe_custom_layer/shufflechannel.py +++ b/x2paddle/op_mapper/caffe_custom_layer/shufflechannel.py @@ -17,7 +17,8 @@ def shufflechannel_weights(name, data=None): return weights_name -register(kind='ShuffleChannel', - shape=shufflechannel_shape, - layer=shufflechannel_layer, - weights=shufflechannel_weights) +register( + kind='ShuffleChannel', + shape=shufflechannel_shape, + layer=shufflechannel_layer, + weights=shufflechannel_weights) diff --git a/x2paddle/op_mapper/caffe_op_mapper.py b/x2paddle/op_mapper/caffe_op_mapper.py index 8e46fdcb8a3e402e589d78110b3ebeb589fb404f..f350c50ea707518e296c6e807e7d89cca686bbcf 100644 --- a/x2paddle/op_mapper/caffe_op_mapper.py +++ b/x2paddle/op_mapper/caffe_op_mapper.py @@ -144,8 +144,8 @@ class CaffeOpMapper(OpMapper): [s_h, s_w] = [params.stride] * 2 elif len(params.stride) > 0: s_h = params.stride_h if params.stride_h > 0 else params.stride[0] - s_w = params.stride_w if params.stride_w > 0 else params.stride[ - len(params.stride) - 1] + s_w = params.stride_w if params.stride_w > 0 else params.stride[len( + params.stride) - 1] elif params.stride_h > 0 or params.stride_w > 0: s_h = params.stride_h s_w = params.stride_w @@ -154,8 +154,8 @@ class CaffeOpMapper(OpMapper): [p_h, p_w] = [params.pad] * 2 elif len(params.pad) > 0: p_h = params.pad_h if params.pad_h > 0 else params.pad[0] - p_w = params.pad_w if params.pad_w > 0 else params.pad[ - len(params.pad) - 1] + p_w = params.pad_w if params.pad_w > 0 else params.pad[len( + params.pad) - 1] elif params.pad_h > 0 or params.pad_w > 0: p_h = params.pad_h p_w = params.pad_w @@ -195,10 +195,8 @@ class CaffeOpMapper(OpMapper): 'shape': shape, 'name': string(node.layer_name) } - node.fluid_code.add_layer("data", - inputs=None, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "data", inputs=None, output=node, param_attr=attr) def MemoryData(self, node): # TODO(syf): Paddlepaddle can't fully support @@ -209,10 +207,8 @@ class CaffeOpMapper(OpMapper): 'shape': shape, 'name': string(node.layer_name) } - node.fluid_code.add_layer("data", - inputs=None, - output=node.layer_name + '0', - param_attr=attr) + node.fluid_code.add_layer( + "data", inputs=None, output=node.layer_name + '0', param_attr=attr) node.fluid_code.add_note('{} = [{}]'.format(node.layer_name, node.layer_name + '0')) @@ -229,11 +225,9 @@ class CaffeOpMapper(OpMapper): input_c = node.input_shape[0][1] output_c = channel data.append( - np.zeros([output_c, input_c, kernel[0], - kernel[1]]).astype('float32')) - data.append(np.zeros([ - output_c, - ])).astype('float32') + np.zeros([output_c, input_c, kernel[0], kernel[1]]).astype( + 'float32')) + data.append(np.zeros([output_c, ])).astype('float32') else: data = self.adjust_parameters(node) self.weights[node.layer_name + '_weights'] = data[0] @@ -244,29 +238,19 @@ class CaffeOpMapper(OpMapper): input = self.graph.get_bottom_node(node, idx=0, copy=True) attr = { - 'filter_size': - kernel, - 'num_filters': - channel, - 'stride': - stride, - 'padding': - pad, - 'dilation': - dilation, - 'groups': - group, - 'name': - string(node.layer_name), - 'param_attr': - string(node.layer_name + '_weights'), - 'bias_attr': - False if len(data) == 1 else string(node.layer_name + '_bias'), + 'filter_size': kernel, + 'num_filters': channel, + 'stride': stride, + 'padding': pad, + 'dilation': dilation, + 'groups': group, + 'name': string(node.layer_name), + 'param_attr': string(node.layer_name + '_weights'), + 'bias_attr': False + if len(data) == 1 else string(node.layer_name + '_bias'), } - node.fluid_code.add_layer("conv2d", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "conv2d", inputs=input, output=node, param_attr=attr) def Deconvolution(self, node): data = node.data @@ -281,11 +265,9 @@ class CaffeOpMapper(OpMapper): input_c = node.input_shape[0][1] output_c = channel data.append( - np.zeros([output_c, input_c, kernel[0], - kernel[1]]).astype('float32')) - data.append(np.zeros([ - output_c, - ]).astype('float32')) + np.zeros([output_c, input_c, kernel[0], kernel[1]]).astype( + 'float32')) + data.append(np.zeros([output_c, ]).astype('float32')) else: data = self.adjust_parameters(node) self.weights[node.layer_name + '_weights'] = data[0] @@ -295,31 +277,20 @@ class CaffeOpMapper(OpMapper): ) == 1, 'The count of Deconvolution node\'s input is not 1.' input = self.graph.get_bottom_node(node, idx=0, copy=True) attr = { - 'output_size': - None, - 'filter_size': - kernel, - 'num_filters': - channel, - 'stride': - stride, - 'padding': - pad, - 'dilation': - dilation, - 'groups': - group, - 'name': - string(node.layer_name), - 'param_attr': - string(node.layer_name + '_weights'), - 'bias_attr': - False if len(data) == 1 else string(node.layer_name + '_bias') + 'output_size': None, + 'filter_size': kernel, + 'num_filters': channel, + 'stride': stride, + 'padding': pad, + 'dilation': dilation, + 'groups': group, + 'name': string(node.layer_name), + 'param_attr': string(node.layer_name + '_weights'), + 'bias_attr': False + if len(data) == 1 else string(node.layer_name + '_bias') } - node.fluid_code.add_layer("conv2d_transpose", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "conv2d_transpose", inputs=input, output=node, param_attr=attr) def Pooling(self, node): params = node.layer.pooling_param @@ -345,10 +316,8 @@ class CaffeOpMapper(OpMapper): 'global_pooling': global_pool, 'name': string(node.layer_name) } - node.fluid_code.add_layer("pool2d", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "pool2d", inputs=input, output=node, param_attr=attr) def LRN(self, node): assert len(node.inputs) == 1, 'The count of LRN node\'s input is not 1.' @@ -368,10 +337,8 @@ class CaffeOpMapper(OpMapper): 'beta': params.beta, 'name': string(node.layer_name) } - node.fluid_code.add_layer("lrn", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "lrn", inputs=input, output=node, param_attr=attr) def InnerProduct(self, node): data = node.data @@ -384,8 +351,8 @@ class CaffeOpMapper(OpMapper): output_c = params.num_output data = [] data.append( - np.zeros([input_c, - output_c]).astype('float32').astype('float32')) + np.zeros([input_c, output_c]).astype('float32').astype( + 'float32')) data.append( np.zeros([output_c]).astype('float32').astype('float32')) else: @@ -409,21 +376,15 @@ class CaffeOpMapper(OpMapper): assert params.bias_term == True input = self.graph.get_bottom_node(node, idx=0, copy=True) attr = { - 'size': - params.num_output, - 'name': - string(node.layer_name), - 'act': - None, - 'param_attr': - string(node.layer_name + '_weights'), - 'bias_attr': - False if len(data) == 1 else string(node.layer_name + '_bias') + 'size': params.num_output, + 'name': string(node.layer_name), + 'act': None, + 'param_attr': string(node.layer_name + '_weights'), + 'bias_attr': False + if len(data) == 1 else string(node.layer_name + '_bias') } - node.fluid_code.add_layer("fc", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "fc", inputs=input, output=node, param_attr=attr) def Softmax(self, node): assert len( @@ -435,10 +396,8 @@ class CaffeOpMapper(OpMapper): dims = len(shape) axis = axis + dims if axis < 0 else axis attr = {'axis': axis, 'name': string(node.layer_name + '_softmax')} - node.fluid_code.add_layer("softmax", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "softmax", inputs=input, output=node, param_attr=attr) def Slice(self, node): assert len( @@ -459,10 +418,8 @@ class CaffeOpMapper(OpMapper): 'dim': axis, 'name': string(node.layer_name) } - node.fluid_code.add_layer("split", - inputs=input, - output=node.layer_name, - param_attr=attr) + node.fluid_code.add_layer( + "split", inputs=input, output=node.layer_name, param_attr=attr) def Concat(self, node): assert len( @@ -475,10 +432,8 @@ class CaffeOpMapper(OpMapper): params = node.layer.concat_param axis = params.axis attr = {'axis': axis, 'name': string(node.layer_name)} - node.fluid_code.add_layer("concat", - inputs=inputs, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "concat", inputs=inputs, output=node, param_attr=attr) def PReLU(self, node): assert len( @@ -499,10 +454,8 @@ class CaffeOpMapper(OpMapper): 'param_attr': string(node.layer_name + '_weights'), 'name': string(node.layer_name) } - node.fluid_code.add_layer("prelu", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "prelu", inputs=input, output=node, param_attr=attr) def Accuracy(self, node): assert len( @@ -526,10 +479,8 @@ class CaffeOpMapper(OpMapper): assert axis == 1, 'PaddlePaddle can not support the situation when the axis is not 1.' assert not ignore_label >= 0, 'PaddlePaddle can not support the situation when the model has ignore label.' attr = {'k': top_k} - node.fluid_code.add_layer("accuracy", - inputs=inputs, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "accuracy", inputs=inputs, output=node, param_attr=attr) def Eltwise(self, node): assert len( @@ -546,10 +497,11 @@ class CaffeOpMapper(OpMapper): inputs_dict['x'] = inputs[0] inputs_dict['y'] = inputs[1] attr = {'act': None, 'name': string(node.layer_name)} - node.fluid_code.add_layer("elementwise_mul", - inputs=inputs_dict, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "elementwise_mul", + inputs=inputs_dict, + output=node, + param_attr=attr) elif mode == 1: if hasattr(params, 'coeff') and len(params.coeff) == 2: coeff = params.coeff @@ -559,57 +511,62 @@ class CaffeOpMapper(OpMapper): 'value': coeff[0], 'dtype': '{}.dtype'.format(input1_name) } - node.fluid_code.add_layer("fill_constant", - inputs=None, - output=node.layer_name + '_const1', - param_attr=attr) + node.fluid_code.add_layer( + "fill_constant", + inputs=None, + output=node.layer_name + '_const1', + param_attr=attr) attr = {'act': None, 'name': string(node.layer_name + '_mul1')} - node.fluid_code.add_layer("elementwise_mul", - inputs=input1_name + ', ' + - node.layer_name + '_const1', - output=node.layer_name + '_mul1', - param_attr=attr) + node.fluid_code.add_layer( + "elementwise_mul", + inputs=input1_name + ', ' + node.layer_name + '_const1', + output=node.layer_name + '_mul1', + param_attr=attr) input2_name = self.get_input_name(inputs[1]) attr = { 'shape': [1], 'value': coeff[1], 'dtype': '{}.dtype'.format(input2_name) } - node.fluid_code.add_layer("fill_constant", - inputs=None, - output=node.layer_name + '_const2', - param_attr=attr) + node.fluid_code.add_layer( + "fill_constant", + inputs=None, + output=node.layer_name + '_const2', + param_attr=attr) attr = {'act': None, 'name': string(node.layer_name + '_mul2')} - node.fluid_code.add_layer("elementwise_mul", - inputs=input2_name + ', ' + - node.layer_name + '_const2', - output=node.layer_name + '_mul2', - param_attr=attr) + node.fluid_code.add_layer( + "elementwise_mul", + inputs=input2_name + ', ' + node.layer_name + '_const2', + output=node.layer_name + '_mul2', + param_attr=attr) attr = {'act': None, 'name': string(node.layer_name)} - node.fluid_code.add_layer("elementwise_add", - inputs='{}_mul1, {}_mul2'.format( - node.layer_name, node.layer_name), - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "elementwise_add", + inputs='{}_mul1, {}_mul2'.format(node.layer_name, + node.layer_name), + output=node, + param_attr=attr) else: inputs_dict = {} inputs_dict['x'] = inputs[0] inputs_dict['y'] = inputs[1] attr = {'act': None, 'name': string(node.layer_name)} - node.fluid_code.add_layer("elementwise_add", - inputs=inputs_dict, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "elementwise_add", + inputs=inputs_dict, + output=node, + param_attr=attr) else: inputs_dict = {} inputs_dict['x'] = inputs[0] inputs_dict['y'] = inputs[1] attr = {'act': None, 'name': string(node.layer_name)} - node.fluid_code.add_layer("elementwise_max", - inputs=inputs_dict, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "elementwise_max", + inputs=inputs_dict, + output=node, + param_attr=attr) def BatchNorm(self, node): assert len( @@ -625,12 +582,8 @@ class CaffeOpMapper(OpMapper): 'The parameter of {} (type is {}) is not set. So we set the parameters as 0' .format(node.layer_name, node.layer_type)) input_c = node.input_shape[0][1] - mean = np.zeros([ - input_c, - ]).astype('float32') - variance = np.zeros([ - input_c, - ]).astype('float32') + mean = np.zeros([input_c, ]).astype('float32') + variance = np.zeros([input_c, ]).astype('float32') scale = 0 else: @@ -651,10 +604,8 @@ class CaffeOpMapper(OpMapper): 'epsilon': eps, 'name': string(node.layer_name) } - node.fluid_code.add_layer("batch_norm", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "batch_norm", inputs=input, output=node, param_attr=attr) def Scale(self, node): if node.data is None: @@ -669,10 +620,10 @@ class CaffeOpMapper(OpMapper): input_c, ]).astype('float32') else: - self.weights[node.layer_name + '_scale'] = np.squeeze( - node.data[0]).astype('float32') - self.weights[node.layer_name + '_offset'] = np.squeeze( - node.data[1]).astype('float32') + self.weights[node.layer_name + '_scale'] = np.squeeze(node.data[ + 0]).astype('float32') + self.weights[node.layer_name + '_offset'] = np.squeeze(node.data[ + 1]).astype('float32') params = node.layer.scale_param axis = params.axis num_axes = params.num_axes @@ -687,10 +638,11 @@ class CaffeOpMapper(OpMapper): inputs_dict['x'] = input0 inputs_dict['y'] = input1 attr = {'axis': axis, 'name': string(node.layer_name + '_mul')} - node.fluid_code.add_layer("elementwise_mul", - inputs=inputs_dict, - output=node.layer_name + '_mul', - param_attr=attr) + node.fluid_code.add_layer( + "elementwise_mul", + inputs=inputs_dict, + output=node.layer_name + '_mul', + param_attr=attr) else: bias_shape = node.input_shape[0][axis:axis + num_axes] input0 = self.graph.get_bottom_node(node, idx=0, copy=True) @@ -703,18 +655,17 @@ class CaffeOpMapper(OpMapper): 'is_bias': True, 'default_initializer': 'Constant(value=1.0)' } - node.fluid_code.add_layer("create_parameter", - inputs=None, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "create_parameter", inputs=None, output=node, param_attr=attr) inputs_dict = {} inputs_dict['x'] = input0 inputs_dict['y'] = node attr = {'axis': axis, 'name': string(node.layer_name + '_mul')} - node.fluid_code.add_layer("elementwise_mul", - inputs=inputs_dict, - output=node.layer_name + '_mul', - param_attr=attr) + node.fluid_code.add_layer( + "elementwise_mul", + inputs=inputs_dict, + output=node.layer_name + '_mul', + param_attr=attr) scale_shape = bias_shape input0_name = self.get_input_name(input0) attr = { @@ -725,16 +676,18 @@ class CaffeOpMapper(OpMapper): 'is_bias': True, 'default_initializer': 'Constant(value=1.0)' } - node.fluid_code.add_layer("create_parameter", - inputs=None, - output=node.layer_name + '_offset_param', - param_attr=attr) + node.fluid_code.add_layer( + "create_parameter", + inputs=None, + output=node.layer_name + '_offset_param', + param_attr=attr) attr = {'axis': axis, 'name': string(node.layer_name + '_add')} - node.fluid_code.add_layer("elementwise_add", - inputs='{}_mul, {}_offset_param'.format( - node.layer_name, node.layer_name), - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "elementwise_add", + inputs='{}_mul, {}_offset_param'.format(node.layer_name, + node.layer_name), + output=node, + param_attr=attr) def Reshape(self, node): input = self.graph.get_bottom_node(node, idx=0, copy=True) @@ -747,10 +700,8 @@ class CaffeOpMapper(OpMapper): 'act': None, 'name': string(node.layer_name) } - node.fluid_code.add_layer("reshape", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "reshape", inputs=input, output=node, param_attr=attr) def ArgMax(self, node): assert len(node.inputs) == 1 and len( @@ -767,11 +718,12 @@ class CaffeOpMapper(OpMapper): axis += len(input_shape) if out_max_val is True: attr = {'k': top_k, 'name': string(node.layer_name + '_topk')} - node.fluid_code.add_layer("topk", - inputs=input, - output='{}_topk_var, {}_index_var'.format( - node.layer_name, node.layer_name), - param_attr=attr) + node.fluid_code.add_layer( + "topk", + inputs=input, + output='{}_topk_var, {}_index_var'.format(node.layer_name, + node.layer_name), + param_attr=attr) attr = {'dtype': '{}_topk_var.dtype'.format(node.layer_name)} node.fluid_code.add_layer( "cast", @@ -779,17 +731,19 @@ class CaffeOpMapper(OpMapper): output='{}_index_var'.format(node.layer_name), param_attr=attr) attr = {'axis': axis, 'name': string(node.layer_name)} - node.fluid_code.add_layer("concat", - inputs='{}_topk_var, {}_index_var'.format( - node.layer_name, node.layer_name), - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "concat", + inputs='{}_topk_var, {}_index_var'.format(node.layer_name, + node.layer_name), + output=node, + param_attr=attr) else: attr = {'k': top_k, 'name': string(node.layer_name)} - node.fluid_code.add_layer("topk", - inputs=input, - output='_, {}'.format(node.layer_name), - param_attr=attr) + node.fluid_code.add_layer( + "topk", + inputs=input, + output='_, {}'.format(node.layer_name), + param_attr=attr) def Crop(self, node): assert len( @@ -804,29 +758,27 @@ class CaffeOpMapper(OpMapper): offset_real = [0] * len(input_shape) if hasattr(params, "offset") and len(params.offset) > 0: offset = list(params.offset) - assert (len(input_shape) - axis) == len( - offset), "invalid offset[%s] in crop layer" % (str(offset)) + assert (len(input_shape) - axis + ) == len(offset), "invalid offset[%s] in crop layer" % ( + str(offset)) offset_real = [0] * axis + offset attr = {'offsets': list(offset_real), 'name': string(node.layer_name)} - node.fluid_code.add_layer("crop", - inputs={ - 'x': input, - 'shape': node.input_shape[1] - }, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "crop", + inputs={'x': input, + 'shape': node.input_shape[1]}, + output=node, + param_attr=attr) def Flatten(self, node): assert len( - node.inputs - ) == 1, 'The count of DetectionOutput node\'s input is not 1.' + node. + inputs) == 1, 'The count of DetectionOutput node\'s input is not 1.' input = self.graph.get_bottom_node(node, idx=0, copy=True) shape = node.output_shape[0] attr = {'shape': shape, 'name': string(node.layer_name)} - node.fluid_code.add_layer("reshape", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "reshape", inputs=input, output=node, param_attr=attr) def Power(self, node): assert len( @@ -842,15 +794,11 @@ class CaffeOpMapper(OpMapper): 'bias_after_scale': True, 'name': string(node.layer_name + '_scale') } - node.fluid_code.add_layer("scale", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "scale", inputs=input, output=node, param_attr=attr) attr = {'factor': power, 'name': string(node.layer_name)} - node.fluid_code.add_layer("pow", - inputs=node, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "pow", inputs=node, output=node, param_attr=attr) def Reduction(self, node): assert len( @@ -872,55 +820,41 @@ class CaffeOpMapper(OpMapper): 'keep_dim': False, 'name': string(node.layer_name) } - node.fluid_code.add_layer("reduce_sum", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "reduce_sum", inputs=input, output=node, param_attr=attr) elif operation == 2: ## operation = ASUM attr = {'name': string(node.layer_name + '_abs')} - node.fluid_code.add_layer("abs", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "abs", inputs=input, output=node, param_attr=attr) attr = { 'dim': dim[axis:], 'keep_dim': False, 'name': string(node.layer_name) } - node.fluid_code.add_layer("reduce_sum", - inputs=node, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "reduce_sum", inputs=node, output=node, param_attr=attr) elif operation == 3: ## operation = SUMSQ attr = {'factor': 2.0, 'name': string(node.layer_name + '_pow')} - node.fluid_code.add_layer("pow", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "pow", inputs=input, output=node, param_attr=attr) attr = { 'dim': dim[axis:], 'keep_dim': False, 'name': string(node.layer_name) } - node.fluid_code.add_layer("reduce_sum", - inputs=node, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "reduce_sum", inputs=node, output=node, param_attr=attr) else: ## operation = MEAN attr = { 'dim': dim[axis:], 'keep_dim': False, 'name': string(node.layer_name) } - node.fluid_code.add_layer("reduce_mean", - inputs=node, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "reduce_mean", inputs=node, output=node, param_attr=attr) attr = {'scale': coeff} - node.fluid_code.add_layer("scale", - inputs=node, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "scale", inputs=node, output=node, param_attr=attr) def deal_custom_layer(self, node): op = node.layer_type @@ -947,11 +881,12 @@ class CaffeOpMapper(OpMapper): assert input is not None, 'This kind of DetectionOutput is not supported!' input = self.graph.get_bottom_node(input, idx=0, copy=True) inputs_node.append(input) - node.fluid_code.add_layer(func.__code__.co_name, - inputs=inputs_node, - output=node, - param_attr=kwargs, - is_custom_layer=True) + node.fluid_code.add_layer( + func.__code__.co_name, + inputs=inputs_node, + output=node, + param_attr=kwargs, + is_custom_layer=True) if op not in self.used_custom_layers: self.used_custom_layers[op] = custom_code @@ -960,7 +895,5 @@ class CaffeOpMapper(OpMapper): op_info = self.directly_map_ops[node.layer_type] input = self.graph.get_bottom_node(node, idx=0, copy=True) attr = {'name': string(node.layer_name)} - node.fluid_code.add_layer(op_info, - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + op_info, inputs=input, output=node, param_attr=attr) diff --git a/x2paddle/op_mapper/caffe_shape.py b/x2paddle/op_mapper/caffe_shape.py index 608a9d242aa7b703a070916f6cfaebc29e574dad..c574193d45400f351da0058f5d80b152cd6505cb 100644 --- a/x2paddle/op_mapper/caffe_shape.py +++ b/x2paddle/op_mapper/caffe_shape.py @@ -33,8 +33,8 @@ def get_kernel_parameters(params): [s_h, s_w] = [params.stride] * 2 elif len(params.stride) > 0: s_h = params.stride_h if params.stride_h > 0 else params.stride[0] - s_w = params.stride_w if params.stride_w > 0 else params.stride[ - len(params.stride) - 1] + s_w = params.stride_w if params.stride_w > 0 else params.stride[len( + params.stride) - 1] elif params.stride_h > 0 or params.stride_w > 0: s_h = params.stride_h s_w = params.stride_w diff --git a/x2paddle/op_mapper/onnx_custom_layer/InstanceNormalization.py b/x2paddle/op_mapper/onnx_custom_layer/InstanceNormalization.py index db5e2088613a334d9b29dfaca9b6d486e61ee1ee..f47273206d3108316e5a3a8a72da5a1f96f8fe9e 100644 --- a/x2paddle/op_mapper/onnx_custom_layer/InstanceNormalization.py +++ b/x2paddle/op_mapper/onnx_custom_layer/InstanceNormalization.py @@ -24,21 +24,18 @@ def InstanceNormalization_layer(inputs, name=None): epsilon = 1e-5 input_ = inputs[0] mean = fluid.layers.reduce_mean(input_, dim=[2, 3], keep_dim=True) - var = fluid.layers.reduce_mean(fluid.layers.square(input_ - mean), - dim=[2, 3], - keep_dim=True) + var = fluid.layers.reduce_mean( + fluid.layers.square(input_ - mean), dim=[2, 3], keep_dim=True) if name is not None: scale_name = name + "_scale" offset_name = name + "_offset" scale_param = inputs[1] offset_param = inputs[2] - scale = fluid.layers.create_parameter(name=scale_param.name, - shape=input_.shape[1:2], - dtype="float32") - offset = fluid.layers.create_parameter(name=offset_param.name, - shape=input_.shape[1:2], - dtype="float32") + scale = fluid.layers.create_parameter( + name=scale_param.name, shape=input_.shape[1:2], dtype="float32") + offset = fluid.layers.create_parameter( + name=offset_param.name, shape=input_.shape[1:2], dtype="float32") tmp = fluid.layers.elementwise_mul(x=(input_ - mean), y=scale, axis=1) tmp = tmp / fluid.layers.sqrt(var + epsilon) @@ -51,8 +48,9 @@ def InstanceNormalization_weights(name, data=None): return weights_name -register(kind='InstanceNormalization', - shape=InstanceNormalization_shape, - layer=InstanceNormalization_layer, - child_func=None, - weights=InstanceNormalization_weights) +register( + kind='InstanceNormalization', + shape=InstanceNormalization_shape, + layer=InstanceNormalization_layer, + child_func=None, + weights=InstanceNormalization_weights) diff --git a/x2paddle/op_mapper/onnx_custom_layer/register.py b/x2paddle/op_mapper/onnx_custom_layer/register.py index f68bbaf03232ae65dc891353f3ae902fdf21f135..37a593648a0b8717eb22cde7f1bc8b85c4990e39 100644 --- a/x2paddle/op_mapper/onnx_custom_layer/register.py +++ b/x2paddle/op_mapper/onnx_custom_layer/register.py @@ -36,8 +36,7 @@ def register(kind, shape, layer, child_func, weights): kind = [kind] else: assert type( - kind - ) is list, 'invalid param "kind" for register, not a list or str' + kind) is list, 'invalid param "kind" for register, not a list or str' for k in kind: assert type( diff --git a/x2paddle/op_mapper/onnx_directly_map.py b/x2paddle/op_mapper/onnx_directly_map.py index ecd1a30321af5032366e3503f5e36ce95c3ed2fd..a1765f064153dca3e698857065a23b6650254f33 100644 --- a/x2paddle/op_mapper/onnx_directly_map.py +++ b/x2paddle/op_mapper/onnx_directly_map.py @@ -28,60 +28,49 @@ default_op_mapping_field_values['FILL_NAME_FIELD'] = True default_op_mapping = { 'Shape': ['shape', ['X'], ['Out']], 'Clip': [ - 'clip', ['X'], ['Out'], - dict(), - dict( - min=(_np.asarray([255, 255, 127, 255], - dtype=_np.uint8).view(_np.float32)[0]), - max=(_np.asarray([255, 255, 127, 127], - dtype=_np.uint8).view(_np.float32)[0]), - ) + 'clip', ['X'], ['Out'], dict(), dict( + min=(_np.asarray( + [255, 255, 127, 255], dtype=_np.uint8).view(_np.float32)[0]), + max=(_np.asarray( + [255, 255, 127, 127], dtype=_np.uint8).view(_np.float32)[0]), ) ], 'Erf': ['erf', ['X'], ['Out']], 'Ceil': ['ceil', ['X'], ['Out']], 'ReduceMean': [ - 'reduce_mean', ['X'], ['Out'], - dict(axes='dim', keepdims='keep_dim'), - dict(keep_dim=1) + 'reduce_mean', ['X'], ['Out'], dict( + axes='dim', keepdims='keep_dim'), dict(keep_dim=1) ], 'ReduceSum': [ - 'reduce_sum', ['X'], ['Out'], - dict(axes='dim', keepdims='keep_dim'), - dict(keep_dim=1) + 'reduce_sum', ['X'], ['Out'], dict( + axes='dim', keepdims='keep_dim'), dict(keep_dim=1) ], 'ReduceMin': [ - 'reduce_min', ['X'], ['Out'], - dict(axes='dim', keepdims='keep_dim'), - dict(keep_dim=1) + 'reduce_min', ['X'], ['Out'], dict( + axes='dim', keepdims='keep_dim'), dict(keep_dim=1) ], 'ReduceMax': [ - 'reduce_max', ['X'], ['Out'], - dict(axes='dim', keepdims='keep_dim'), - dict(keep_dim=1) + 'reduce_max', ['X'], ['Out'], dict( + axes='dim', keepdims='keep_dim'), dict(keep_dim=1) ], #active function 'Relu': ['relu', ['X'], ['Out']], - 'LeakyRelu': ['leaky_relu', ['X'], ['Out'], - dict(), dict(alpha=.01)], - 'Elu': ['elu', ['X'], ['Out'], - dict(), dict(alpha=1.)], + 'LeakyRelu': ['leaky_relu', ['X'], ['Out'], dict(), dict(alpha=.01)], + 'Elu': ['elu', ['X'], ['Out'], dict(), dict(alpha=1.)], 'ThresholdedRelu': [ - 'thresholded_relu', ['X'], ['Out'], - dict(alpha='threshold'), + 'thresholded_relu', ['X'], ['Out'], dict(alpha='threshold'), dict(alpha=1.) ], 'Tanh': ['tanh', ['X'], ['Out']], 'Sigmoid': ['sigmoid', ['X'], ['Out']], 'HardSigmoid': [ - 'hard_sigmoid', ['X'], ['Out'], - dict(alpha='slope', beta='offset'), - dict(slope=.2, offset=.5) + 'hard_sigmoid', ['X'], ['Out'], dict( + alpha='slope', beta='offset'), dict( + slope=.2, offset=.5) ], 'Softsign': ['softsign', ['X'], ['Out']], 'Softplus': ['softplus', ['X'], ['Out']], 'Exp': ['exp', ['X'], ['Out']], - 'Softmax': ['softmax', ['X'], ['Out'], - dict(), dict(axis=1)], + 'Softmax': ['softmax', ['X'], ['Out'], dict(), dict(axis=1)], 'Sqrt': ['sqrt', ['X'], ['Out']], 'Floor': ['floor', ['X'], ['Out']], 'Abs': ['abs', ['X'], ['Out']], diff --git a/x2paddle/op_mapper/onnx_op_mapper.py b/x2paddle/op_mapper/onnx_op_mapper.py index a8c656cb4487bb0af178f1b74f190fb47a4c2715..a50670b29f774a496dc0d6da450560848a47abed 100644 --- a/x2paddle/op_mapper/onnx_op_mapper.py +++ b/x2paddle/op_mapper/onnx_op_mapper.py @@ -140,8 +140,8 @@ class ONNXOpMapper(OpMapper): model.graph.ClearField('output') model.graph.output.MergeFrom(model.graph.value_info) - onnx.save(model, os.path.join(self.tmp_data_dir, - 'onnx_model_infer.onnx')) + onnx.save(model, + os.path.join(self.tmp_data_dir, 'onnx_model_infer.onnx')) sess = rt.InferenceSession( os.path.join(self.tmp_data_dir, 'onnx_model_infer.onnx')) res = sess.run(None, input_feed=inputs_dict) @@ -217,8 +217,7 @@ class ONNXOpMapper(OpMapper): default_attrs, input_perm, output_perm, - fill_name_field, - ) = info + fill_name_field, ) = info if fluid_op in default_ioa_constraint: for predicate, message in default_ioa_constraint[fluid_op]: @@ -429,10 +428,8 @@ class ONNXOpMapper(OpMapper): } node.fluid_code.add_layer( 'roi_align', - inputs={ - 'input': val_x, - 'rois': val_rois - }, + inputs={'input': val_x, + 'rois': val_rois}, output=node, param_attr=attr) @@ -449,10 +446,8 @@ class ONNXOpMapper(OpMapper): } node.fluid_code.add_layer( 'roi_pool', - inputs={ - 'input': val_x, - 'rois': val_rois - }, + inputs={'input': val_x, + 'rois': val_rois}, output=node, param_attr=attr) @@ -527,10 +522,8 @@ class ONNXOpMapper(OpMapper): val_y = self.graph.get_input_node(node, idx=1, copy=True) node.fluid_code.add_layer( 'greater_than', - inputs={ - 'x': val_x, - 'y': val_y - }, + inputs={'x': val_x, + 'y': val_y}, output=node, param_attr=None) @@ -549,11 +542,10 @@ class ONNXOpMapper(OpMapper): shape = val_output.out_shapes[0] if shape is None: shape = list(value.shape) - _logger.warning( - 'in (Constant -> %s): ' - 'attribute "shape" of %s not inferred, ' - 'using value as 1-D tensor may lead to fails', - val_output.layer_name, val_output.layer_name) + _logger.warning('in (Constant -> %s): ' + 'attribute "shape" of %s not inferred, ' + 'using value as 1-D tensor may lead to fails', + val_output.layer_name, val_output.layer_name) if len(value) == 1: value = value.tolist() @@ -616,10 +608,8 @@ class ONNXOpMapper(OpMapper): if axis == 0 and len(indices_shape) <= 1: node.fluid_code.add_layer( 'gather', - inputs={ - 'input': val_x, - 'index': indices - }, + inputs={'input': val_x, + 'index': indices}, output=node, param_attr=None) elif axis > 0 and len(indices_shape) <= 1: @@ -634,10 +624,8 @@ class ONNXOpMapper(OpMapper): param_attr=attr_trans) node.fluid_code.add_layer( 'gather', - inputs={ - 'input': name_trans, - 'index': indices - }, + inputs={'input': name_trans, + 'index': indices}, output=node, param_attr=None) node.fluid_code.add_layer( @@ -649,9 +637,7 @@ class ONNXOpMapper(OpMapper): 'reshape', inputs=indices, output=indices, - param_attr={'shape': [ - reshape_shape, - ]}) + param_attr={'shape': [reshape_shape, ]}) perm = list(range(len(val_x.out_shapes[0]))) perm = [axis] + perm[:axis] + perm[axis + 1:] @@ -664,10 +650,8 @@ class ONNXOpMapper(OpMapper): param_attr=attr_trans) node.fluid_code.add_layer( 'gather', - inputs={ - 'input': name_trans, - 'index': indices - }, + inputs={'input': name_trans, + 'index': indices}, output=node, param_attr=None) node.fluid_code.add_layer( @@ -926,8 +910,10 @@ class ONNXOpMapper(OpMapper): def Sum(self, node): val_inps = node.layer.input inputs = { - "x": self.graph.get_input_node(node, idx=0, copy=True), - "y": self.graph.get_input_node(node, idx=1, copy=True), + "x": self.graph.get_input_node( + node, idx=0, copy=True), + "y": self.graph.get_input_node( + node, idx=1, copy=True), } node.fluid_code.add_layer("elementwise_add", inputs=inputs, output=node) @@ -1022,10 +1008,8 @@ class ONNXOpMapper(OpMapper): val_y = self.graph.get_input_node(node, idx=1, copy=True) node.fluid_code.add_layer( "equal", - inputs={ - 'x': val_x, - 'y': val_y - }, + inputs={'x': val_x, + 'y': val_y}, output=node, param_attr=None) @@ -1055,29 +1039,23 @@ class ONNXOpMapper(OpMapper): mul_val_x = val_x.layer_name + '_mul' node.fluid_code.add_layer( "elementwise_mul", - inputs={ - 'x': val_x, - 'y': cast_condition - }, + inputs={'x': val_x, + 'y': cast_condition}, output=mul_val_x, param_attr=None) mul_val_y = val_y.layer_name + '_mul' node.fluid_code.add_layer( "elementwise_mul", - inputs={ - 'x': val_y, - 'y': cast_not_condition - }, + inputs={'x': val_y, + 'y': cast_not_condition}, output=mul_val_y, param_attr=None) node.fluid_code.add_layer( "elementwise_add", - inputs={ - 'x': mul_val_x, - 'y': mul_val_y - }, + inputs={'x': mul_val_x, + 'y': mul_val_y}, output=node, param_attr=None) @@ -1106,7 +1084,8 @@ class ONNXOpMapper(OpMapper): output=flatten_name, param_attr={'axis': 0}) node.fluid_code.add_layer( - "concat", inputs=flatten_names, output=node, param_attr={'axis': 0}) + "concat", inputs=flatten_names, output=node, + param_attr={'axis': 0}) def Identity(self, node): val_x = self.graph.get_input_node(node, idx=0, copy=True) @@ -1280,11 +1259,11 @@ class ONNXOpMapper(OpMapper): output_size = [0, 0] - output_size[0] = (val_x.out_shapes[0][2] - - 1) * strides[0] - 2 * paddings[0] + dilations[0] * ( + output_size[0] = (val_x.out_shapes[0][2] - 1 + ) * strides[0] - 2 * paddings[0] + dilations[0] * ( kernel_shape[0] - 1) + 1 + out_padding[0] - output_size[1] = (val_x.out_shapes[0][3] - - 1) * strides[1] - 2 * paddings[1] + dilations[1] * ( + output_size[1] = (val_x.out_shapes[0][3] - 1 + ) * strides[1] - 2 * paddings[1] + dilations[1] * ( kernel_shape[1] - 1) + 1 + out_padding[1] attr = { 'num_filters': num_out_channels, @@ -1367,29 +1346,23 @@ class ONNXOpMapper(OpMapper): 'squeeze', inputs=val_x, output=var_x0, - param_attr={ - 'axes': [1], - 'name': string(var_x0) - }) + param_attr={'axes': [1], + 'name': string(var_x0)}) var_w0 = node.layer_name + '_w0' node.fluid_code.add_layer( 'squeeze', inputs=val_w, output=var_w0, - param_attr={ - 'axes': [0], - 'name': string(var_w0) - }) + param_attr={'axes': [0], + 'name': string(var_w0)}) var_fc = node.layer_name + '_fc' var_mm = (node.layer_name + '_mm') if val_b else var_fc node.fluid_code.add_layer( 'matmul', - inputs={ - 'x': var_x0, - 'y': var_w0 - }, + inputs={'x': var_x0, + 'y': var_w0}, output=var_mm, param_attr={ 'transpose_x': 0, @@ -1402,10 +1375,8 @@ class ONNXOpMapper(OpMapper): 'squeeze', inputs=val_r, output=var_r0, - param_attr={ - 'axes': [0], - 'name': string(var_r0) - }) + param_attr={'axes': [0], + 'name': string(var_r0)}) var_r0t = node.layer_name + '_r0t' @@ -1413,10 +1384,8 @@ class ONNXOpMapper(OpMapper): 'transpose', inputs=var_r0, output=var_r0t, - param_attr={ - 'perm': [1, 0], - 'name': string(var_r0t) - }) + param_attr={'perm': [1, 0], + 'name': string(var_r0t)}) if val_b: var_bi = node.layer_name + '_bi' var_bh = node.layer_name + '_bh' @@ -1434,10 +1403,8 @@ class ONNXOpMapper(OpMapper): 'squeeze', inputs=var_bi, output=var_bi0, - param_attr={ - 'axes': [0], - 'name': string(var_bi0) - }) + param_attr={'axes': [0], + 'name': string(var_bi0)}) node.fluid_code.add_layer( 'elmentwise_add', @@ -1454,10 +1421,8 @@ class ONNXOpMapper(OpMapper): 'squeeze', inputs=val_xh, output=var_xh0, - param_attr={ - 'axes': [1], - 'name': string(var_xh0) - }) + param_attr={'axes': [1], + 'name': string(var_xh0)}) var_y00 = node.layer_name + '_y00' attr = { diff --git a/x2paddle/op_mapper/paddle_custom_layer/im2sequence.py b/x2paddle/op_mapper/paddle_custom_layer/im2sequence.py index 91fa2ca30a2a472d6190a424a522103f46b0f9d1..aeb4a9ceddb280295aafb2ebcafa3a25f8767d75 100644 --- a/x2paddle/op_mapper/paddle_custom_layer/im2sequence.py +++ b/x2paddle/op_mapper/paddle_custom_layer/im2sequence.py @@ -30,8 +30,8 @@ def im2sequence(op, block): slice_blocks = list() for i in range(out_h): for j in range(out_w): - starts_name = "im2sequence.starts.{}.{}.{}".format( - im2seq_counter, i, j) + starts_name = "im2sequence.starts.{}.{}.{}".format(im2seq_counter, + i, j) starts_tensor = helper.make_tensor( name=starts_name, data_type=onnx_pb.TensorProto.INT64, diff --git a/x2paddle/op_mapper/paddle_custom_layer/multiclass_nms.py b/x2paddle/op_mapper/paddle_custom_layer/multiclass_nms.py index 743c20f3fe001c2105dbc268bbfc7a8597a833b2..5d30f651bbe2e51a2d328a81ceb8c52391374d60 100644 --- a/x2paddle/op_mapper/paddle_custom_layer/multiclass_nms.py +++ b/x2paddle/op_mapper/paddle_custom_layer/multiclass_nms.py @@ -44,8 +44,7 @@ def multiclass_nms(op, block): if normalized == False: warnings.warn( 'The parameter normalized of multiclass_nms OP of Paddle is False, which has diff with ONNX. \ - Please set normalized=True in multiclass_nms of Paddle' - ) + Please set normalized=True in multiclass_nms of Paddle') #convert the paddle attribute to onnx tensor name_score_threshold = [outputs['Out'][0] + "@score_threshold"] @@ -353,7 +352,8 @@ def multiclass_nms(op, block): outputs_gather_topk_class = [result_name + "@gather_topk_class"] node_gather_topk_class = onnx.helper.make_node( 'Gather', - inputs=outputs_gather_1_nonzero + [outputs_topk_select_topk_indices[1]], + inputs=outputs_gather_1_nonzero + + [outputs_topk_select_topk_indices[1]], outputs=outputs_gather_topk_class, axis=1) node_list.append(node_gather_topk_class) @@ -362,7 +362,8 @@ def multiclass_nms(op, block): outputs_gather_topk_boxes_id = [result_name + "@gather_topk_boxes_id"] node_gather_topk_boxes_id = onnx.helper.make_node( 'Gather', - inputs=outputs_gather_2_nonzero + [outputs_topk_select_topk_indices[1]], + inputs=outputs_gather_2_nonzero + + [outputs_topk_select_topk_indices[1]], outputs=outputs_gather_topk_boxes_id, axis=1) node_list.append(node_gather_topk_boxes_id) diff --git a/x2paddle/op_mapper/paddle_custom_layer/yolo_box.py b/x2paddle/op_mapper/paddle_custom_layer/yolo_box.py index 50e048c4aaadcedf9ef566d39487bdb11a32f5d1..a1e49e77b32ce1d64c11ca35ac69ed6cb20ee51c 100644 --- a/x2paddle/op_mapper/paddle_custom_layer/yolo_box.py +++ b/x2paddle/op_mapper/paddle_custom_layer/yolo_box.py @@ -4,8 +4,6 @@ from onnx import onnx_pb, helper def get_old_name(arg, name_prefix=''): - """Get the old rame for a possible renamed argument - """ prefix_index = arg.find(name_prefix) if prefix_index != -1: @@ -40,8 +38,8 @@ def yolo_box(op, block): downsample_ratio = attrs['downsample_ratio'] input_size = input_height * downsample_ratio conf_thresh = attrs['conf_thresh'] - conf_thresh_mat = np.ones([num_anchors * input_height * input_width - ]) * conf_thresh + conf_thresh_mat = np.ones([num_anchors * input_height * + input_width]) * conf_thresh node_list = [] im_outputs = [] diff --git a/x2paddle/op_mapper/paddle_op_mapper.py b/x2paddle/op_mapper/paddle_op_mapper.py index e38555e91eb2e65959e3b0bf76934078a682cb72..0ba7ad682528b4062dea381964835271f0177432 100644 --- a/x2paddle/op_mapper/paddle_op_mapper.py +++ b/x2paddle/op_mapper/paddle_op_mapper.py @@ -250,8 +250,7 @@ class PaddleOpMapper(object): node = helper.make_node( pool_type[op.attr('pooling_type')][1], inputs=op.input('X'), - outputs=op.output('Out'), - ) + outputs=op.output('Out'), ) else: input_shape = block.var(op.input('X')[0]).shape k_size = op.attr('ksize') @@ -407,8 +406,7 @@ class PaddleOpMapper(object): node = helper.make_node( 'Clip', inputs=[op.input('X')[0], min_name, max_name], - outputs=op.output('Out'), - ) + outputs=op.output('Out'), ) return [min_node, max_node, node] def shape(self, op, block): @@ -450,8 +448,7 @@ class PaddleOpMapper(object): node = helper.make_node( "Slice", inputs=[op.input('Input')[0], starts_name, ends_name, axes_name], - outputs=op.output('Out'), - ) + outputs=op.output('Out'), ) return [starts_node, ends_node, axes_node, node] def fill_constant(self, op, block): @@ -551,8 +548,8 @@ class PaddleOpMapper(object): if op.attr('align_corners'): coordinate_transformation_mode = 'align_corners' if ('OutSize' in input_names and len(op.input('OutSize')) > 0) or ( - 'SizeTensor' in input_names - and len(op.input('SizeTensor')) > 0): + 'SizeTensor' in input_names and + len(op.input('SizeTensor')) > 0): node_list = list() roi_node = self.make_constant_node( self.get_name(op.type, 'roi'), onnx_pb.TensorProto.FLOAT, @@ -631,8 +628,7 @@ class PaddleOpMapper(object): elif 'Scale' in input_names and len(op.input('Scale')) > 0: node = helper.make_node( 'Resize', - inputs=[op.input('X')[0], - op.input('Scale')[0]], + inputs=[op.input('X')[0], op.input('Scale')[0]], outputs=op.output('Out'), mode='linear', coordinate_transformation_mode=coordinate_transformation_mode) @@ -641,8 +637,9 @@ class PaddleOpMapper(object): scale = op.attr('scale') if out_shape.count(-1) > 0: scale_name = self.get_name(op.type, 'scale') - scale_node = self.make_constant_node( - scale_name, onnx_pb.TensorProto.FLOAT, [1, 1, scale, scale]) + scale_node = self.make_constant_node(scale_name, + onnx_pb.TensorProto.FLOAT, + [1, 1, scale, scale]) roi_name = self.get_name(op.type, 'roi') roi_node = self.make_constant_node(roi_name, onnx_pb.TensorProto.FLOAT, @@ -667,16 +664,14 @@ class PaddleOpMapper(object): if 'OutSize' in input_names and len(op.input('OutSize')) > 0: node = helper.make_node( 'Resize', - inputs=[op.input('X')[0], '', - op.input('OutSize')[0]], + inputs=[op.input('X')[0], '', op.input('OutSize')[0]], outputs=op.output('Out'), mode='nearest', coordinate_transformation_mode=coordinate_transformation_mode) elif 'Scale' in input_names and len(op.input('Scale')) > 0: node = helper.make_node( 'Resize', - inputs=[op.input('X')[0], - op.input('Scale')[0]], + inputs=[op.input('X')[0], op.input('Scale')[0]], outputs=op.output('Out'), mode='nearest', coordinate_transformation_mode=coordinate_transformation_mode) @@ -685,8 +680,9 @@ class PaddleOpMapper(object): scale = op.attr('scale') if out_shape.count(-1) > 0: scale_name = self.get_name(op.type, 'scale') - scale_node = self.make_constant_node( - scale_name, onnx_pb.TensorProto.FLOAT, [1, 1, scale, scale]) + scale_node = self.make_constant_node(scale_name, + onnx_pb.TensorProto.FLOAT, + [1, 1, scale, scale]) roi_name = self.get_name(op.type, 'roi') roi_node = self.make_constant_node(roi_name, onnx_pb.TensorProto.FLOAT, @@ -737,8 +733,7 @@ class PaddleOpMapper(object): node1 = helper.make_node( 'Clip', inputs=[name0, min_name, max_name], - outputs=[name1], - ) + outputs=[name1], ) name2 = self.get_name(op.type, 'mul') node2 = helper.make_node( 'Mul', inputs=[op.input('X')[0], name1], outputs=[name2]) @@ -814,14 +809,6 @@ class PaddleOpMapper(object): keepdims=0) return node - def yolo_box(self, op, block): - from .paddle_custom_layer.yolo_box import yolo_box - return yolo_box(op, block) - - def multiclass_nms(self, op, block): - from .paddle_custom_layer.multiclass_nms import multiclass_nms - return multiclass_nms(op, block) - def reciprocal(self, op, block): inputs = op.input(op.input_names[0]) outputs = op.output(op.output_names[0]) diff --git a/x2paddle/op_mapper/tf_op_mapper.py b/x2paddle/op_mapper/tf_op_mapper.py index aba5f950081bc7ada25b5dd2fdeca4e70f2da714..3a88f47330a7d63cff02358aa476d5216c183e9f 100644 --- a/x2paddle/op_mapper/tf_op_mapper.py +++ b/x2paddle/op_mapper/tf_op_mapper.py @@ -114,9 +114,8 @@ class TFOpMapper(OpMapper): else: unsupported_ops.add(op) if len(unsupported_ops) > 0: - sys.stderr.write( - "=========={} Ops are not supported yet======\n".format( - len(unsupported_ops))) + sys.stderr.write("=========={} Ops are not supported yet======\n". + format(len(unsupported_ops))) for op in unsupported_ops: sys.stderr.write("========== {} ==========\n".format(op)) sys.exit(-1) @@ -141,10 +140,8 @@ class TFOpMapper(OpMapper): pd_param_name = list(param.values())[0] tf_param = node.get_attr(tf_param_name) attr[pd_param_name] = tf_param - node.fluid_code.add_layer(op_info[0], - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + op_info[0], inputs=input, output=node, param_attr=attr) def elementwise_map(self, node): assert node.layer_type in self.elementwise_ops @@ -179,21 +176,21 @@ class TFOpMapper(OpMapper): 0] == y_shape[-1] and y_shape.count(-1) < 1: shape = [1, x_shape[0], 1, 1] attr = {"shape": shape} - node.fluid_code.add_layer("reshape", - inputs=x_input, - output="reshape_x", - param_attr=attr) + node.fluid_code.add_layer( + "reshape", + inputs=x_input, + output="reshape_x", + param_attr=attr) if y_shape[0] != 1: attr = {"expand_times": [y_shape[0], 1, 1, 1]} - node.fluid_code.add_layer("expand", - inputs="reshape_x", - output="reshape_x", - param_attr=attr) + node.fluid_code.add_layer( + "expand", + inputs="reshape_x", + output="reshape_x", + param_attr=attr) inputs = {"x": "reshape_x", "y": y_input} - node.fluid_code.add_layer(op_type, - inputs=inputs, - output=node, - param_attr=None) + node.fluid_code.add_layer( + op_type, inputs=inputs, output=node, param_attr=None) return else: raise Exception("Unexpected situation happend") @@ -205,10 +202,8 @@ class TFOpMapper(OpMapper): axis = -1 attr = {"axis": axis} inputs = {"x": x_input, "y": y_input} - node.fluid_code.add_layer(op_type, - inputs=inputs, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + op_type, inputs=inputs, output=node, param_attr=attr) return is_sub_seq = True @@ -242,10 +237,8 @@ class TFOpMapper(OpMapper): if len(x_expand_times) == 4 and x.tf_data_format == "NHWC": x_expand_times = [x_expand_times[i] for i in [0, 3, 1, 2]] attr = {"expand_times": x_expand_times} - node.fluid_code.add_layer("expand", - inputs=x_input, - output="x_tmp", - param_attr=attr) + node.fluid_code.add_layer( + "expand", inputs=x_input, output="x_tmp", param_attr=attr) x_input = "x_tmp" if y_need_expand: if len(y_expand_times) == 3 and y.tf_data_format == "NHWC": @@ -253,16 +246,12 @@ class TFOpMapper(OpMapper): if len(y_expand_times) == 4 and y.tf_data_format == "NHWC": y_expand_times = [y_expand_times[i] for i in [0, 3, 1, 2]] attr = {"expand_times": y_expand_times} - node.fluid_code.add_layer("expand", - inputs=y_input, - output="y_tmp", - param_attr=attr) + node.fluid_code.add_layer( + "expand", inputs=y_input, output="y_tmp", param_attr=attr) y_input = "y_tmp" inputs = {"x": x_input, "y": y_input} - node.fluid_code.add_layer(op_type, - inputs=inputs, - output=node, - param_attr=None) + node.fluid_code.add_layer( + op_type, inputs=inputs, output=node, param_attr=None) def Placeholder(self, node): shape = node.out_shapes[0] @@ -283,10 +272,8 @@ class TFOpMapper(OpMapper): if shape[0] < 0: self.batch_node = node - node.fluid_code.add_layer("data", - inputs=None, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "data", inputs=None, output=node, param_attr=attr) def OneShotIterator(self, node): return self.Placeholder(node) @@ -308,8 +295,8 @@ class TFOpMapper(OpMapper): shape = [shape[i] for i in [0, 3, 1, 2]] if len(shape) == 3: shape = [shape[i] for i in [2, 0, 1]] - self.weights[node.layer_name] = numpy.transpose( - node.value, (2, 0, 1)) + self.weights[node.layer_name] = numpy.transpose(node.value, + (2, 0, 1)) elif node.tf_data_format == "NCHW": if len(shape) == 4: self.graph.data_format_propagation(node) @@ -320,10 +307,8 @@ class TFOpMapper(OpMapper): 'name': string(node.layer_name), 'default_initializer': initializer } - node.fluid_code.add_layer("create_parameter", - inputs=None, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "create_parameter", inputs=None, output=node, param_attr=attr) def Transpose(self, node): input = self.graph.get_node(node.layer.input[0], copy=True) @@ -362,16 +347,12 @@ class TFOpMapper(OpMapper): node.tf_data_format = [tf_data_format[i] for i in perm] node.pd_data_format = [pd_data_format[i] for i in perm] attr = {'perm': new_perm} - node.fluid_code.add_layer("transpose", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "transpose", inputs=input, output=node, param_attr=attr) elif len(node.out_shapes[0]) != 4: attr = {'perm': perm} - node.fluid_code.add_layer("transpose", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "transpose", inputs=input, output=node, param_attr=attr) else: raise Exception("Unexpected situation happend in Transpose OP") @@ -401,10 +382,8 @@ class TFOpMapper(OpMapper): "pool_padding": string(pad_mode), "pool_stride": strides[2:4] } - node.fluid_code.add_layer("pool2d", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "pool2d", inputs=input, output=node, param_attr=attr) def Conv2D(self, node): input = self.graph.get_node(node.layer.input[0], copy=True) @@ -444,10 +423,8 @@ class TFOpMapper(OpMapper): "dilation": dilations[2:4], "padding": string(pad_mode) } - node.fluid_code.add_layer("conv2d", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "conv2d", inputs=input, output=node, param_attr=attr) def BiasAdd(self, node): input = self.graph.get_node(node.layer.input[0], copy=True) @@ -457,10 +434,8 @@ class TFOpMapper(OpMapper): axis = 1 inputs = {"x": input, "y": bias} attr = {"axis": axis} - node.fluid_code.add_layer("elementwise_add", - inputs=inputs, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "elementwise_add", inputs=inputs, output=node, param_attr=attr) def FusedBatchNorm(self, node): input = self.graph.get_node(node.layer.input[0], copy=True) @@ -491,10 +466,8 @@ class TFOpMapper(OpMapper): "is_test": True } - node.fluid_code.add_layer("batch_norm", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "batch_norm", inputs=input, output=node, param_attr=attr) def FusedBatchNormV3(self, node): return self.FusedBatchNorm(node) @@ -539,10 +512,8 @@ class TFOpMapper(OpMapper): "use_cudnn": False, "padding": string(pad_mode) } - node.fluid_code.add_layer("conv2d", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "conv2d", inputs=input, output=node, param_attr=attr) def Reshape(self, node): input = self.graph.get_node(node.layer.input[0], copy=True) @@ -562,18 +533,17 @@ class TFOpMapper(OpMapper): attr = {"shape": shape} self.add_omit_nodes(param.layer_name, node.layer_name) else: - assert len(param.out_shapes[0] - ) == 1, "Unexpected situation of shape parameter" + assert len(param.out_shapes[ + 0]) == 1, "Unexpected situation of shape parameter" attr = {"shape": [-1]} - node.fluid_code.add_layer("reshape", - inputs=param, - output="shape_param", - param_attr=attr) + node.fluid_code.add_layer( + "reshape", + inputs=param, + output="shape_param", + param_attr=attr) attr = {"num_or_sections": param.out_shapes[0][0], "dim": 0} - node.fluid_code.add_layer("split", - inputs="shape_param", - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "split", inputs="shape_param", output=node, param_attr=attr) new_param = "[" for i in range(param.out_shapes[0][0]): new_param += (node.layer_name + "[{}]".format(i) + ", ") @@ -601,14 +571,10 @@ class TFOpMapper(OpMapper): if len(input.out_shapes[0]) == 4 and node.tf_data_format == "NHWC": if len(attr["shape"]) < 3: perm = {"perm": [0, 2, 3, 1]} - node.fluid_code.add_layer("transpose", - inputs=input, - output=node, - param_attr=perm) - node.fluid_code.add_layer("reshape", - inputs=node, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "transpose", inputs=input, output=node, param_attr=perm) + node.fluid_code.add_layer( + "reshape", inputs=node, output=node, param_attr=attr) return if len(attr["shape"]) == 4 and node.tf_data_format == "NHWC": @@ -617,27 +583,19 @@ class TFOpMapper(OpMapper): attr["shape"] = [attr["shape"][i] for i in [0, 3, 1, 2]] else: perm = {"perm": [0, 2, 3, 1]} - node.fluid_code.add_layer("transpose", - inputs=input, - output=node, - param_attr=perm) - node.fluid_code.add_layer("reshape", - inputs=node, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "transpose", inputs=input, output=node, param_attr=perm) + node.fluid_code.add_layer( + "reshape", inputs=node, output=node, param_attr=attr) perm = {"perm": [0, 3, 1, 2]} - node.fluid_code.add_layer("transpose", - inputs=node, - output=node, - param_attr=perm) + node.fluid_code.add_layer( + "transpose", inputs=node, output=node, param_attr=perm) return if len(attr["shape"]) == 5: attr["shape"] = [attr["shape"][i] for i in [0, 1, 4, 2, 3]] - node.fluid_code.add_layer("reshape", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "reshape", inputs=input, output=node, param_attr=attr) def AvgPool(self, node): input = self.graph.get_node(node.layer.input[0], copy=True) @@ -665,10 +623,8 @@ class TFOpMapper(OpMapper): "pool_stride": strides[2:4], "pool_padding": string(pad_mode) } - node.fluid_code.add_layer("pool2d", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "pool2d", inputs=input, output=node, param_attr=attr) def SplitV(self, node): input = self.graph.get_node(node.layer.input[0], copy=True) @@ -685,28 +641,24 @@ class TFOpMapper(OpMapper): "num_or_sections": num_sections.value.tolist(), "dim": dim.value } - node.fluid_code.add_layer("split", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "split", inputs=input, output=node, param_attr=attr) def ConcatV2(self, node): inputs = [ - self.graph.get_node(name, copy=True) - for name in node.layer.input[:-1] + self.graph.get_node( + name, copy=True) for name in node.layer.input[:-1] ] axis = self.graph.get_node(node.layer.input[-1], copy=True) assert axis.layer_type == "Const" self.add_omit_nodes(axis.layer_name, node.layer_name) axis = axis.value - if inputs[0].tf_data_format == "NHWC" and len( - inputs[0].out_shapes[0]) == 4: + if inputs[0].tf_data_format == "NHWC" and len(inputs[0].out_shapes[ + 0]) == 4: axis = nhwc_dim_to_nchw(inputs[0], axis) attr = {"axis": axis} - node.fluid_code.add_layer("concat", - inputs=inputs, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "concat", inputs=inputs, output=node, param_attr=attr) def Tile(self, node): input = self.graph.get_node(node.layer.input[0], copy=True) @@ -726,18 +678,17 @@ class TFOpMapper(OpMapper): expand_times[i] = 1 attr = {"expand_times": expand_times} - node.fluid_code.add_layer("expand", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "expand", inputs=input, output=node, param_attr=attr) def Pack(self, node): inputs = [ - self.graph.get_node(name, copy=True) for name in node.layer.input + self.graph.get_node( + name, copy=True) for name in node.layer.input ] axis = node.get_attr("axis") - if inputs[0].tf_data_format == "NHWC" and len( - inputs[0].out_shapes[0]) == 4: + if inputs[0].tf_data_format == "NHWC" and len(inputs[0].out_shapes[ + 0]) == 4: tf_data_format = list(inputs[0].tf_data_format) tf_data_format.insert(axis, str(len(tf_data_format))) axis = nhwc_dim_to_nchw(inputs[0], axis) @@ -747,10 +698,8 @@ class TFOpMapper(OpMapper): node.pd_data_format = "".join(pd_data_format) attr = {"axis": axis} - node.fluid_code.add_layer("stack", - inputs=inputs, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "stack", inputs=inputs, output=node, param_attr=attr) def Pad(self, node): input = self.graph.get_node(node.layer.input[0], copy=True) @@ -767,10 +716,8 @@ class TFOpMapper(OpMapper): paddings = paddings[4:] pad_op = "pad2d" attr = {"paddings": paddings} - node.fluid_code.add_layer(pad_op, - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + pad_op, inputs=input, output=node, param_attr=attr) def MirrorPad(self, node): input = self.graph.get_node(node.layer.input[0], copy=True) @@ -789,10 +736,8 @@ class TFOpMapper(OpMapper): paddings = paddings[4:] pad_op = "pad2d" attr = {"paddings": paddings, "mode": string("reflect")} - node.fluid_code.add_layer(pad_op, - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + pad_op, inputs=input, output=node, param_attr=attr) def Range(self, node): start = self.graph.get_node(node.layer.input[0], copy=True) @@ -816,10 +761,8 @@ class TFOpMapper(OpMapper): inputs = {"start": start, "end": limit, "step": delta} attr = {"dtype": string(node.dtype)} - node.fluid_code.add_layer("range", - inputs=inputs, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "range", inputs=inputs, output=node, param_attr=attr) def Mean(self, node): input = self.graph.get_node(node.layer.input[0], copy=True) @@ -833,10 +776,8 @@ class TFOpMapper(OpMapper): dims[i] = nhwc_dim_to_nchw(input, dims[i]) attr = {"dim": dims, "keep_dim": keep_dims} - node.fluid_code.add_layer("reduce_mean", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "reduce_mean", inputs=input, output=node, param_attr=attr) def MatMul(self, node): x = self.graph.get_node(node.layer.input[0], copy=True) @@ -850,15 +791,11 @@ class TFOpMapper(OpMapper): shape = x.out_shapes[0] shape[-1] = y.out_shapes[0][0] attr = {"shape": shape} - node.fluid_code.add_layer("reshape", - inputs=x, - output=x, - param_attr=attr) + node.fluid_code.add_layer( + "reshape", inputs=x, output=x, param_attr=attr) attr = {"transpose_x": transpose_a, "transpose_y": transpose_b} - node.fluid_code.add_layer("matmul", - inputs=inputs, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "matmul", inputs=inputs, output=node, param_attr=attr) def ArgMax(self, node): input = self.graph.get_node(node.layer.input[0], copy=True) @@ -869,10 +806,8 @@ class TFOpMapper(OpMapper): if input.tf_data_format == "NHWC" and len(input.out_shapes[0]) == 4: axis = nhwc_dim_to_nchw(input, axis) attr = {"axis": axis} - node.fluid_code.add_layer("argmax", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "argmax", inputs=input, output=node, param_attr=attr) def StridedSlice(self, node): input = self.graph.get_node(node.layer.input[0], copy=True) @@ -910,16 +845,12 @@ class TFOpMapper(OpMapper): x = shrink_axis_mask >> i & 1 if x == 1: squeeze_dims.append(i) - node.fluid_code.add_layer("slice", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "slice", inputs=input, output=node, param_attr=attr) if shrink_axis_mask > 0 and len(input.out_shapes[0]) == 5: attr = {"axes": squeeze_dims} - node.fluid_code.add_layer("squeeze", - inputs=node, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "squeeze", inputs=node, output=node, param_attr=attr) def Slice(self, node): input = self.graph.get_node(node.layer.input[0], copy=True) @@ -951,10 +882,8 @@ class TFOpMapper(OpMapper): "starts": begin, "ends": size } - node.fluid_code.add_layer("slice", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "slice", inputs=input, output=node, param_attr=attr) def Conv2DBackpropInput(self, node): out_shape = self.graph.get_node(node.layer.input[0], copy=True) @@ -1004,10 +933,8 @@ class TFOpMapper(OpMapper): "padding": string(pad_mode), "output_size": out_shape[1:3] } - node.fluid_code.add_layer("conv2d_transpose", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "conv2d_transpose", inputs=input, output=node, param_attr=attr) def Max(self, node): input = self.graph.get_node(node.layer.input[0], copy=True) @@ -1019,10 +946,8 @@ class TFOpMapper(OpMapper): dim = nhwc_dim_to_nchw(input, dim) attr = {"dim": dim, "keep_dim": keep_dims} - node.fluid_code.add_layer("reduce_max", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "reduce_max", inputs=input, output=node, param_attr=attr) def Sum(self, node): input = self.graph.get_node(node.layer.input[0], copy=True) @@ -1034,19 +959,15 @@ class TFOpMapper(OpMapper): dim = nhwc_dim_to_nchw(input, dim) attr = {"dim": dim, "keep_dim": keep_dims} - node.fluid_code.add_layer("reduce_sum", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "reduce_sum", inputs=input, output=node, param_attr=attr) def Cast(self, node): input = self.graph.get_node(node.layer.input[0], copy=True) dtype = node.dtype_map[node.get_attr('DstT')] attr = {"dtype": string(dtype)} - node.fluid_code.add_layer("cast", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "cast", inputs=input, output=node, param_attr=attr) def Split(self, node): dim = self.graph.get_node(node.layer.input[0], copy=True) @@ -1058,10 +979,8 @@ class TFOpMapper(OpMapper): dim = nhwc_dim_to_nchw(input, dim) attr = {"num_or_sections": num_split, "dim": dim} - node.fluid_code.add_layer("split", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "split", inputs=input, output=node, param_attr=attr) def Squeeze(self, node): input = self.graph.get_node(node.layer.input[0], copy=True) @@ -1070,10 +989,8 @@ class TFOpMapper(OpMapper): for i in range(len(squeeze_dims)): squeeze_dims[i] = nhwc_dim_to_nchw(input, squeeze_dims[i]) attr = {"axes": squeeze_dims} - node.fluid_code.add_layer("squeeze", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "squeeze", inputs=input, output=node, param_attr=attr) def Softmax(self, node): input = self.graph.get_node(node.layer.input[0], copy=True) @@ -1083,10 +1000,8 @@ class TFOpMapper(OpMapper): if input.tf_data_format == "NHWC" and len(input.out_shapes[0]) == 4: axis = nhwc_dim_to_nchw(input, axis) attr = {"axis": axis} - node.fluid_code.add_layer("softmax", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "softmax", inputs=input, output=node, param_attr=attr) def ResizeNearestNeighbor(self, node): input = self.graph.get_node(node.layer.input[0], copy=True) @@ -1095,14 +1010,12 @@ class TFOpMapper(OpMapper): if resize_shape.layer_type == "Const": resize_shape = resize_shape.value.tolist() else: - resize_shape = self.decoder.infer_shape_tensor( - resize_shape, node.out_shapes[0]) + resize_shape = self.decoder.infer_shape_tensor(resize_shape, + node.out_shapes[0]) align_corners = node.get_attr("align_corners") attr = {"align_corners": align_corners, "out_shape": resize_shape} - node.fluid_code.add_layer("resize_nearest", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "resize_nearest", inputs=input, output=node, param_attr=attr) def ResizeBilinear(self, node): input = self.graph.get_node(node.layer.input[0], copy=True) @@ -1111,27 +1024,23 @@ class TFOpMapper(OpMapper): if resize_shape.layer_type == "Const": resize_shape = resize_shape.value.tolist() else: - resize_shape = self.decoder.infer_shape_tensor( - resize_shape, node.out_shapes[0]) + resize_shape = self.decoder.infer_shape_tensor(resize_shape, + node.out_shapes[0]) align_corners = node.get_attr("align_corners") attr = { "align_corners": align_corners, "out_shape": resize_shape, "align_mode": 1 } - node.fluid_code.add_layer("resize_bilinear", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "resize_bilinear", inputs=input, output=node, param_attr=attr) def GreaterEqual(self, node): x = self.graph.get_node(node.layer.input[0], copy=True) y = self.graph.get_node(node.layer.input[1], copy=True) inputs = {"x": x, "y": y} - node.fluid_code.add_layer("greater_equal", - inputs=inputs, - output=node, - param_attr=None) + node.fluid_code.add_layer( + "greater_equal", inputs=inputs, output=node, param_attr=None) def RandomUniform(self, node): shape = self.graph.get_node(node.layer.input[0], copy=True) @@ -1145,26 +1054,21 @@ class TFOpMapper(OpMapper): attr = {"shape": shape, "min": 0.0, "max": 0.9999} if shape[0] < 0: input = self.batch_node - node.fluid_code.add_layer("uniform_random_batch_size_like", - inputs=input, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "uniform_random_batch_size_like", + inputs=input, + output=node, + param_attr=attr) else: - node.fluid_code.add_layer("uniform_random", - inputs=None, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "uniform_random", inputs=None, output=node, param_attr=attr) def SquaredDifference(self, node): x = self.graph.get_node(node.layer.input[0], copy=True) y = self.graph.get_node(node.layer.input[1], copy=True) inputs = {"x": x, "y": y} - node.fluid_code.add_layer("elementwise_sub", - inputs=inputs, - output=node, - param_attr=None) + node.fluid_code.add_layer( + "elementwise_sub", inputs=inputs, output=node, param_attr=None) inputs = {"x": node, "y": node} - node.fluid_code.add_layer("elementwise_mul", - inputs=inputs, - output=node, - param_attr=None) + node.fluid_code.add_layer( + "elementwise_mul", inputs=inputs, output=node, param_attr=None) diff --git a/x2paddle/op_mapper/tf_op_mapper_nhwc.py b/x2paddle/op_mapper/tf_op_mapper_nhwc.py index 05a06ac91247b11111200c1406a5466a616f4847..6f36c8dbaaccdc03ea3c482a1a011bd3a6b275ee 100644 --- a/x2paddle/op_mapper/tf_op_mapper_nhwc.py +++ b/x2paddle/op_mapper/tf_op_mapper_nhwc.py @@ -486,8 +486,8 @@ class TFOpMapperNHWC(OpMapper): attr = {"shape": shape} self.add_omit_nodes(param.layer_name, node.layer_name) else: - assert len(param.out_shapes[0] - ) == 1, "Unexpected situation of shape parameter" + assert len(param.out_shapes[ + 0]) == 1, "Unexpected situation of shape parameter" attr = {"shape": [-1]} node.fluid_code.add_layer( "reshape", @@ -577,8 +577,8 @@ class TFOpMapperNHWC(OpMapper): def ConcatV2(self, node): inputs = [ - self.graph.get_node(name, copy=True) - for name in node.layer.input[:-1] + self.graph.get_node( + name, copy=True) for name in node.layer.input[:-1] ] axis = self.graph.get_node(node.layer.input[-1], copy=True) assert axis.layer_type == "Const" @@ -608,7 +608,8 @@ class TFOpMapperNHWC(OpMapper): def Pack(self, node): inputs = [ - self.graph.get_node(name, copy=True) for name in node.layer.input + self.graph.get_node( + name, copy=True) for name in node.layer.input ] axis = node.get_attr("axis") attr = {"axis": axis} @@ -949,8 +950,8 @@ class TFOpMapperNHWC(OpMapper): if resize_shape.layer_type == "Const": resize_shape = resize_shape.value.tolist() else: - resize_shape = self.decoder.infer_shape_tensor( - resize_shape, node.out_shapes[0]) + resize_shape = self.decoder.infer_shape_tensor(resize_shape, + node.out_shapes[0]) align_corners = node.get_attr("align_corners") attr = {"perm": [0, 3, 1, 2]} node.fluid_code.add_layer( @@ -969,8 +970,8 @@ class TFOpMapperNHWC(OpMapper): if resize_shape.layer_type == "Const": resize_shape = resize_shape.value.tolist() else: - resize_shape = self.decoder.infer_shape_tensor( - resize_shape, node.out_shapes[0]) + resize_shape = self.decoder.infer_shape_tensor(resize_shape, + node.out_shapes[0]) align_corners = node.get_attr("align_corners") attr = {"perm": [0, 3, 1, 2]} node.fluid_code.add_layer( diff --git a/x2paddle/optimizer/caffe_optimizer.py b/x2paddle/optimizer/caffe_optimizer.py index 52786ce5b3c03e5eda5e830262e96bf388f37e76..a9f373ae6ed6affddec3e8377073ae58c6eec38d 100644 --- a/x2paddle/optimizer/caffe_optimizer.py +++ b/x2paddle/optimizer/caffe_optimizer.py @@ -41,10 +41,11 @@ class CaffeOptimizer(object): if is_delete_node: parent_node.fluid_code.clear() node.fluid_code.clear() - node.fluid_code.add_layer("batch_norm", - inputs=input, - output=node, - param_attr=parent_param_attr) + node.fluid_code.add_layer( + "batch_norm", + inputs=input, + output=node, + param_attr=parent_param_attr) def merge_op_activation(self): for node_name in self.graph.topo_sort: @@ -62,7 +63,8 @@ class CaffeOptimizer(object): if is_delete_node: parent_node.fluid_code.clear() node.fluid_code.clear() - node.fluid_code.add_layer(op, - inputs=input, - output=node, - param_attr=parent_param_attr) + node.fluid_code.add_layer( + op, + inputs=input, + output=node, + param_attr=parent_param_attr) diff --git a/x2paddle/optimizer/tf_optimizer.py b/x2paddle/optimizer/tf_optimizer.py index 6431e439ccf463a1aa2c08ba1b1eac3f34d41175..8b63df18017295aefb25155d85e0c1595edec88a 100644 --- a/x2paddle/optimizer/tf_optimizer.py +++ b/x2paddle/optimizer/tf_optimizer.py @@ -554,10 +554,11 @@ class TFOptimizer(object): node.fluid_code.layers[0].param_attr["shape"] = shape node.fluid_code.layers[0].output = "nhwc_" + name attr = {"perm": [0, 2, 3, 1]} - node.fluid_code.add_layer("transpose", - inputs="nhwc_" + name, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "transpose", + inputs="nhwc_" + name, + output=node, + param_attr=attr) self.graph.input_nodes[i] = "nhwc_" + name for i, name in enumerate(self.graph.output_nodes): node = self.graph.get_node(name) @@ -767,8 +768,8 @@ class TFOptimizer(object): is_prelu = False continue - if len(in_nodes0[0].outputs) != 1 or len( - in_nodes0[1].outputs) != 1: + if len(in_nodes0[0].outputs) != 1 or len(in_nodes0[1] + .outputs) != 1: is_prelu = False continue @@ -777,8 +778,8 @@ class TFOptimizer(object): self.graph.get_node(in_name) for in_name in in_nodes0[1].inputs ] - if in_nodes2[1].layer_type != "Const" or numpy.fabs( - in_nodes2[1].value - 0.5) > 1e-06: + if in_nodes2[1].layer_type != "Const" or numpy.fabs(in_nodes2[ + 1].value - 0.5) > 1e-06: is_prelu = False continue if in_nodes2[0].layer_type != "Mul": @@ -787,8 +788,8 @@ class TFOptimizer(object): if exist_act(in_nodes2[0]): is_prelu = False continue - if len(in_nodes2[1].outputs) != 1 or len( - in_nodes2[0].outputs) != 1: + if len(in_nodes2[1].outputs) != 1 or len(in_nodes2[0] + .outputs) != 1: is_prelu = False continue @@ -803,8 +804,8 @@ class TFOptimizer(object): if exist_act(in_nodes3[1]): is_prelu = False continue - if len(in_nodes3[0].outputs) != 1 or len( - in_nodes3[1].outputs) != 1: + if len(in_nodes3[0].outputs) != 1 or len(in_nodes3[1] + .outputs) != 1: is_prelu = False continue @@ -856,12 +857,12 @@ class TFOptimizer(object): mode = "element" elif len(in_nodes3[0].value.shape) == 0: mode = "all" - elif len(in_nodes3[0].value.shape - ) == 1 and in_nodes3[0].value.shape[0] == 1: + elif len(in_nodes3[0].value.shape) == 1 and in_nodes3[ + 0].value.shape[0] == 1: mode = "all" - elif len(in_shape) == 4 and len( - in_nodes3[0].value.shape - ) == 1 and in_nodes3[0].value.shape[0] == in_shape[-1]: + elif len(in_shape) == 4 and len(in_nodes3[ + 0].value.shape) == 1 and in_nodes3[0].value.shape[ + 0] == in_shape[-1]: mode = "channel" weight = self.op_mapper.weights[in_nodes3[0].layer_name] weight = numpy.expand_dims(weight, 0) @@ -916,14 +917,15 @@ class TFOptimizer(object): self.graph.get_node(in_name) for in_name in node.inputs ] if in_nodes0[0].layer_type != "Mul" or in_nodes0[ - 1].layer_type != "Const" or in_nodes0[1].value.size != 1: + 1].layer_type != "Const" or in_nodes0[ + 1].value.size != 1: is_scale = False continue if exist_act(in_nodes0[0]): is_scale = False continue - if len(in_nodes0[0].outputs) != 1 or len( - in_nodes0[1].outputs) != 1: + if len(in_nodes0[0].outputs) != 1 or len(in_nodes0[1] + .outputs) != 1: is_scale = False continue @@ -939,8 +941,8 @@ class TFOptimizer(object): if exist_act(in_nodes1[1]): is_scale = False continue - if len(in_nodes1[0].outputs) != 1 or len( - in_nodes1[1].outputs) != 1: + if len(in_nodes1[0].outputs) != 1 or len(in_nodes1[1] + .outputs) != 1: is_scale = False continue @@ -962,8 +964,8 @@ class TFOptimizer(object): scale = 1.0 / in_nodes2[1].value * in_nodes1[0].value act = None if node.fluid_code.layers[0].param_attr is not None: - act = node.fluid_code.layers[0].param_attr.get( - "act", None) + act = node.fluid_code.layers[0].param_attr.get("act", + None) node.fluid_code.clear() attr = { @@ -972,10 +974,8 @@ class TFOptimizer(object): "bias_after_scale": True, "act": act } - node.fluid_code.add_layer("scale", - inputs=in_node, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "scale", inputs=in_node, output=node, param_attr=attr) del self.graph.node_map[in_nodes0[0].layer_name] del self.graph.node_map[in_nodes0[1].layer_name] @@ -1004,17 +1004,17 @@ class TFOptimizer(object): if exist_act(in_nodes0[0]): is_affine_channel = False continue - if len(in_nodes0[0].outputs) != 1 or len( - in_nodes0[1].outputs) != 1: + if len(in_nodes0[0].outputs) != 1 or len(in_nodes0[1] + .outputs) != 1: is_affine_channel = False continue in_nodes1 = [ self.graph.get_node(in_name) for in_name in in_nodes0[0].inputs ] - if len(in_nodes1[0].out_shapes[0] - ) != 4 or in_nodes1[1].layer_type != "Const" or len( - in_nodes1[1].value.shape) != 3: + if len(in_nodes1[0].out_shapes[0]) != 4 or in_nodes1[ + 1].layer_type != "Const" or len(in_nodes1[1] + .value.shape) != 3: is_affine_channel = False continue if len(in_nodes1[1].outputs) != 1: @@ -1037,8 +1037,8 @@ class TFOptimizer(object): node.layer_type = "AffineChannel" node.inputs = [in_node.layer_name] scale = 1.0 / in_nodes0[1].value.flatten() - bias = in_nodes1[1].value.flatten( - ) / in_nodes0[1].value.flatten() + bias = in_nodes1[1].value.flatten() / in_nodes0[ + 1].value.flatten() if not bias_add: bias *= -1.0 self.op_mapper.weights[node.layer_name + "_scale"] = scale @@ -1046,8 +1046,8 @@ class TFOptimizer(object): act = None if node.fluid_code.layers[0].param_attr is not None: - act = node.fluid_code.layers[0].param_attr.get( - "act", None) + act = node.fluid_code.layers[0].param_attr.get("act", + None) node.fluid_code.clear() attr = { @@ -1055,29 +1055,32 @@ class TFOptimizer(object): "shape": [channel], "name": string(node.layer_name + "_scale") } - node.fluid_code.add_layer("create_parameter", - inputs=None, - output=node.layer_name + "_scale", - param_attr=attr) + node.fluid_code.add_layer( + "create_parameter", + inputs=None, + output=node.layer_name + "_scale", + param_attr=attr) attr = { "dtype": string(scale.dtype), "shape": [channel], "name": string(node.layer_name + "_bias") } - node.fluid_code.add_layer("create_parameter", - inputs=None, - output=node.layer_name + "_bias", - param_attr=attr) + node.fluid_code.add_layer( + "create_parameter", + inputs=None, + output=node.layer_name + "_bias", + param_attr=attr) inputs = { "x": in_node, "scale": node.layer_name + "_scale", "bias": node.layer_name + "_bias" } attr = {"act": act} - node.fluid_code.add_layer("affine_channel", - inputs=inputs, - output=node, - param_attr=attr) + node.fluid_code.add_layer( + "affine_channel", + inputs=inputs, + output=node, + param_attr=attr) del self.graph.node_map[in_nodes0[0].layer_name] del self.graph.node_map[in_nodes0[1].layer_name]