diff --git a/research/object_detection/environment.yml b/research/object_detection/environment.yml new file mode 100644 index 00000000000..e367309298e --- /dev/null +++ b/research/object_detection/environment.yml @@ -0,0 +1,20 @@ +# Create with `conda env create -f environment.yml` +name: object_detection + +channels: + - conda-forge + +dependencies: + - python=3.6 + - python + - pillow + - matplotlib + - cython + - protobuf + - jupyter + - jupyterlab + - pip: + - pycocotools + - tensorflow-gpu + # or + # - tensorflow \ No newline at end of file diff --git a/research/object_detection/g3doc/installation.md b/research/object_detection/g3doc/installation.md index 4bb682d88d9..3883ea5a1d6 100644 --- a/research/object_detection/g3doc/installation.md +++ b/research/object_detection/g3doc/installation.md @@ -1,112 +1,21 @@ # Installation -## Dependencies - -Tensorflow Object Detection API depends on the following libraries: - -* Protobuf 3.0.0 -* Python-tk -* Pillow 1.0 -* lxml -* tf Slim (which is included in the "tensorflow/models/research/" checkout) -* Jupyter notebook -* Matplotlib -* Tensorflow -* Cython -* contextlib2 -* cocoapi - -For detailed steps to install Tensorflow, follow the [Tensorflow installation -instructions](https://www.tensorflow.org/install/). A typical user can install -Tensorflow using one of the following commands: - -``` bash -# For CPU -pip install tensorflow -# For GPU -pip install tensorflow-gpu -``` - -The remaining libraries can be installed on Ubuntu 16.04 using via apt-get: +It is recommended to use the [Anaconda Python distribution](https://www.anaconda.com/downloa) to install and use the `object_detection` API. Then you can setup a new Conda environment with all the necessary dependencies: ``` bash -sudo apt-get install protobuf-compiler python-pil python-lxml python-tk -pip install --user Cython -pip install --user contextlib2 -pip install --user jupyter -pip install --user matplotlib -``` +wget https://raw.githubusercontent.com/hadim/models/master/research/object_detection/environment.yml +conda env create -f environment.yml -Alternatively, users can install dependencies using pip: +source activate object_detection +pip install object_detection -``` bash -pip install --user Cython -pip install --user contextlib2 -pip install --user pillow -pip install --user lxml -pip install --user jupyter -pip install --user matplotlib -``` - -Note that sometimes "sudo apt-get install protobuf-compiler" will install -Protobuf 3+ versions for you and some users have issues when using 3.5. -If that is your case, you're suggested to download and install Protobuf 3.0.0 -(available [here](https://github.com/google/protobuf/releases/tag/v3.0.0)). - -## COCO API installation - -Download the -cocoapi and -copy the pycocotools subfolder to the tensorflow/models/research directory if -you are interested in using COCO evaluation metrics. The default metrics are -based on those used in Pascal VOC evaluation. To use the COCO object detection -metrics add `metrics_set: "coco_detection_metrics"` to the `eval_config` message -in the config file. To use the COCO instance segmentation metrics add -`metrics_set: "coco_mask_metrics"` to the `eval_config` message in the config -file. - -```bash -git clone https://github.com/cocodataset/cocoapi.git -cd cocoapi/PythonAPI -make -cp -r pycocotools /models/research/ +# or to install from a local directory +# pip install -e . ``` -## Protobuf Compilation - -The Tensorflow Object Detection API uses Protobufs to configure model and -training parameters. Before the framework can be used, the Protobuf libraries -must be compiled. This should be done by running the following command from -the tensorflow/models/research/ directory: - - -``` bash -# From tensorflow/models/research/ -protoc object_detection/protos/*.proto --python_out=. -``` - -## Add Libraries to PYTHONPATH - -When running locally, the tensorflow/models/research/ and slim directories -should be appended to PYTHONPATH. This can be done by running the following from -tensorflow/models/research/: - - -``` bash -# From tensorflow/models/research/ -export PYTHONPATH=$PYTHONPATH:`pwd`:`pwd`/slim -``` - -Note: This command needs to run from every new terminal you start. If you wish -to avoid running this manually, you can add it as a new line to the end of your -~/.bashrc file, replacing \`pwd\` with the absolute path of -tensorflow/models/research on your system. - -# Testing the Installation - -You can test that you have correctly installed the Tensorflow Object Detection\ -API by running the following command: +Check the installation is correct by running tests: ```bash -python object_detection/builders/model_builder_test.py +source activate object_detection +model_builder_test ``` diff --git a/research/object_detection/__init__.py b/research/object_detection/object_detection/__init__.py similarity index 100% rename from research/object_detection/__init__.py rename to research/object_detection/object_detection/__init__.py diff --git a/research/object_detection/anchor_generators/__init__.py b/research/object_detection/object_detection/anchor_generators/__init__.py similarity index 100% rename from research/object_detection/anchor_generators/__init__.py rename to research/object_detection/object_detection/anchor_generators/__init__.py diff --git a/research/object_detection/anchor_generators/grid_anchor_generator.py b/research/object_detection/object_detection/anchor_generators/grid_anchor_generator.py similarity index 100% rename from research/object_detection/anchor_generators/grid_anchor_generator.py rename to research/object_detection/object_detection/anchor_generators/grid_anchor_generator.py diff --git a/research/object_detection/anchor_generators/grid_anchor_generator_test.py b/research/object_detection/object_detection/anchor_generators/grid_anchor_generator_test.py similarity index 100% rename from research/object_detection/anchor_generators/grid_anchor_generator_test.py rename to research/object_detection/object_detection/anchor_generators/grid_anchor_generator_test.py diff --git a/research/object_detection/anchor_generators/multiple_grid_anchor_generator.py b/research/object_detection/object_detection/anchor_generators/multiple_grid_anchor_generator.py similarity index 100% rename from research/object_detection/anchor_generators/multiple_grid_anchor_generator.py rename to research/object_detection/object_detection/anchor_generators/multiple_grid_anchor_generator.py diff --git a/research/object_detection/anchor_generators/multiple_grid_anchor_generator_test.py b/research/object_detection/object_detection/anchor_generators/multiple_grid_anchor_generator_test.py similarity index 100% rename from research/object_detection/anchor_generators/multiple_grid_anchor_generator_test.py rename to research/object_detection/object_detection/anchor_generators/multiple_grid_anchor_generator_test.py diff --git a/research/object_detection/anchor_generators/multiscale_grid_anchor_generator.py b/research/object_detection/object_detection/anchor_generators/multiscale_grid_anchor_generator.py similarity index 100% rename from research/object_detection/anchor_generators/multiscale_grid_anchor_generator.py rename to research/object_detection/object_detection/anchor_generators/multiscale_grid_anchor_generator.py diff --git a/research/object_detection/anchor_generators/multiscale_grid_anchor_generator_test.py b/research/object_detection/object_detection/anchor_generators/multiscale_grid_anchor_generator_test.py similarity index 100% rename from research/object_detection/anchor_generators/multiscale_grid_anchor_generator_test.py rename to research/object_detection/object_detection/anchor_generators/multiscale_grid_anchor_generator_test.py diff --git a/research/object_detection/box_coders/__init__.py b/research/object_detection/object_detection/box_coders/__init__.py similarity index 100% rename from research/object_detection/box_coders/__init__.py rename to research/object_detection/object_detection/box_coders/__init__.py diff --git a/research/object_detection/box_coders/faster_rcnn_box_coder.py b/research/object_detection/object_detection/box_coders/faster_rcnn_box_coder.py similarity index 100% rename from research/object_detection/box_coders/faster_rcnn_box_coder.py rename to research/object_detection/object_detection/box_coders/faster_rcnn_box_coder.py diff --git a/research/object_detection/box_coders/faster_rcnn_box_coder_test.py b/research/object_detection/object_detection/box_coders/faster_rcnn_box_coder_test.py similarity index 100% rename from research/object_detection/box_coders/faster_rcnn_box_coder_test.py rename to research/object_detection/object_detection/box_coders/faster_rcnn_box_coder_test.py diff --git a/research/object_detection/box_coders/keypoint_box_coder.py b/research/object_detection/object_detection/box_coders/keypoint_box_coder.py similarity index 100% rename from research/object_detection/box_coders/keypoint_box_coder.py rename to research/object_detection/object_detection/box_coders/keypoint_box_coder.py diff --git a/research/object_detection/box_coders/keypoint_box_coder_test.py b/research/object_detection/object_detection/box_coders/keypoint_box_coder_test.py similarity index 100% rename from research/object_detection/box_coders/keypoint_box_coder_test.py rename to research/object_detection/object_detection/box_coders/keypoint_box_coder_test.py diff --git a/research/object_detection/box_coders/mean_stddev_box_coder.py b/research/object_detection/object_detection/box_coders/mean_stddev_box_coder.py similarity index 100% rename from research/object_detection/box_coders/mean_stddev_box_coder.py rename to research/object_detection/object_detection/box_coders/mean_stddev_box_coder.py diff --git a/research/object_detection/box_coders/mean_stddev_box_coder_test.py b/research/object_detection/object_detection/box_coders/mean_stddev_box_coder_test.py similarity index 100% rename from research/object_detection/box_coders/mean_stddev_box_coder_test.py rename to research/object_detection/object_detection/box_coders/mean_stddev_box_coder_test.py diff --git a/research/object_detection/box_coders/square_box_coder.py b/research/object_detection/object_detection/box_coders/square_box_coder.py similarity index 100% rename from research/object_detection/box_coders/square_box_coder.py rename to research/object_detection/object_detection/box_coders/square_box_coder.py diff --git a/research/object_detection/box_coders/square_box_coder_test.py b/research/object_detection/object_detection/box_coders/square_box_coder_test.py similarity index 100% rename from research/object_detection/box_coders/square_box_coder_test.py rename to research/object_detection/object_detection/box_coders/square_box_coder_test.py diff --git a/research/object_detection/builders/__init__.py b/research/object_detection/object_detection/builders/__init__.py similarity index 100% rename from research/object_detection/builders/__init__.py rename to research/object_detection/object_detection/builders/__init__.py diff --git a/research/object_detection/builders/anchor_generator_builder.py b/research/object_detection/object_detection/builders/anchor_generator_builder.py similarity index 100% rename from research/object_detection/builders/anchor_generator_builder.py rename to research/object_detection/object_detection/builders/anchor_generator_builder.py diff --git a/research/object_detection/builders/anchor_generator_builder_test.py b/research/object_detection/object_detection/builders/anchor_generator_builder_test.py similarity index 100% rename from research/object_detection/builders/anchor_generator_builder_test.py rename to research/object_detection/object_detection/builders/anchor_generator_builder_test.py diff --git a/research/object_detection/builders/box_coder_builder.py b/research/object_detection/object_detection/builders/box_coder_builder.py similarity index 100% rename from research/object_detection/builders/box_coder_builder.py rename to research/object_detection/object_detection/builders/box_coder_builder.py diff --git a/research/object_detection/builders/box_coder_builder_test.py b/research/object_detection/object_detection/builders/box_coder_builder_test.py similarity index 100% rename from research/object_detection/builders/box_coder_builder_test.py rename to research/object_detection/object_detection/builders/box_coder_builder_test.py diff --git a/research/object_detection/builders/box_predictor_builder.py b/research/object_detection/object_detection/builders/box_predictor_builder.py similarity index 100% rename from research/object_detection/builders/box_predictor_builder.py rename to research/object_detection/object_detection/builders/box_predictor_builder.py diff --git a/research/object_detection/builders/box_predictor_builder_test.py b/research/object_detection/object_detection/builders/box_predictor_builder_test.py similarity index 100% rename from research/object_detection/builders/box_predictor_builder_test.py rename to research/object_detection/object_detection/builders/box_predictor_builder_test.py diff --git a/research/object_detection/builders/dataset_builder.py b/research/object_detection/object_detection/builders/dataset_builder.py similarity index 100% rename from research/object_detection/builders/dataset_builder.py rename to research/object_detection/object_detection/builders/dataset_builder.py diff --git a/research/object_detection/builders/dataset_builder_test.py b/research/object_detection/object_detection/builders/dataset_builder_test.py similarity index 100% rename from research/object_detection/builders/dataset_builder_test.py rename to research/object_detection/object_detection/builders/dataset_builder_test.py diff --git a/research/object_detection/builders/graph_rewriter_builder.py b/research/object_detection/object_detection/builders/graph_rewriter_builder.py similarity index 100% rename from research/object_detection/builders/graph_rewriter_builder.py rename to research/object_detection/object_detection/builders/graph_rewriter_builder.py diff --git a/research/object_detection/builders/graph_rewriter_builder_test.py b/research/object_detection/object_detection/builders/graph_rewriter_builder_test.py similarity index 100% rename from research/object_detection/builders/graph_rewriter_builder_test.py rename to research/object_detection/object_detection/builders/graph_rewriter_builder_test.py diff --git a/research/object_detection/builders/hyperparams_builder.py b/research/object_detection/object_detection/builders/hyperparams_builder.py similarity index 100% rename from research/object_detection/builders/hyperparams_builder.py rename to research/object_detection/object_detection/builders/hyperparams_builder.py diff --git a/research/object_detection/builders/hyperparams_builder_test.py b/research/object_detection/object_detection/builders/hyperparams_builder_test.py similarity index 100% rename from research/object_detection/builders/hyperparams_builder_test.py rename to research/object_detection/object_detection/builders/hyperparams_builder_test.py diff --git a/research/object_detection/builders/image_resizer_builder.py b/research/object_detection/object_detection/builders/image_resizer_builder.py similarity index 100% rename from research/object_detection/builders/image_resizer_builder.py rename to research/object_detection/object_detection/builders/image_resizer_builder.py diff --git a/research/object_detection/builders/image_resizer_builder_test.py b/research/object_detection/object_detection/builders/image_resizer_builder_test.py similarity index 100% rename from research/object_detection/builders/image_resizer_builder_test.py rename to research/object_detection/object_detection/builders/image_resizer_builder_test.py diff --git a/research/object_detection/builders/input_reader_builder.py b/research/object_detection/object_detection/builders/input_reader_builder.py similarity index 100% rename from research/object_detection/builders/input_reader_builder.py rename to research/object_detection/object_detection/builders/input_reader_builder.py diff --git a/research/object_detection/builders/input_reader_builder_test.py b/research/object_detection/object_detection/builders/input_reader_builder_test.py similarity index 100% rename from research/object_detection/builders/input_reader_builder_test.py rename to research/object_detection/object_detection/builders/input_reader_builder_test.py diff --git a/research/object_detection/builders/losses_builder.py b/research/object_detection/object_detection/builders/losses_builder.py similarity index 100% rename from research/object_detection/builders/losses_builder.py rename to research/object_detection/object_detection/builders/losses_builder.py diff --git a/research/object_detection/builders/losses_builder_test.py b/research/object_detection/object_detection/builders/losses_builder_test.py similarity index 100% rename from research/object_detection/builders/losses_builder_test.py rename to research/object_detection/object_detection/builders/losses_builder_test.py diff --git a/research/object_detection/builders/matcher_builder.py b/research/object_detection/object_detection/builders/matcher_builder.py similarity index 100% rename from research/object_detection/builders/matcher_builder.py rename to research/object_detection/object_detection/builders/matcher_builder.py diff --git a/research/object_detection/builders/matcher_builder_test.py b/research/object_detection/object_detection/builders/matcher_builder_test.py similarity index 100% rename from research/object_detection/builders/matcher_builder_test.py rename to research/object_detection/object_detection/builders/matcher_builder_test.py diff --git a/research/object_detection/builders/model_builder.py b/research/object_detection/object_detection/builders/model_builder.py similarity index 100% rename from research/object_detection/builders/model_builder.py rename to research/object_detection/object_detection/builders/model_builder.py diff --git a/research/object_detection/builders/model_builder_test.py b/research/object_detection/object_detection/builders/model_builder_test.py similarity index 100% rename from research/object_detection/builders/model_builder_test.py rename to research/object_detection/object_detection/builders/model_builder_test.py diff --git a/research/object_detection/builders/optimizer_builder.py b/research/object_detection/object_detection/builders/optimizer_builder.py similarity index 100% rename from research/object_detection/builders/optimizer_builder.py rename to research/object_detection/object_detection/builders/optimizer_builder.py diff --git a/research/object_detection/builders/optimizer_builder_test.py b/research/object_detection/object_detection/builders/optimizer_builder_test.py similarity index 100% rename from research/object_detection/builders/optimizer_builder_test.py rename to research/object_detection/object_detection/builders/optimizer_builder_test.py diff --git a/research/object_detection/builders/post_processing_builder.py b/research/object_detection/object_detection/builders/post_processing_builder.py similarity index 100% rename from research/object_detection/builders/post_processing_builder.py rename to research/object_detection/object_detection/builders/post_processing_builder.py diff --git a/research/object_detection/builders/post_processing_builder_test.py b/research/object_detection/object_detection/builders/post_processing_builder_test.py similarity index 100% rename from research/object_detection/builders/post_processing_builder_test.py rename to research/object_detection/object_detection/builders/post_processing_builder_test.py diff --git a/research/object_detection/builders/preprocessor_builder.py b/research/object_detection/object_detection/builders/preprocessor_builder.py similarity index 100% rename from research/object_detection/builders/preprocessor_builder.py rename to research/object_detection/object_detection/builders/preprocessor_builder.py diff --git a/research/object_detection/builders/preprocessor_builder_test.py b/research/object_detection/object_detection/builders/preprocessor_builder_test.py similarity index 100% rename from research/object_detection/builders/preprocessor_builder_test.py rename to research/object_detection/object_detection/builders/preprocessor_builder_test.py diff --git a/research/object_detection/builders/region_similarity_calculator_builder.py b/research/object_detection/object_detection/builders/region_similarity_calculator_builder.py similarity index 100% rename from research/object_detection/builders/region_similarity_calculator_builder.py rename to research/object_detection/object_detection/builders/region_similarity_calculator_builder.py diff --git a/research/object_detection/builders/region_similarity_calculator_builder_test.py b/research/object_detection/object_detection/builders/region_similarity_calculator_builder_test.py similarity index 100% rename from research/object_detection/builders/region_similarity_calculator_builder_test.py rename to research/object_detection/object_detection/builders/region_similarity_calculator_builder_test.py diff --git a/research/object_detection/core/__init__.py b/research/object_detection/object_detection/core/__init__.py similarity index 100% rename from research/object_detection/core/__init__.py rename to research/object_detection/object_detection/core/__init__.py diff --git a/research/object_detection/core/anchor_generator.py b/research/object_detection/object_detection/core/anchor_generator.py similarity index 100% rename from research/object_detection/core/anchor_generator.py rename to research/object_detection/object_detection/core/anchor_generator.py diff --git a/research/object_detection/core/balanced_positive_negative_sampler.py b/research/object_detection/object_detection/core/balanced_positive_negative_sampler.py similarity index 100% rename from research/object_detection/core/balanced_positive_negative_sampler.py rename to research/object_detection/object_detection/core/balanced_positive_negative_sampler.py diff --git a/research/object_detection/core/balanced_positive_negative_sampler_test.py b/research/object_detection/object_detection/core/balanced_positive_negative_sampler_test.py similarity index 100% rename from research/object_detection/core/balanced_positive_negative_sampler_test.py rename to research/object_detection/object_detection/core/balanced_positive_negative_sampler_test.py diff --git a/research/object_detection/core/batcher.py b/research/object_detection/object_detection/core/batcher.py similarity index 100% rename from research/object_detection/core/batcher.py rename to research/object_detection/object_detection/core/batcher.py diff --git a/research/object_detection/core/batcher_test.py b/research/object_detection/object_detection/core/batcher_test.py similarity index 100% rename from research/object_detection/core/batcher_test.py rename to research/object_detection/object_detection/core/batcher_test.py diff --git a/research/object_detection/core/box_coder.py b/research/object_detection/object_detection/core/box_coder.py similarity index 100% rename from research/object_detection/core/box_coder.py rename to research/object_detection/object_detection/core/box_coder.py diff --git a/research/object_detection/core/box_coder_test.py b/research/object_detection/object_detection/core/box_coder_test.py similarity index 100% rename from research/object_detection/core/box_coder_test.py rename to research/object_detection/object_detection/core/box_coder_test.py diff --git a/research/object_detection/core/box_list.py b/research/object_detection/object_detection/core/box_list.py similarity index 100% rename from research/object_detection/core/box_list.py rename to research/object_detection/object_detection/core/box_list.py diff --git a/research/object_detection/core/box_list_ops.py b/research/object_detection/object_detection/core/box_list_ops.py similarity index 100% rename from research/object_detection/core/box_list_ops.py rename to research/object_detection/object_detection/core/box_list_ops.py diff --git a/research/object_detection/core/box_list_ops_test.py b/research/object_detection/object_detection/core/box_list_ops_test.py similarity index 100% rename from research/object_detection/core/box_list_ops_test.py rename to research/object_detection/object_detection/core/box_list_ops_test.py diff --git a/research/object_detection/core/box_list_test.py b/research/object_detection/object_detection/core/box_list_test.py similarity index 100% rename from research/object_detection/core/box_list_test.py rename to research/object_detection/object_detection/core/box_list_test.py diff --git a/research/object_detection/core/box_predictor.py b/research/object_detection/object_detection/core/box_predictor.py similarity index 100% rename from research/object_detection/core/box_predictor.py rename to research/object_detection/object_detection/core/box_predictor.py diff --git a/research/object_detection/core/box_predictor_test.py b/research/object_detection/object_detection/core/box_predictor_test.py similarity index 100% rename from research/object_detection/core/box_predictor_test.py rename to research/object_detection/object_detection/core/box_predictor_test.py diff --git a/research/object_detection/core/data_decoder.py b/research/object_detection/object_detection/core/data_decoder.py similarity index 100% rename from research/object_detection/core/data_decoder.py rename to research/object_detection/object_detection/core/data_decoder.py diff --git a/research/object_detection/core/data_parser.py b/research/object_detection/object_detection/core/data_parser.py similarity index 100% rename from research/object_detection/core/data_parser.py rename to research/object_detection/object_detection/core/data_parser.py diff --git a/research/object_detection/core/keypoint_ops.py b/research/object_detection/object_detection/core/keypoint_ops.py similarity index 100% rename from research/object_detection/core/keypoint_ops.py rename to research/object_detection/object_detection/core/keypoint_ops.py diff --git a/research/object_detection/core/keypoint_ops_test.py b/research/object_detection/object_detection/core/keypoint_ops_test.py similarity index 100% rename from research/object_detection/core/keypoint_ops_test.py rename to research/object_detection/object_detection/core/keypoint_ops_test.py diff --git a/research/object_detection/core/losses.py b/research/object_detection/object_detection/core/losses.py similarity index 100% rename from research/object_detection/core/losses.py rename to research/object_detection/object_detection/core/losses.py diff --git a/research/object_detection/core/losses_test.py b/research/object_detection/object_detection/core/losses_test.py similarity index 100% rename from research/object_detection/core/losses_test.py rename to research/object_detection/object_detection/core/losses_test.py diff --git a/research/object_detection/core/matcher.py b/research/object_detection/object_detection/core/matcher.py similarity index 100% rename from research/object_detection/core/matcher.py rename to research/object_detection/object_detection/core/matcher.py diff --git a/research/object_detection/core/matcher_test.py b/research/object_detection/object_detection/core/matcher_test.py similarity index 100% rename from research/object_detection/core/matcher_test.py rename to research/object_detection/object_detection/core/matcher_test.py diff --git a/research/object_detection/core/minibatch_sampler.py b/research/object_detection/object_detection/core/minibatch_sampler.py similarity index 100% rename from research/object_detection/core/minibatch_sampler.py rename to research/object_detection/object_detection/core/minibatch_sampler.py diff --git a/research/object_detection/core/minibatch_sampler_test.py b/research/object_detection/object_detection/core/minibatch_sampler_test.py similarity index 100% rename from research/object_detection/core/minibatch_sampler_test.py rename to research/object_detection/object_detection/core/minibatch_sampler_test.py diff --git a/research/object_detection/core/model.py b/research/object_detection/object_detection/core/model.py similarity index 100% rename from research/object_detection/core/model.py rename to research/object_detection/object_detection/core/model.py diff --git a/research/object_detection/core/post_processing.py b/research/object_detection/object_detection/core/post_processing.py similarity index 100% rename from research/object_detection/core/post_processing.py rename to research/object_detection/object_detection/core/post_processing.py diff --git a/research/object_detection/core/post_processing_test.py b/research/object_detection/object_detection/core/post_processing_test.py similarity index 100% rename from research/object_detection/core/post_processing_test.py rename to research/object_detection/object_detection/core/post_processing_test.py diff --git a/research/object_detection/core/prefetcher.py b/research/object_detection/object_detection/core/prefetcher.py similarity index 100% rename from research/object_detection/core/prefetcher.py rename to research/object_detection/object_detection/core/prefetcher.py diff --git a/research/object_detection/core/prefetcher_test.py b/research/object_detection/object_detection/core/prefetcher_test.py similarity index 100% rename from research/object_detection/core/prefetcher_test.py rename to research/object_detection/object_detection/core/prefetcher_test.py diff --git a/research/object_detection/core/preprocessor.py b/research/object_detection/object_detection/core/preprocessor.py similarity index 100% rename from research/object_detection/core/preprocessor.py rename to research/object_detection/object_detection/core/preprocessor.py diff --git a/research/object_detection/core/preprocessor_cache.py b/research/object_detection/object_detection/core/preprocessor_cache.py similarity index 100% rename from research/object_detection/core/preprocessor_cache.py rename to research/object_detection/object_detection/core/preprocessor_cache.py diff --git a/research/object_detection/core/preprocessor_test.py b/research/object_detection/object_detection/core/preprocessor_test.py similarity index 100% rename from research/object_detection/core/preprocessor_test.py rename to research/object_detection/object_detection/core/preprocessor_test.py diff --git a/research/object_detection/core/region_similarity_calculator.py b/research/object_detection/object_detection/core/region_similarity_calculator.py similarity index 100% rename from research/object_detection/core/region_similarity_calculator.py rename to research/object_detection/object_detection/core/region_similarity_calculator.py diff --git a/research/object_detection/core/region_similarity_calculator_test.py b/research/object_detection/object_detection/core/region_similarity_calculator_test.py similarity index 100% rename from research/object_detection/core/region_similarity_calculator_test.py rename to research/object_detection/object_detection/core/region_similarity_calculator_test.py diff --git a/research/object_detection/core/standard_fields.py b/research/object_detection/object_detection/core/standard_fields.py similarity index 100% rename from research/object_detection/core/standard_fields.py rename to research/object_detection/object_detection/core/standard_fields.py diff --git a/research/object_detection/core/target_assigner.py b/research/object_detection/object_detection/core/target_assigner.py similarity index 100% rename from research/object_detection/core/target_assigner.py rename to research/object_detection/object_detection/core/target_assigner.py diff --git a/research/object_detection/core/target_assigner_test.py b/research/object_detection/object_detection/core/target_assigner_test.py similarity index 100% rename from research/object_detection/core/target_assigner_test.py rename to research/object_detection/object_detection/core/target_assigner_test.py diff --git a/research/object_detection/data/ava_label_map_v2.1.pbtxt b/research/object_detection/object_detection/data/ava_label_map_v2.1.pbtxt similarity index 100% rename from research/object_detection/data/ava_label_map_v2.1.pbtxt rename to research/object_detection/object_detection/data/ava_label_map_v2.1.pbtxt diff --git a/research/object_detection/data/kitti_label_map.pbtxt b/research/object_detection/object_detection/data/kitti_label_map.pbtxt similarity index 100% rename from research/object_detection/data/kitti_label_map.pbtxt rename to research/object_detection/object_detection/data/kitti_label_map.pbtxt diff --git a/research/object_detection/data/mscoco_label_map.pbtxt b/research/object_detection/object_detection/data/mscoco_label_map.pbtxt similarity index 100% rename from research/object_detection/data/mscoco_label_map.pbtxt rename to research/object_detection/object_detection/data/mscoco_label_map.pbtxt diff --git a/research/object_detection/data/oid_bbox_trainable_label_map.pbtxt b/research/object_detection/object_detection/data/oid_bbox_trainable_label_map.pbtxt similarity index 100% rename from research/object_detection/data/oid_bbox_trainable_label_map.pbtxt rename to research/object_detection/object_detection/data/oid_bbox_trainable_label_map.pbtxt diff --git a/research/object_detection/data/oid_object_detection_challenge_500_label_map.pbtxt b/research/object_detection/object_detection/data/oid_object_detection_challenge_500_label_map.pbtxt similarity index 100% rename from research/object_detection/data/oid_object_detection_challenge_500_label_map.pbtxt rename to research/object_detection/object_detection/data/oid_object_detection_challenge_500_label_map.pbtxt diff --git a/research/object_detection/data/pascal_label_map.pbtxt b/research/object_detection/object_detection/data/pascal_label_map.pbtxt similarity index 100% rename from research/object_detection/data/pascal_label_map.pbtxt rename to research/object_detection/object_detection/data/pascal_label_map.pbtxt diff --git a/research/object_detection/data/pet_label_map.pbtxt b/research/object_detection/object_detection/data/pet_label_map.pbtxt similarity index 100% rename from research/object_detection/data/pet_label_map.pbtxt rename to research/object_detection/object_detection/data/pet_label_map.pbtxt diff --git a/research/object_detection/data_decoders/__init__.py b/research/object_detection/object_detection/data_decoders/__init__.py similarity index 100% rename from research/object_detection/data_decoders/__init__.py rename to research/object_detection/object_detection/data_decoders/__init__.py diff --git a/research/object_detection/data_decoders/tf_example_decoder.py b/research/object_detection/object_detection/data_decoders/tf_example_decoder.py similarity index 100% rename from research/object_detection/data_decoders/tf_example_decoder.py rename to research/object_detection/object_detection/data_decoders/tf_example_decoder.py diff --git a/research/object_detection/data_decoders/tf_example_decoder_test.py b/research/object_detection/object_detection/data_decoders/tf_example_decoder_test.py similarity index 100% rename from research/object_detection/data_decoders/tf_example_decoder_test.py rename to research/object_detection/object_detection/data_decoders/tf_example_decoder_test.py diff --git a/research/object_detection/dataset_tools/__init__.py b/research/object_detection/object_detection/dataset_tools/__init__.py similarity index 100% rename from research/object_detection/dataset_tools/__init__.py rename to research/object_detection/object_detection/dataset_tools/__init__.py diff --git a/research/object_detection/dataset_tools/create_coco_tf_record.py b/research/object_detection/object_detection/dataset_tools/create_coco_tf_record.py similarity index 100% rename from research/object_detection/dataset_tools/create_coco_tf_record.py rename to research/object_detection/object_detection/dataset_tools/create_coco_tf_record.py diff --git a/research/object_detection/dataset_tools/create_coco_tf_record_test.py b/research/object_detection/object_detection/dataset_tools/create_coco_tf_record_test.py similarity index 100% rename from research/object_detection/dataset_tools/create_coco_tf_record_test.py rename to research/object_detection/object_detection/dataset_tools/create_coco_tf_record_test.py diff --git a/research/object_detection/dataset_tools/create_kitti_tf_record.py b/research/object_detection/object_detection/dataset_tools/create_kitti_tf_record.py similarity index 100% rename from research/object_detection/dataset_tools/create_kitti_tf_record.py rename to research/object_detection/object_detection/dataset_tools/create_kitti_tf_record.py diff --git a/research/object_detection/dataset_tools/create_kitti_tf_record_test.py b/research/object_detection/object_detection/dataset_tools/create_kitti_tf_record_test.py similarity index 100% rename from research/object_detection/dataset_tools/create_kitti_tf_record_test.py rename to research/object_detection/object_detection/dataset_tools/create_kitti_tf_record_test.py diff --git a/research/object_detection/dataset_tools/create_oid_tf_record.py b/research/object_detection/object_detection/dataset_tools/create_oid_tf_record.py similarity index 100% rename from research/object_detection/dataset_tools/create_oid_tf_record.py rename to research/object_detection/object_detection/dataset_tools/create_oid_tf_record.py diff --git a/research/object_detection/dataset_tools/create_pascal_tf_record.py b/research/object_detection/object_detection/dataset_tools/create_pascal_tf_record.py similarity index 100% rename from research/object_detection/dataset_tools/create_pascal_tf_record.py rename to research/object_detection/object_detection/dataset_tools/create_pascal_tf_record.py diff --git a/research/object_detection/dataset_tools/create_pascal_tf_record_test.py b/research/object_detection/object_detection/dataset_tools/create_pascal_tf_record_test.py similarity index 100% rename from research/object_detection/dataset_tools/create_pascal_tf_record_test.py rename to research/object_detection/object_detection/dataset_tools/create_pascal_tf_record_test.py diff --git a/research/object_detection/dataset_tools/create_pet_tf_record.py b/research/object_detection/object_detection/dataset_tools/create_pet_tf_record.py similarity index 100% rename from research/object_detection/dataset_tools/create_pet_tf_record.py rename to research/object_detection/object_detection/dataset_tools/create_pet_tf_record.py diff --git a/research/object_detection/dataset_tools/create_pycocotools_package.sh b/research/object_detection/object_detection/dataset_tools/create_pycocotools_package.sh similarity index 100% rename from research/object_detection/dataset_tools/create_pycocotools_package.sh rename to research/object_detection/object_detection/dataset_tools/create_pycocotools_package.sh diff --git a/research/object_detection/dataset_tools/download_and_preprocess_mscoco.sh b/research/object_detection/object_detection/dataset_tools/download_and_preprocess_mscoco.sh similarity index 100% rename from research/object_detection/dataset_tools/download_and_preprocess_mscoco.sh rename to research/object_detection/object_detection/dataset_tools/download_and_preprocess_mscoco.sh diff --git a/research/object_detection/dataset_tools/oid_hierarchical_labels_expansion.py b/research/object_detection/object_detection/dataset_tools/oid_hierarchical_labels_expansion.py similarity index 100% rename from research/object_detection/dataset_tools/oid_hierarchical_labels_expansion.py rename to research/object_detection/object_detection/dataset_tools/oid_hierarchical_labels_expansion.py diff --git a/research/object_detection/dataset_tools/oid_hierarchical_labels_expansion_test.py b/research/object_detection/object_detection/dataset_tools/oid_hierarchical_labels_expansion_test.py similarity index 100% rename from research/object_detection/dataset_tools/oid_hierarchical_labels_expansion_test.py rename to research/object_detection/object_detection/dataset_tools/oid_hierarchical_labels_expansion_test.py diff --git a/research/object_detection/dataset_tools/oid_tfrecord_creation.py b/research/object_detection/object_detection/dataset_tools/oid_tfrecord_creation.py similarity index 100% rename from research/object_detection/dataset_tools/oid_tfrecord_creation.py rename to research/object_detection/object_detection/dataset_tools/oid_tfrecord_creation.py diff --git a/research/object_detection/dataset_tools/oid_tfrecord_creation_test.py b/research/object_detection/object_detection/dataset_tools/oid_tfrecord_creation_test.py similarity index 100% rename from research/object_detection/dataset_tools/oid_tfrecord_creation_test.py rename to research/object_detection/object_detection/dataset_tools/oid_tfrecord_creation_test.py diff --git a/research/object_detection/dataset_tools/tf_record_creation_util.py b/research/object_detection/object_detection/dataset_tools/tf_record_creation_util.py similarity index 100% rename from research/object_detection/dataset_tools/tf_record_creation_util.py rename to research/object_detection/object_detection/dataset_tools/tf_record_creation_util.py diff --git a/research/object_detection/dataset_tools/tf_record_creation_util_test.py b/research/object_detection/object_detection/dataset_tools/tf_record_creation_util_test.py similarity index 100% rename from research/object_detection/dataset_tools/tf_record_creation_util_test.py rename to research/object_detection/object_detection/dataset_tools/tf_record_creation_util_test.py diff --git a/research/object_detection/dockerfiles/android/Dockerfile b/research/object_detection/object_detection/dockerfiles/android/Dockerfile similarity index 100% rename from research/object_detection/dockerfiles/android/Dockerfile rename to research/object_detection/object_detection/dockerfiles/android/Dockerfile diff --git a/research/object_detection/dockerfiles/android/README.md b/research/object_detection/object_detection/dockerfiles/android/README.md similarity index 100% rename from research/object_detection/dockerfiles/android/README.md rename to research/object_detection/object_detection/dockerfiles/android/README.md diff --git a/research/object_detection/eval_util.py b/research/object_detection/object_detection/eval_util.py similarity index 100% rename from research/object_detection/eval_util.py rename to research/object_detection/object_detection/eval_util.py diff --git a/research/object_detection/eval_util_test.py b/research/object_detection/object_detection/eval_util_test.py similarity index 100% rename from research/object_detection/eval_util_test.py rename to research/object_detection/object_detection/eval_util_test.py diff --git a/research/object_detection/export_inference_graph.py b/research/object_detection/object_detection/export_inference_graph.py similarity index 100% rename from research/object_detection/export_inference_graph.py rename to research/object_detection/object_detection/export_inference_graph.py diff --git a/research/object_detection/export_tflite_ssd_graph.py b/research/object_detection/object_detection/export_tflite_ssd_graph.py similarity index 100% rename from research/object_detection/export_tflite_ssd_graph.py rename to research/object_detection/object_detection/export_tflite_ssd_graph.py diff --git a/research/object_detection/export_tflite_ssd_graph_lib.py b/research/object_detection/object_detection/export_tflite_ssd_graph_lib.py similarity index 100% rename from research/object_detection/export_tflite_ssd_graph_lib.py rename to research/object_detection/object_detection/export_tflite_ssd_graph_lib.py diff --git a/research/object_detection/export_tflite_ssd_graph_lib_test.py b/research/object_detection/object_detection/export_tflite_ssd_graph_lib_test.py similarity index 100% rename from research/object_detection/export_tflite_ssd_graph_lib_test.py rename to research/object_detection/object_detection/export_tflite_ssd_graph_lib_test.py diff --git a/research/object_detection/exporter.py b/research/object_detection/object_detection/exporter.py similarity index 100% rename from research/object_detection/exporter.py rename to research/object_detection/object_detection/exporter.py diff --git a/research/object_detection/exporter_test.py b/research/object_detection/object_detection/exporter_test.py similarity index 100% rename from research/object_detection/exporter_test.py rename to research/object_detection/object_detection/exporter_test.py diff --git a/research/object_detection/inference/__init__.py b/research/object_detection/object_detection/inference/__init__.py similarity index 100% rename from research/object_detection/inference/__init__.py rename to research/object_detection/object_detection/inference/__init__.py diff --git a/research/object_detection/inference/detection_inference.py b/research/object_detection/object_detection/inference/detection_inference.py similarity index 100% rename from research/object_detection/inference/detection_inference.py rename to research/object_detection/object_detection/inference/detection_inference.py diff --git a/research/object_detection/inference/detection_inference_test.py b/research/object_detection/object_detection/inference/detection_inference_test.py similarity index 100% rename from research/object_detection/inference/detection_inference_test.py rename to research/object_detection/object_detection/inference/detection_inference_test.py diff --git a/research/object_detection/inference/infer_detections.py b/research/object_detection/object_detection/inference/infer_detections.py similarity index 100% rename from research/object_detection/inference/infer_detections.py rename to research/object_detection/object_detection/inference/infer_detections.py diff --git a/research/object_detection/inputs.py b/research/object_detection/object_detection/inputs.py similarity index 100% rename from research/object_detection/inputs.py rename to research/object_detection/object_detection/inputs.py diff --git a/research/object_detection/inputs_test.py b/research/object_detection/object_detection/inputs_test.py similarity index 100% rename from research/object_detection/inputs_test.py rename to research/object_detection/object_detection/inputs_test.py diff --git a/research/object_detection/legacy/__init__.py b/research/object_detection/object_detection/legacy/__init__.py similarity index 100% rename from research/object_detection/legacy/__init__.py rename to research/object_detection/object_detection/legacy/__init__.py diff --git a/research/object_detection/legacy/eval.py b/research/object_detection/object_detection/legacy/eval.py similarity index 100% rename from research/object_detection/legacy/eval.py rename to research/object_detection/object_detection/legacy/eval.py diff --git a/research/object_detection/legacy/evaluator.py b/research/object_detection/object_detection/legacy/evaluator.py similarity index 100% rename from research/object_detection/legacy/evaluator.py rename to research/object_detection/object_detection/legacy/evaluator.py diff --git a/research/object_detection/legacy/train.py b/research/object_detection/object_detection/legacy/train.py similarity index 100% rename from research/object_detection/legacy/train.py rename to research/object_detection/object_detection/legacy/train.py diff --git a/research/object_detection/legacy/trainer.py b/research/object_detection/object_detection/legacy/trainer.py similarity index 100% rename from research/object_detection/legacy/trainer.py rename to research/object_detection/object_detection/legacy/trainer.py diff --git a/research/object_detection/legacy/trainer_test.py b/research/object_detection/object_detection/legacy/trainer_test.py similarity index 100% rename from research/object_detection/legacy/trainer_test.py rename to research/object_detection/object_detection/legacy/trainer_test.py diff --git a/research/object_detection/matchers/__init__.py b/research/object_detection/object_detection/matchers/__init__.py similarity index 100% rename from research/object_detection/matchers/__init__.py rename to research/object_detection/object_detection/matchers/__init__.py diff --git a/research/object_detection/matchers/argmax_matcher.py b/research/object_detection/object_detection/matchers/argmax_matcher.py similarity index 100% rename from research/object_detection/matchers/argmax_matcher.py rename to research/object_detection/object_detection/matchers/argmax_matcher.py diff --git a/research/object_detection/matchers/argmax_matcher_test.py b/research/object_detection/object_detection/matchers/argmax_matcher_test.py similarity index 100% rename from research/object_detection/matchers/argmax_matcher_test.py rename to research/object_detection/object_detection/matchers/argmax_matcher_test.py diff --git a/research/object_detection/matchers/bipartite_matcher.py b/research/object_detection/object_detection/matchers/bipartite_matcher.py similarity index 100% rename from research/object_detection/matchers/bipartite_matcher.py rename to research/object_detection/object_detection/matchers/bipartite_matcher.py diff --git a/research/object_detection/matchers/bipartite_matcher_test.py b/research/object_detection/object_detection/matchers/bipartite_matcher_test.py similarity index 100% rename from research/object_detection/matchers/bipartite_matcher_test.py rename to research/object_detection/object_detection/matchers/bipartite_matcher_test.py diff --git a/research/object_detection/meta_architectures/__init__.py b/research/object_detection/object_detection/meta_architectures/__init__.py similarity index 100% rename from research/object_detection/meta_architectures/__init__.py rename to research/object_detection/object_detection/meta_architectures/__init__.py diff --git a/research/object_detection/meta_architectures/faster_rcnn_meta_arch.py b/research/object_detection/object_detection/meta_architectures/faster_rcnn_meta_arch.py similarity index 100% rename from research/object_detection/meta_architectures/faster_rcnn_meta_arch.py rename to research/object_detection/object_detection/meta_architectures/faster_rcnn_meta_arch.py diff --git a/research/object_detection/meta_architectures/faster_rcnn_meta_arch_test.py b/research/object_detection/object_detection/meta_architectures/faster_rcnn_meta_arch_test.py similarity index 100% rename from research/object_detection/meta_architectures/faster_rcnn_meta_arch_test.py rename to research/object_detection/object_detection/meta_architectures/faster_rcnn_meta_arch_test.py diff --git a/research/object_detection/meta_architectures/faster_rcnn_meta_arch_test_lib.py b/research/object_detection/object_detection/meta_architectures/faster_rcnn_meta_arch_test_lib.py similarity index 100% rename from research/object_detection/meta_architectures/faster_rcnn_meta_arch_test_lib.py rename to research/object_detection/object_detection/meta_architectures/faster_rcnn_meta_arch_test_lib.py diff --git a/research/object_detection/meta_architectures/rfcn_meta_arch.py b/research/object_detection/object_detection/meta_architectures/rfcn_meta_arch.py similarity index 100% rename from research/object_detection/meta_architectures/rfcn_meta_arch.py rename to research/object_detection/object_detection/meta_architectures/rfcn_meta_arch.py diff --git a/research/object_detection/meta_architectures/rfcn_meta_arch_test.py b/research/object_detection/object_detection/meta_architectures/rfcn_meta_arch_test.py similarity index 100% rename from research/object_detection/meta_architectures/rfcn_meta_arch_test.py rename to research/object_detection/object_detection/meta_architectures/rfcn_meta_arch_test.py diff --git a/research/object_detection/meta_architectures/ssd_meta_arch.py b/research/object_detection/object_detection/meta_architectures/ssd_meta_arch.py similarity index 100% rename from research/object_detection/meta_architectures/ssd_meta_arch.py rename to research/object_detection/object_detection/meta_architectures/ssd_meta_arch.py diff --git a/research/object_detection/meta_architectures/ssd_meta_arch_test.py b/research/object_detection/object_detection/meta_architectures/ssd_meta_arch_test.py similarity index 100% rename from research/object_detection/meta_architectures/ssd_meta_arch_test.py rename to research/object_detection/object_detection/meta_architectures/ssd_meta_arch_test.py diff --git a/research/object_detection/metrics/__init__.py b/research/object_detection/object_detection/metrics/__init__.py similarity index 100% rename from research/object_detection/metrics/__init__.py rename to research/object_detection/object_detection/metrics/__init__.py diff --git a/research/object_detection/metrics/coco_evaluation.py b/research/object_detection/object_detection/metrics/coco_evaluation.py similarity index 100% rename from research/object_detection/metrics/coco_evaluation.py rename to research/object_detection/object_detection/metrics/coco_evaluation.py diff --git a/research/object_detection/metrics/coco_evaluation_test.py b/research/object_detection/object_detection/metrics/coco_evaluation_test.py similarity index 100% rename from research/object_detection/metrics/coco_evaluation_test.py rename to research/object_detection/object_detection/metrics/coco_evaluation_test.py diff --git a/research/object_detection/metrics/coco_tools.py b/research/object_detection/object_detection/metrics/coco_tools.py similarity index 100% rename from research/object_detection/metrics/coco_tools.py rename to research/object_detection/object_detection/metrics/coco_tools.py diff --git a/research/object_detection/metrics/coco_tools_test.py b/research/object_detection/object_detection/metrics/coco_tools_test.py similarity index 100% rename from research/object_detection/metrics/coco_tools_test.py rename to research/object_detection/object_detection/metrics/coco_tools_test.py diff --git a/research/object_detection/metrics/io_utils.py b/research/object_detection/object_detection/metrics/io_utils.py similarity index 100% rename from research/object_detection/metrics/io_utils.py rename to research/object_detection/object_detection/metrics/io_utils.py diff --git a/research/object_detection/metrics/offline_eval_map_corloc.py b/research/object_detection/object_detection/metrics/offline_eval_map_corloc.py similarity index 100% rename from research/object_detection/metrics/offline_eval_map_corloc.py rename to research/object_detection/object_detection/metrics/offline_eval_map_corloc.py diff --git a/research/object_detection/metrics/offline_eval_map_corloc_test.py b/research/object_detection/object_detection/metrics/offline_eval_map_corloc_test.py similarity index 100% rename from research/object_detection/metrics/offline_eval_map_corloc_test.py rename to research/object_detection/object_detection/metrics/offline_eval_map_corloc_test.py diff --git a/research/object_detection/metrics/oid_od_challenge_evaluation.py b/research/object_detection/object_detection/metrics/oid_od_challenge_evaluation.py similarity index 100% rename from research/object_detection/metrics/oid_od_challenge_evaluation.py rename to research/object_detection/object_detection/metrics/oid_od_challenge_evaluation.py diff --git a/research/object_detection/metrics/oid_od_challenge_evaluation_utils.py b/research/object_detection/object_detection/metrics/oid_od_challenge_evaluation_utils.py similarity index 100% rename from research/object_detection/metrics/oid_od_challenge_evaluation_utils.py rename to research/object_detection/object_detection/metrics/oid_od_challenge_evaluation_utils.py diff --git a/research/object_detection/metrics/oid_od_challenge_evaluation_utils_test.py b/research/object_detection/object_detection/metrics/oid_od_challenge_evaluation_utils_test.py similarity index 100% rename from research/object_detection/metrics/oid_od_challenge_evaluation_utils_test.py rename to research/object_detection/object_detection/metrics/oid_od_challenge_evaluation_utils_test.py diff --git a/research/object_detection/metrics/oid_vrd_challenge_evaluation.py b/research/object_detection/object_detection/metrics/oid_vrd_challenge_evaluation.py similarity index 100% rename from research/object_detection/metrics/oid_vrd_challenge_evaluation.py rename to research/object_detection/object_detection/metrics/oid_vrd_challenge_evaluation.py diff --git a/research/object_detection/metrics/oid_vrd_challenge_evaluation_utils.py b/research/object_detection/object_detection/metrics/oid_vrd_challenge_evaluation_utils.py similarity index 100% rename from research/object_detection/metrics/oid_vrd_challenge_evaluation_utils.py rename to research/object_detection/object_detection/metrics/oid_vrd_challenge_evaluation_utils.py diff --git a/research/object_detection/metrics/oid_vrd_challenge_evaluation_utils_test.py b/research/object_detection/object_detection/metrics/oid_vrd_challenge_evaluation_utils_test.py similarity index 100% rename from research/object_detection/metrics/oid_vrd_challenge_evaluation_utils_test.py rename to research/object_detection/object_detection/metrics/oid_vrd_challenge_evaluation_utils_test.py diff --git a/research/object_detection/metrics/tf_example_parser.py b/research/object_detection/object_detection/metrics/tf_example_parser.py similarity index 100% rename from research/object_detection/metrics/tf_example_parser.py rename to research/object_detection/object_detection/metrics/tf_example_parser.py diff --git a/research/object_detection/metrics/tf_example_parser_test.py b/research/object_detection/object_detection/metrics/tf_example_parser_test.py similarity index 100% rename from research/object_detection/metrics/tf_example_parser_test.py rename to research/object_detection/object_detection/metrics/tf_example_parser_test.py diff --git a/research/object_detection/model_hparams.py b/research/object_detection/object_detection/model_hparams.py similarity index 100% rename from research/object_detection/model_hparams.py rename to research/object_detection/object_detection/model_hparams.py diff --git a/research/object_detection/model_lib.py b/research/object_detection/object_detection/model_lib.py similarity index 100% rename from research/object_detection/model_lib.py rename to research/object_detection/object_detection/model_lib.py diff --git a/research/object_detection/model_lib_test.py b/research/object_detection/object_detection/model_lib_test.py similarity index 100% rename from research/object_detection/model_lib_test.py rename to research/object_detection/object_detection/model_lib_test.py diff --git a/research/object_detection/model_main.py b/research/object_detection/object_detection/model_main.py similarity index 100% rename from research/object_detection/model_main.py rename to research/object_detection/object_detection/model_main.py diff --git a/research/object_detection/model_tpu_main.py b/research/object_detection/object_detection/model_tpu_main.py similarity index 100% rename from research/object_detection/model_tpu_main.py rename to research/object_detection/object_detection/model_tpu_main.py diff --git a/research/object_detection/models/__init__.py b/research/object_detection/object_detection/models/__init__.py similarity index 100% rename from research/object_detection/models/__init__.py rename to research/object_detection/object_detection/models/__init__.py diff --git a/research/object_detection/models/embedded_ssd_mobilenet_v1_feature_extractor.py b/research/object_detection/object_detection/models/embedded_ssd_mobilenet_v1_feature_extractor.py similarity index 99% rename from research/object_detection/models/embedded_ssd_mobilenet_v1_feature_extractor.py rename to research/object_detection/object_detection/models/embedded_ssd_mobilenet_v1_feature_extractor.py index f0cad235408..84b620c1f41 100644 --- a/research/object_detection/models/embedded_ssd_mobilenet_v1_feature_extractor.py +++ b/research/object_detection/object_detection/models/embedded_ssd_mobilenet_v1_feature_extractor.py @@ -21,7 +21,7 @@ from object_detection.models import feature_map_generators from object_detection.utils import context_manager from object_detection.utils import ops -from nets import mobilenet_v1 +from slim.nets import mobilenet_v1 slim = tf.contrib.slim diff --git a/research/object_detection/models/embedded_ssd_mobilenet_v1_feature_extractor_test.py b/research/object_detection/object_detection/models/embedded_ssd_mobilenet_v1_feature_extractor_test.py similarity index 100% rename from research/object_detection/models/embedded_ssd_mobilenet_v1_feature_extractor_test.py rename to research/object_detection/object_detection/models/embedded_ssd_mobilenet_v1_feature_extractor_test.py diff --git a/research/object_detection/models/faster_rcnn_inception_resnet_v2_feature_extractor.py b/research/object_detection/object_detection/models/faster_rcnn_inception_resnet_v2_feature_extractor.py similarity index 99% rename from research/object_detection/models/faster_rcnn_inception_resnet_v2_feature_extractor.py rename to research/object_detection/object_detection/models/faster_rcnn_inception_resnet_v2_feature_extractor.py index 30b3dd4eccb..03c2bc5ff5d 100644 --- a/research/object_detection/models/faster_rcnn_inception_resnet_v2_feature_extractor.py +++ b/research/object_detection/object_detection/models/faster_rcnn_inception_resnet_v2_feature_extractor.py @@ -25,7 +25,7 @@ import tensorflow as tf from object_detection.meta_architectures import faster_rcnn_meta_arch -from nets import inception_resnet_v2 +from slim.nets import inception_resnet_v2 slim = tf.contrib.slim diff --git a/research/object_detection/models/faster_rcnn_inception_resnet_v2_feature_extractor_test.py b/research/object_detection/object_detection/models/faster_rcnn_inception_resnet_v2_feature_extractor_test.py similarity index 100% rename from research/object_detection/models/faster_rcnn_inception_resnet_v2_feature_extractor_test.py rename to research/object_detection/object_detection/models/faster_rcnn_inception_resnet_v2_feature_extractor_test.py diff --git a/research/object_detection/models/faster_rcnn_inception_v2_feature_extractor.py b/research/object_detection/object_detection/models/faster_rcnn_inception_v2_feature_extractor.py similarity index 99% rename from research/object_detection/models/faster_rcnn_inception_v2_feature_extractor.py rename to research/object_detection/object_detection/models/faster_rcnn_inception_v2_feature_extractor.py index fe0675169bf..79e30e68ba1 100644 --- a/research/object_detection/models/faster_rcnn_inception_v2_feature_extractor.py +++ b/research/object_detection/object_detection/models/faster_rcnn_inception_v2_feature_extractor.py @@ -21,7 +21,7 @@ import tensorflow as tf from object_detection.meta_architectures import faster_rcnn_meta_arch -from nets import inception_v2 +from slim.nets import inception_v2 slim = tf.contrib.slim diff --git a/research/object_detection/models/faster_rcnn_inception_v2_feature_extractor_test.py b/research/object_detection/object_detection/models/faster_rcnn_inception_v2_feature_extractor_test.py similarity index 100% rename from research/object_detection/models/faster_rcnn_inception_v2_feature_extractor_test.py rename to research/object_detection/object_detection/models/faster_rcnn_inception_v2_feature_extractor_test.py diff --git a/research/object_detection/models/faster_rcnn_mobilenet_v1_feature_extractor.py b/research/object_detection/object_detection/models/faster_rcnn_mobilenet_v1_feature_extractor.py similarity index 99% rename from research/object_detection/models/faster_rcnn_mobilenet_v1_feature_extractor.py rename to research/object_detection/object_detection/models/faster_rcnn_mobilenet_v1_feature_extractor.py index 52c744b8293..ce7c76c2ca6 100644 --- a/research/object_detection/models/faster_rcnn_mobilenet_v1_feature_extractor.py +++ b/research/object_detection/object_detection/models/faster_rcnn_mobilenet_v1_feature_extractor.py @@ -20,7 +20,7 @@ from object_detection.meta_architectures import faster_rcnn_meta_arch from object_detection.utils import shape_utils -from nets import mobilenet_v1 +from slim.nets import mobilenet_v1 slim = tf.contrib.slim diff --git a/research/object_detection/models/faster_rcnn_mobilenet_v1_feature_extractor_test.py b/research/object_detection/object_detection/models/faster_rcnn_mobilenet_v1_feature_extractor_test.py similarity index 100% rename from research/object_detection/models/faster_rcnn_mobilenet_v1_feature_extractor_test.py rename to research/object_detection/object_detection/models/faster_rcnn_mobilenet_v1_feature_extractor_test.py diff --git a/research/object_detection/models/faster_rcnn_nas_feature_extractor.py b/research/object_detection/object_detection/models/faster_rcnn_nas_feature_extractor.py similarity index 99% rename from research/object_detection/models/faster_rcnn_nas_feature_extractor.py rename to research/object_detection/object_detection/models/faster_rcnn_nas_feature_extractor.py index 5fa6bf75315..c212fc70682 100644 --- a/research/object_detection/models/faster_rcnn_nas_feature_extractor.py +++ b/research/object_detection/object_detection/models/faster_rcnn_nas_feature_extractor.py @@ -23,8 +23,8 @@ import tensorflow as tf from object_detection.meta_architectures import faster_rcnn_meta_arch -from nets.nasnet import nasnet -from nets.nasnet import nasnet_utils +from slim.nets.nasnet import nasnet +from slim.nets.nasnet import nasnet_utils arg_scope = tf.contrib.framework.arg_scope slim = tf.contrib.slim diff --git a/research/object_detection/models/faster_rcnn_nas_feature_extractor_test.py b/research/object_detection/object_detection/models/faster_rcnn_nas_feature_extractor_test.py similarity index 100% rename from research/object_detection/models/faster_rcnn_nas_feature_extractor_test.py rename to research/object_detection/object_detection/models/faster_rcnn_nas_feature_extractor_test.py diff --git a/research/object_detection/models/faster_rcnn_pnas_feature_extractor.py b/research/object_detection/object_detection/models/faster_rcnn_pnas_feature_extractor.py similarity index 99% rename from research/object_detection/models/faster_rcnn_pnas_feature_extractor.py rename to research/object_detection/object_detection/models/faster_rcnn_pnas_feature_extractor.py index b5d0f43afd4..83528f30a7b 100644 --- a/research/object_detection/models/faster_rcnn_pnas_feature_extractor.py +++ b/research/object_detection/object_detection/models/faster_rcnn_pnas_feature_extractor.py @@ -21,8 +21,8 @@ import tensorflow as tf from object_detection.meta_architectures import faster_rcnn_meta_arch -from nets.nasnet import nasnet_utils -from nets.nasnet import pnasnet +from slim.nets.nasnet import nasnet_utils +from slim.nets.nasnet import pnasnet arg_scope = tf.contrib.framework.arg_scope slim = tf.contrib.slim diff --git a/research/object_detection/models/faster_rcnn_pnas_feature_extractor_test.py b/research/object_detection/object_detection/models/faster_rcnn_pnas_feature_extractor_test.py similarity index 100% rename from research/object_detection/models/faster_rcnn_pnas_feature_extractor_test.py rename to research/object_detection/object_detection/models/faster_rcnn_pnas_feature_extractor_test.py diff --git a/research/object_detection/models/faster_rcnn_resnet_v1_feature_extractor.py b/research/object_detection/object_detection/models/faster_rcnn_resnet_v1_feature_extractor.py similarity index 99% rename from research/object_detection/models/faster_rcnn_resnet_v1_feature_extractor.py rename to research/object_detection/object_detection/models/faster_rcnn_resnet_v1_feature_extractor.py index 286deae3de7..7ec7ac752bf 100644 --- a/research/object_detection/models/faster_rcnn_resnet_v1_feature_extractor.py +++ b/research/object_detection/object_detection/models/faster_rcnn_resnet_v1_feature_extractor.py @@ -27,8 +27,8 @@ import tensorflow as tf from object_detection.meta_architectures import faster_rcnn_meta_arch -from nets import resnet_utils -from nets import resnet_v1 +from slim.nets import resnet_utils +from slim.nets import resnet_v1 slim = tf.contrib.slim diff --git a/research/object_detection/models/faster_rcnn_resnet_v1_feature_extractor_test.py b/research/object_detection/object_detection/models/faster_rcnn_resnet_v1_feature_extractor_test.py similarity index 100% rename from research/object_detection/models/faster_rcnn_resnet_v1_feature_extractor_test.py rename to research/object_detection/object_detection/models/faster_rcnn_resnet_v1_feature_extractor_test.py diff --git a/research/object_detection/models/feature_map_generators.py b/research/object_detection/object_detection/models/feature_map_generators.py similarity index 100% rename from research/object_detection/models/feature_map_generators.py rename to research/object_detection/object_detection/models/feature_map_generators.py diff --git a/research/object_detection/models/feature_map_generators_test.py b/research/object_detection/object_detection/models/feature_map_generators_test.py similarity index 100% rename from research/object_detection/models/feature_map_generators_test.py rename to research/object_detection/object_detection/models/feature_map_generators_test.py diff --git a/research/object_detection/models/ssd_feature_extractor_test.py b/research/object_detection/object_detection/models/ssd_feature_extractor_test.py similarity index 100% rename from research/object_detection/models/ssd_feature_extractor_test.py rename to research/object_detection/object_detection/models/ssd_feature_extractor_test.py diff --git a/research/object_detection/models/ssd_inception_v2_feature_extractor.py b/research/object_detection/object_detection/models/ssd_inception_v2_feature_extractor.py similarity index 99% rename from research/object_detection/models/ssd_inception_v2_feature_extractor.py rename to research/object_detection/object_detection/models/ssd_inception_v2_feature_extractor.py index b97b0f2bc99..b9ed233d0e3 100644 --- a/research/object_detection/models/ssd_inception_v2_feature_extractor.py +++ b/research/object_detection/object_detection/models/ssd_inception_v2_feature_extractor.py @@ -20,7 +20,7 @@ from object_detection.models import feature_map_generators from object_detection.utils import ops from object_detection.utils import shape_utils -from nets import inception_v2 +from slim.nets import inception_v2 slim = tf.contrib.slim diff --git a/research/object_detection/models/ssd_inception_v2_feature_extractor_test.py b/research/object_detection/object_detection/models/ssd_inception_v2_feature_extractor_test.py similarity index 100% rename from research/object_detection/models/ssd_inception_v2_feature_extractor_test.py rename to research/object_detection/object_detection/models/ssd_inception_v2_feature_extractor_test.py diff --git a/research/object_detection/models/ssd_inception_v3_feature_extractor.py b/research/object_detection/object_detection/models/ssd_inception_v3_feature_extractor.py similarity index 99% rename from research/object_detection/models/ssd_inception_v3_feature_extractor.py rename to research/object_detection/object_detection/models/ssd_inception_v3_feature_extractor.py index 5d97e7b5c9c..e4cdef9fb3c 100644 --- a/research/object_detection/models/ssd_inception_v3_feature_extractor.py +++ b/research/object_detection/object_detection/models/ssd_inception_v3_feature_extractor.py @@ -20,7 +20,7 @@ from object_detection.models import feature_map_generators from object_detection.utils import ops from object_detection.utils import shape_utils -from nets import inception_v3 +from slim.nets import inception_v3 slim = tf.contrib.slim diff --git a/research/object_detection/models/ssd_inception_v3_feature_extractor_test.py b/research/object_detection/object_detection/models/ssd_inception_v3_feature_extractor_test.py similarity index 100% rename from research/object_detection/models/ssd_inception_v3_feature_extractor_test.py rename to research/object_detection/object_detection/models/ssd_inception_v3_feature_extractor_test.py diff --git a/research/object_detection/models/ssd_mobilenet_v1_feature_extractor.py b/research/object_detection/object_detection/models/ssd_mobilenet_v1_feature_extractor.py similarity index 99% rename from research/object_detection/models/ssd_mobilenet_v1_feature_extractor.py rename to research/object_detection/object_detection/models/ssd_mobilenet_v1_feature_extractor.py index aada1111ed6..083adcc2ab9 100644 --- a/research/object_detection/models/ssd_mobilenet_v1_feature_extractor.py +++ b/research/object_detection/object_detection/models/ssd_mobilenet_v1_feature_extractor.py @@ -22,7 +22,7 @@ from object_detection.utils import context_manager from object_detection.utils import ops from object_detection.utils import shape_utils -from nets import mobilenet_v1 +from slim.nets import mobilenet_v1 slim = tf.contrib.slim diff --git a/research/object_detection/models/ssd_mobilenet_v1_feature_extractor_test.py b/research/object_detection/object_detection/models/ssd_mobilenet_v1_feature_extractor_test.py similarity index 100% rename from research/object_detection/models/ssd_mobilenet_v1_feature_extractor_test.py rename to research/object_detection/object_detection/models/ssd_mobilenet_v1_feature_extractor_test.py diff --git a/research/object_detection/models/ssd_mobilenet_v1_fpn_feature_extractor.py b/research/object_detection/object_detection/models/ssd_mobilenet_v1_fpn_feature_extractor.py similarity index 99% rename from research/object_detection/models/ssd_mobilenet_v1_fpn_feature_extractor.py rename to research/object_detection/object_detection/models/ssd_mobilenet_v1_fpn_feature_extractor.py index a52b7572ee7..9613835a58c 100644 --- a/research/object_detection/models/ssd_mobilenet_v1_fpn_feature_extractor.py +++ b/research/object_detection/object_detection/models/ssd_mobilenet_v1_fpn_feature_extractor.py @@ -22,7 +22,7 @@ from object_detection.utils import context_manager from object_detection.utils import ops from object_detection.utils import shape_utils -from nets import mobilenet_v1 +from slim.nets import mobilenet_v1 slim = tf.contrib.slim diff --git a/research/object_detection/models/ssd_mobilenet_v1_fpn_feature_extractor_test.py b/research/object_detection/object_detection/models/ssd_mobilenet_v1_fpn_feature_extractor_test.py similarity index 100% rename from research/object_detection/models/ssd_mobilenet_v1_fpn_feature_extractor_test.py rename to research/object_detection/object_detection/models/ssd_mobilenet_v1_fpn_feature_extractor_test.py diff --git a/research/object_detection/models/ssd_mobilenet_v1_ppn_feature_extractor.py b/research/object_detection/object_detection/models/ssd_mobilenet_v1_ppn_feature_extractor.py similarity index 98% rename from research/object_detection/models/ssd_mobilenet_v1_ppn_feature_extractor.py rename to research/object_detection/object_detection/models/ssd_mobilenet_v1_ppn_feature_extractor.py index 34e9cb4e88b..fce285c4649 100644 --- a/research/object_detection/models/ssd_mobilenet_v1_ppn_feature_extractor.py +++ b/research/object_detection/object_detection/models/ssd_mobilenet_v1_ppn_feature_extractor.py @@ -22,7 +22,7 @@ from object_detection.utils import context_manager from object_detection.utils import ops from object_detection.utils import shape_utils -from nets import mobilenet_v1 +from slim.nets import mobilenet_v1 slim = tf.contrib.slim diff --git a/research/object_detection/models/ssd_mobilenet_v1_ppn_feature_extractor_test.py b/research/object_detection/object_detection/models/ssd_mobilenet_v1_ppn_feature_extractor_test.py similarity index 100% rename from research/object_detection/models/ssd_mobilenet_v1_ppn_feature_extractor_test.py rename to research/object_detection/object_detection/models/ssd_mobilenet_v1_ppn_feature_extractor_test.py diff --git a/research/object_detection/models/ssd_mobilenet_v2_feature_extractor.py b/research/object_detection/object_detection/models/ssd_mobilenet_v2_feature_extractor.py similarity index 98% rename from research/object_detection/models/ssd_mobilenet_v2_feature_extractor.py rename to research/object_detection/object_detection/models/ssd_mobilenet_v2_feature_extractor.py index 014b93a8e29..b6468e06177 100644 --- a/research/object_detection/models/ssd_mobilenet_v2_feature_extractor.py +++ b/research/object_detection/object_detection/models/ssd_mobilenet_v2_feature_extractor.py @@ -22,8 +22,8 @@ from object_detection.utils import context_manager from object_detection.utils import ops from object_detection.utils import shape_utils -from nets.mobilenet import mobilenet -from nets.mobilenet import mobilenet_v2 +from slim.nets.mobilenet import mobilenet +from slim.nets.mobilenet import mobilenet_v2 slim = tf.contrib.slim diff --git a/research/object_detection/models/ssd_mobilenet_v2_feature_extractor_test.py b/research/object_detection/object_detection/models/ssd_mobilenet_v2_feature_extractor_test.py similarity index 100% rename from research/object_detection/models/ssd_mobilenet_v2_feature_extractor_test.py rename to research/object_detection/object_detection/models/ssd_mobilenet_v2_feature_extractor_test.py diff --git a/research/object_detection/models/ssd_resnet_v1_fpn_feature_extractor.py b/research/object_detection/object_detection/models/ssd_resnet_v1_fpn_feature_extractor.py similarity index 99% rename from research/object_detection/models/ssd_resnet_v1_fpn_feature_extractor.py rename to research/object_detection/object_detection/models/ssd_resnet_v1_fpn_feature_extractor.py index 65bda3f4fc3..a0a705e2081 100644 --- a/research/object_detection/models/ssd_resnet_v1_fpn_feature_extractor.py +++ b/research/object_detection/object_detection/models/ssd_resnet_v1_fpn_feature_extractor.py @@ -24,7 +24,7 @@ from object_detection.utils import context_manager from object_detection.utils import ops from object_detection.utils import shape_utils -from nets import resnet_v1 +from slim.nets import resnet_v1 slim = tf.contrib.slim diff --git a/research/object_detection/models/ssd_resnet_v1_fpn_feature_extractor_test.py b/research/object_detection/object_detection/models/ssd_resnet_v1_fpn_feature_extractor_test.py similarity index 100% rename from research/object_detection/models/ssd_resnet_v1_fpn_feature_extractor_test.py rename to research/object_detection/object_detection/models/ssd_resnet_v1_fpn_feature_extractor_test.py diff --git a/research/object_detection/models/ssd_resnet_v1_fpn_feature_extractor_testbase.py b/research/object_detection/object_detection/models/ssd_resnet_v1_fpn_feature_extractor_testbase.py similarity index 100% rename from research/object_detection/models/ssd_resnet_v1_fpn_feature_extractor_testbase.py rename to research/object_detection/object_detection/models/ssd_resnet_v1_fpn_feature_extractor_testbase.py diff --git a/research/object_detection/models/ssd_resnet_v1_ppn_feature_extractor.py b/research/object_detection/object_detection/models/ssd_resnet_v1_ppn_feature_extractor.py similarity index 99% rename from research/object_detection/models/ssd_resnet_v1_ppn_feature_extractor.py rename to research/object_detection/object_detection/models/ssd_resnet_v1_ppn_feature_extractor.py index 13422503c06..f4023a86044 100644 --- a/research/object_detection/models/ssd_resnet_v1_ppn_feature_extractor.py +++ b/research/object_detection/object_detection/models/ssd_resnet_v1_ppn_feature_extractor.py @@ -21,7 +21,7 @@ from object_detection.utils import context_manager from object_detection.utils import ops from object_detection.utils import shape_utils -from nets import resnet_v1 +from slim.nets import resnet_v1 slim = tf.contrib.slim diff --git a/research/object_detection/models/ssd_resnet_v1_ppn_feature_extractor_test.py b/research/object_detection/object_detection/models/ssd_resnet_v1_ppn_feature_extractor_test.py similarity index 100% rename from research/object_detection/models/ssd_resnet_v1_ppn_feature_extractor_test.py rename to research/object_detection/object_detection/models/ssd_resnet_v1_ppn_feature_extractor_test.py diff --git a/research/object_detection/models/ssd_resnet_v1_ppn_feature_extractor_testbase.py b/research/object_detection/object_detection/models/ssd_resnet_v1_ppn_feature_extractor_testbase.py similarity index 100% rename from research/object_detection/models/ssd_resnet_v1_ppn_feature_extractor_testbase.py rename to research/object_detection/object_detection/models/ssd_resnet_v1_ppn_feature_extractor_testbase.py diff --git a/research/object_detection/protos/__init__.py b/research/object_detection/object_detection/protos/__init__.py similarity index 100% rename from research/object_detection/protos/__init__.py rename to research/object_detection/object_detection/protos/__init__.py diff --git a/research/object_detection/protos/anchor_generator.proto b/research/object_detection/object_detection/protos/anchor_generator.proto similarity index 100% rename from research/object_detection/protos/anchor_generator.proto rename to research/object_detection/object_detection/protos/anchor_generator.proto diff --git a/research/object_detection/object_detection/protos/anchor_generator_pb2.py b/research/object_detection/object_detection/protos/anchor_generator_pb2.py new file mode 100644 index 00000000000..646b771e359 --- /dev/null +++ b/research/object_detection/object_detection/protos/anchor_generator_pb2.py @@ -0,0 +1,102 @@ +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: object_detection/protos/anchor_generator.proto + +import sys +_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) +from google.protobuf import descriptor as _descriptor +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +from google.protobuf import descriptor_pb2 +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + +from object_detection.protos import grid_anchor_generator_pb2 as object__detection_dot_protos_dot_grid__anchor__generator__pb2 +from object_detection.protos import ssd_anchor_generator_pb2 as object__detection_dot_protos_dot_ssd__anchor__generator__pb2 +from object_detection.protos import multiscale_anchor_generator_pb2 as object__detection_dot_protos_dot_multiscale__anchor__generator__pb2 + + +DESCRIPTOR = _descriptor.FileDescriptor( + name='object_detection/protos/anchor_generator.proto', + package='object_detection.protos', + syntax='proto2', + serialized_pb=_b('\n.object_detection/protos/anchor_generator.proto\x12\x17object_detection.protos\x1a\x33object_detection/protos/grid_anchor_generator.proto\x1a\x32object_detection/protos/ssd_anchor_generator.proto\x1a\x39object_detection/protos/multiscale_anchor_generator.proto\"\xa2\x02\n\x0f\x41nchorGenerator\x12M\n\x15grid_anchor_generator\x18\x01 \x01(\x0b\x32,.object_detection.protos.GridAnchorGeneratorH\x00\x12K\n\x14ssd_anchor_generator\x18\x02 \x01(\x0b\x32+.object_detection.protos.SsdAnchorGeneratorH\x00\x12Y\n\x1bmultiscale_anchor_generator\x18\x03 \x01(\x0b\x32\x32.object_detection.protos.MultiscaleAnchorGeneratorH\x00\x42\x18\n\x16\x61nchor_generator_oneof') + , + dependencies=[object__detection_dot_protos_dot_grid__anchor__generator__pb2.DESCRIPTOR,object__detection_dot_protos_dot_ssd__anchor__generator__pb2.DESCRIPTOR,object__detection_dot_protos_dot_multiscale__anchor__generator__pb2.DESCRIPTOR,]) + + + + +_ANCHORGENERATOR = _descriptor.Descriptor( + name='AnchorGenerator', + full_name='object_detection.protos.AnchorGenerator', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='grid_anchor_generator', full_name='object_detection.protos.AnchorGenerator.grid_anchor_generator', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='ssd_anchor_generator', full_name='object_detection.protos.AnchorGenerator.ssd_anchor_generator', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='multiscale_anchor_generator', full_name='object_detection.protos.AnchorGenerator.multiscale_anchor_generator', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + _descriptor.OneofDescriptor( + name='anchor_generator_oneof', full_name='object_detection.protos.AnchorGenerator.anchor_generator_oneof', + index=0, containing_type=None, fields=[]), + ], + serialized_start=240, + serialized_end=530, +) + +_ANCHORGENERATOR.fields_by_name['grid_anchor_generator'].message_type = object__detection_dot_protos_dot_grid__anchor__generator__pb2._GRIDANCHORGENERATOR +_ANCHORGENERATOR.fields_by_name['ssd_anchor_generator'].message_type = object__detection_dot_protos_dot_ssd__anchor__generator__pb2._SSDANCHORGENERATOR +_ANCHORGENERATOR.fields_by_name['multiscale_anchor_generator'].message_type = object__detection_dot_protos_dot_multiscale__anchor__generator__pb2._MULTISCALEANCHORGENERATOR +_ANCHORGENERATOR.oneofs_by_name['anchor_generator_oneof'].fields.append( + _ANCHORGENERATOR.fields_by_name['grid_anchor_generator']) +_ANCHORGENERATOR.fields_by_name['grid_anchor_generator'].containing_oneof = _ANCHORGENERATOR.oneofs_by_name['anchor_generator_oneof'] +_ANCHORGENERATOR.oneofs_by_name['anchor_generator_oneof'].fields.append( + _ANCHORGENERATOR.fields_by_name['ssd_anchor_generator']) +_ANCHORGENERATOR.fields_by_name['ssd_anchor_generator'].containing_oneof = _ANCHORGENERATOR.oneofs_by_name['anchor_generator_oneof'] +_ANCHORGENERATOR.oneofs_by_name['anchor_generator_oneof'].fields.append( + _ANCHORGENERATOR.fields_by_name['multiscale_anchor_generator']) +_ANCHORGENERATOR.fields_by_name['multiscale_anchor_generator'].containing_oneof = _ANCHORGENERATOR.oneofs_by_name['anchor_generator_oneof'] +DESCRIPTOR.message_types_by_name['AnchorGenerator'] = _ANCHORGENERATOR +_sym_db.RegisterFileDescriptor(DESCRIPTOR) + +AnchorGenerator = _reflection.GeneratedProtocolMessageType('AnchorGenerator', (_message.Message,), dict( + DESCRIPTOR = _ANCHORGENERATOR, + __module__ = 'object_detection.protos.anchor_generator_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.AnchorGenerator) + )) +_sym_db.RegisterMessage(AnchorGenerator) + + +# @@protoc_insertion_point(module_scope) diff --git a/research/object_detection/protos/argmax_matcher.proto b/research/object_detection/object_detection/protos/argmax_matcher.proto similarity index 100% rename from research/object_detection/protos/argmax_matcher.proto rename to research/object_detection/object_detection/protos/argmax_matcher.proto diff --git a/research/object_detection/object_detection/protos/argmax_matcher_pb2.py b/research/object_detection/object_detection/protos/argmax_matcher_pb2.py new file mode 100644 index 00000000000..08b708248d1 --- /dev/null +++ b/research/object_detection/object_detection/protos/argmax_matcher_pb2.py @@ -0,0 +1,104 @@ +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: object_detection/protos/argmax_matcher.proto + +import sys +_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) +from google.protobuf import descriptor as _descriptor +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +from google.protobuf import descriptor_pb2 +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor.FileDescriptor( + name='object_detection/protos/argmax_matcher.proto', + package='object_detection.protos', + syntax='proto2', + serialized_pb=_b('\n,object_detection/protos/argmax_matcher.proto\x12\x17object_detection.protos\"\xec\x01\n\rArgMaxMatcher\x12\x1e\n\x11matched_threshold\x18\x01 \x01(\x02:\x03\x30.5\x12 \n\x13unmatched_threshold\x18\x02 \x01(\x02:\x03\x30.5\x12 \n\x11ignore_thresholds\x18\x03 \x01(\x08:\x05\x66\x61lse\x12,\n\x1enegatives_lower_than_unmatched\x18\x04 \x01(\x08:\x04true\x12\'\n\x18\x66orce_match_for_each_row\x18\x05 \x01(\x08:\x05\x66\x61lse\x12 \n\x11use_matmul_gather\x18\x06 \x01(\x08:\x05\x66\x61lse') +) + + + + +_ARGMAXMATCHER = _descriptor.Descriptor( + name='ArgMaxMatcher', + full_name='object_detection.protos.ArgMaxMatcher', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='matched_threshold', full_name='object_detection.protos.ArgMaxMatcher.matched_threshold', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='unmatched_threshold', full_name='object_detection.protos.ArgMaxMatcher.unmatched_threshold', index=1, + number=2, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='ignore_thresholds', full_name='object_detection.protos.ArgMaxMatcher.ignore_thresholds', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='negatives_lower_than_unmatched', full_name='object_detection.protos.ArgMaxMatcher.negatives_lower_than_unmatched', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='force_match_for_each_row', full_name='object_detection.protos.ArgMaxMatcher.force_match_for_each_row', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='use_matmul_gather', full_name='object_detection.protos.ArgMaxMatcher.use_matmul_gather', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=74, + serialized_end=310, +) + +DESCRIPTOR.message_types_by_name['ArgMaxMatcher'] = _ARGMAXMATCHER +_sym_db.RegisterFileDescriptor(DESCRIPTOR) + +ArgMaxMatcher = _reflection.GeneratedProtocolMessageType('ArgMaxMatcher', (_message.Message,), dict( + DESCRIPTOR = _ARGMAXMATCHER, + __module__ = 'object_detection.protos.argmax_matcher_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.ArgMaxMatcher) + )) +_sym_db.RegisterMessage(ArgMaxMatcher) + + +# @@protoc_insertion_point(module_scope) diff --git a/research/object_detection/protos/bipartite_matcher.proto b/research/object_detection/object_detection/protos/bipartite_matcher.proto similarity index 100% rename from research/object_detection/protos/bipartite_matcher.proto rename to research/object_detection/object_detection/protos/bipartite_matcher.proto diff --git a/research/object_detection/object_detection/protos/bipartite_matcher_pb2.py b/research/object_detection/object_detection/protos/bipartite_matcher_pb2.py new file mode 100644 index 00000000000..c15e4cfd49b --- /dev/null +++ b/research/object_detection/object_detection/protos/bipartite_matcher_pb2.py @@ -0,0 +1,69 @@ +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: object_detection/protos/bipartite_matcher.proto + +import sys +_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) +from google.protobuf import descriptor as _descriptor +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +from google.protobuf import descriptor_pb2 +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor.FileDescriptor( + name='object_detection/protos/bipartite_matcher.proto', + package='object_detection.protos', + syntax='proto2', + serialized_pb=_b('\n/object_detection/protos/bipartite_matcher.proto\x12\x17object_detection.protos\"4\n\x10\x42ipartiteMatcher\x12 \n\x11use_matmul_gather\x18\x06 \x01(\x08:\x05\x66\x61lse') +) + + + + +_BIPARTITEMATCHER = _descriptor.Descriptor( + name='BipartiteMatcher', + full_name='object_detection.protos.BipartiteMatcher', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='use_matmul_gather', full_name='object_detection.protos.BipartiteMatcher.use_matmul_gather', index=0, + 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=76, + serialized_end=128, +) + +DESCRIPTOR.message_types_by_name['BipartiteMatcher'] = _BIPARTITEMATCHER +_sym_db.RegisterFileDescriptor(DESCRIPTOR) + +BipartiteMatcher = _reflection.GeneratedProtocolMessageType('BipartiteMatcher', (_message.Message,), dict( + DESCRIPTOR = _BIPARTITEMATCHER, + __module__ = 'object_detection.protos.bipartite_matcher_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.BipartiteMatcher) + )) +_sym_db.RegisterMessage(BipartiteMatcher) + + +# @@protoc_insertion_point(module_scope) diff --git a/research/object_detection/protos/box_coder.proto b/research/object_detection/object_detection/protos/box_coder.proto similarity index 100% rename from research/object_detection/protos/box_coder.proto rename to research/object_detection/object_detection/protos/box_coder.proto diff --git a/research/object_detection/object_detection/protos/box_coder_pb2.py b/research/object_detection/object_detection/protos/box_coder_pb2.py new file mode 100644 index 00000000000..6e91d095467 --- /dev/null +++ b/research/object_detection/object_detection/protos/box_coder_pb2.py @@ -0,0 +1,114 @@ +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: object_detection/protos/box_coder.proto + +import sys +_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) +from google.protobuf import descriptor as _descriptor +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +from google.protobuf import descriptor_pb2 +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + +from object_detection.protos import faster_rcnn_box_coder_pb2 as object__detection_dot_protos_dot_faster__rcnn__box__coder__pb2 +from object_detection.protos import keypoint_box_coder_pb2 as object__detection_dot_protos_dot_keypoint__box__coder__pb2 +from object_detection.protos import mean_stddev_box_coder_pb2 as object__detection_dot_protos_dot_mean__stddev__box__coder__pb2 +from object_detection.protos import square_box_coder_pb2 as object__detection_dot_protos_dot_square__box__coder__pb2 + + +DESCRIPTOR = _descriptor.FileDescriptor( + name='object_detection/protos/box_coder.proto', + package='object_detection.protos', + syntax='proto2', + serialized_pb=_b('\n\'object_detection/protos/box_coder.proto\x12\x17object_detection.protos\x1a\x33object_detection/protos/faster_rcnn_box_coder.proto\x1a\x30object_detection/protos/keypoint_box_coder.proto\x1a\x33object_detection/protos/mean_stddev_box_coder.proto\x1a.object_detection/protos/square_box_coder.proto\"\xc7\x02\n\x08\x42oxCoder\x12L\n\x15\x66\x61ster_rcnn_box_coder\x18\x01 \x01(\x0b\x32+.object_detection.protos.FasterRcnnBoxCoderH\x00\x12L\n\x15mean_stddev_box_coder\x18\x02 \x01(\x0b\x32+.object_detection.protos.MeanStddevBoxCoderH\x00\x12\x43\n\x10square_box_coder\x18\x03 \x01(\x0b\x32\'.object_detection.protos.SquareBoxCoderH\x00\x12G\n\x12keypoint_box_coder\x18\x04 \x01(\x0b\x32).object_detection.protos.KeypointBoxCoderH\x00\x42\x11\n\x0f\x62ox_coder_oneof') + , + dependencies=[object__detection_dot_protos_dot_faster__rcnn__box__coder__pb2.DESCRIPTOR,object__detection_dot_protos_dot_keypoint__box__coder__pb2.DESCRIPTOR,object__detection_dot_protos_dot_mean__stddev__box__coder__pb2.DESCRIPTOR,object__detection_dot_protos_dot_square__box__coder__pb2.DESCRIPTOR,]) + + + + +_BOXCODER = _descriptor.Descriptor( + name='BoxCoder', + full_name='object_detection.protos.BoxCoder', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='faster_rcnn_box_coder', full_name='object_detection.protos.BoxCoder.faster_rcnn_box_coder', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='mean_stddev_box_coder', full_name='object_detection.protos.BoxCoder.mean_stddev_box_coder', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='square_box_coder', full_name='object_detection.protos.BoxCoder.square_box_coder', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='keypoint_box_coder', full_name='object_detection.protos.BoxCoder.keypoint_box_coder', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + _descriptor.OneofDescriptor( + name='box_coder_oneof', full_name='object_detection.protos.BoxCoder.box_coder_oneof', + index=0, containing_type=None, fields=[]), + ], + serialized_start=273, + serialized_end=600, +) + +_BOXCODER.fields_by_name['faster_rcnn_box_coder'].message_type = object__detection_dot_protos_dot_faster__rcnn__box__coder__pb2._FASTERRCNNBOXCODER +_BOXCODER.fields_by_name['mean_stddev_box_coder'].message_type = object__detection_dot_protos_dot_mean__stddev__box__coder__pb2._MEANSTDDEVBOXCODER +_BOXCODER.fields_by_name['square_box_coder'].message_type = object__detection_dot_protos_dot_square__box__coder__pb2._SQUAREBOXCODER +_BOXCODER.fields_by_name['keypoint_box_coder'].message_type = object__detection_dot_protos_dot_keypoint__box__coder__pb2._KEYPOINTBOXCODER +_BOXCODER.oneofs_by_name['box_coder_oneof'].fields.append( + _BOXCODER.fields_by_name['faster_rcnn_box_coder']) +_BOXCODER.fields_by_name['faster_rcnn_box_coder'].containing_oneof = _BOXCODER.oneofs_by_name['box_coder_oneof'] +_BOXCODER.oneofs_by_name['box_coder_oneof'].fields.append( + _BOXCODER.fields_by_name['mean_stddev_box_coder']) +_BOXCODER.fields_by_name['mean_stddev_box_coder'].containing_oneof = _BOXCODER.oneofs_by_name['box_coder_oneof'] +_BOXCODER.oneofs_by_name['box_coder_oneof'].fields.append( + _BOXCODER.fields_by_name['square_box_coder']) +_BOXCODER.fields_by_name['square_box_coder'].containing_oneof = _BOXCODER.oneofs_by_name['box_coder_oneof'] +_BOXCODER.oneofs_by_name['box_coder_oneof'].fields.append( + _BOXCODER.fields_by_name['keypoint_box_coder']) +_BOXCODER.fields_by_name['keypoint_box_coder'].containing_oneof = _BOXCODER.oneofs_by_name['box_coder_oneof'] +DESCRIPTOR.message_types_by_name['BoxCoder'] = _BOXCODER +_sym_db.RegisterFileDescriptor(DESCRIPTOR) + +BoxCoder = _reflection.GeneratedProtocolMessageType('BoxCoder', (_message.Message,), dict( + DESCRIPTOR = _BOXCODER, + __module__ = 'object_detection.protos.box_coder_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.BoxCoder) + )) +_sym_db.RegisterMessage(BoxCoder) + + +# @@protoc_insertion_point(module_scope) diff --git a/research/object_detection/protos/box_predictor.proto b/research/object_detection/object_detection/protos/box_predictor.proto similarity index 100% rename from research/object_detection/protos/box_predictor.proto rename to research/object_detection/object_detection/protos/box_predictor.proto diff --git a/research/object_detection/object_detection/protos/box_predictor_pb2.py b/research/object_detection/object_detection/protos/box_predictor_pb2.py new file mode 100644 index 00000000000..8fc43568441 --- /dev/null +++ b/research/object_detection/object_detection/protos/box_predictor_pb2.py @@ -0,0 +1,524 @@ +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: object_detection/protos/box_predictor.proto + +import sys +_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) +from google.protobuf import descriptor as _descriptor +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +from google.protobuf import descriptor_pb2 +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + +from object_detection.protos import hyperparams_pb2 as object__detection_dot_protos_dot_hyperparams__pb2 + + +DESCRIPTOR = _descriptor.FileDescriptor( + name='object_detection/protos/box_predictor.proto', + package='object_detection.protos', + syntax='proto2', + serialized_pb=_b('\n+object_detection/protos/box_predictor.proto\x12\x17object_detection.protos\x1a)object_detection/protos/hyperparams.proto\"\x90\x03\n\x0c\x42oxPredictor\x12Y\n\x1b\x63onvolutional_box_predictor\x18\x01 \x01(\x0b\x32\x32.object_detection.protos.ConvolutionalBoxPredictorH\x00\x12P\n\x17mask_rcnn_box_predictor\x18\x02 \x01(\x0b\x32-.object_detection.protos.MaskRCNNBoxPredictorH\x00\x12G\n\x12rfcn_box_predictor\x18\x03 \x01(\x0b\x32).object_detection.protos.RfcnBoxPredictorH\x00\x12s\n)weight_shared_convolutional_box_predictor\x18\x04 \x01(\x0b\x32>.object_detection.protos.WeightSharedConvolutionalBoxPredictorH\x00\x42\x15\n\x13\x62ox_predictor_oneof\"\x90\x03\n\x19\x43onvolutionalBoxPredictor\x12>\n\x10\x63onv_hyperparams\x18\x01 \x01(\x0b\x32$.object_detection.protos.Hyperparams\x12\x14\n\tmin_depth\x18\x02 \x01(\x05:\x01\x30\x12\x14\n\tmax_depth\x18\x03 \x01(\x05:\x01\x30\x12&\n\x1bnum_layers_before_predictor\x18\x04 \x01(\x05:\x01\x30\x12\x19\n\x0buse_dropout\x18\x05 \x01(\x08:\x04true\x12%\n\x18\x64ropout_keep_probability\x18\x06 \x01(\x02:\x03\x30.8\x12\x16\n\x0bkernel_size\x18\x07 \x01(\x05:\x01\x31\x12\x18\n\rbox_code_size\x18\x08 \x01(\x05:\x01\x34\x12&\n\x17\x61pply_sigmoid_to_scores\x18\t \x01(\x08:\x05\x66\x61lse\x12%\n\x1a\x63lass_prediction_bias_init\x18\n \x01(\x02:\x01\x30\x12\x1c\n\ruse_depthwise\x18\x0b \x01(\x08:\x05\x66\x61lse\"\xe4\x02\n%WeightSharedConvolutionalBoxPredictor\x12>\n\x10\x63onv_hyperparams\x18\x01 \x01(\x0b\x32$.object_detection.protos.Hyperparams\x12&\n\x1bnum_layers_before_predictor\x18\x04 \x01(\x05:\x01\x30\x12\x10\n\x05\x64\x65pth\x18\x02 \x01(\x05:\x01\x30\x12\x16\n\x0bkernel_size\x18\x07 \x01(\x05:\x01\x33\x12\x18\n\rbox_code_size\x18\x08 \x01(\x05:\x01\x34\x12%\n\x1a\x63lass_prediction_bias_init\x18\n \x01(\x02:\x01\x30\x12\x1a\n\x0buse_dropout\x18\x0b \x01(\x08:\x05\x66\x61lse\x12%\n\x18\x64ropout_keep_probability\x18\x0c \x01(\x02:\x03\x30.8\x12%\n\x16share_prediction_tower\x18\r \x01(\x08:\x05\x66\x61lse\"\x92\x04\n\x14MaskRCNNBoxPredictor\x12<\n\x0e\x66\x63_hyperparams\x18\x01 \x01(\x0b\x32$.object_detection.protos.Hyperparams\x12\x1a\n\x0buse_dropout\x18\x02 \x01(\x08:\x05\x66\x61lse\x12%\n\x18\x64ropout_keep_probability\x18\x03 \x01(\x02:\x03\x30.5\x12\x18\n\rbox_code_size\x18\x04 \x01(\x05:\x01\x34\x12>\n\x10\x63onv_hyperparams\x18\x05 \x01(\x0b\x32$.object_detection.protos.Hyperparams\x12%\n\x16predict_instance_masks\x18\x06 \x01(\x08:\x05\x66\x61lse\x12\'\n\x1amask_prediction_conv_depth\x18\x07 \x01(\x05:\x03\x32\x35\x36\x12 \n\x11predict_keypoints\x18\x08 \x01(\x08:\x05\x66\x61lse\x12\x17\n\x0bmask_height\x18\t \x01(\x05:\x02\x31\x35\x12\x16\n\nmask_width\x18\n \x01(\x05:\x02\x31\x35\x12*\n\x1fmask_prediction_num_conv_layers\x18\x0b \x01(\x05:\x01\x32\x12\'\n\x18masks_are_class_agnostic\x18\x0c \x01(\x08:\x05\x66\x61lse\x12\'\n\x18share_box_across_classes\x18\r \x01(\x08:\x05\x66\x61lse\"\xf9\x01\n\x10RfcnBoxPredictor\x12>\n\x10\x63onv_hyperparams\x18\x01 \x01(\x0b\x32$.object_detection.protos.Hyperparams\x12\"\n\x17num_spatial_bins_height\x18\x02 \x01(\x05:\x01\x33\x12!\n\x16num_spatial_bins_width\x18\x03 \x01(\x05:\x01\x33\x12\x13\n\x05\x64\x65pth\x18\x04 \x01(\x05:\x04\x31\x30\x32\x34\x12\x18\n\rbox_code_size\x18\x05 \x01(\x05:\x01\x34\x12\x17\n\x0b\x63rop_height\x18\x06 \x01(\x05:\x02\x31\x32\x12\x16\n\ncrop_width\x18\x07 \x01(\x05:\x02\x31\x32') + , + dependencies=[object__detection_dot_protos_dot_hyperparams__pb2.DESCRIPTOR,]) + + + + +_BOXPREDICTOR = _descriptor.Descriptor( + name='BoxPredictor', + full_name='object_detection.protos.BoxPredictor', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='convolutional_box_predictor', full_name='object_detection.protos.BoxPredictor.convolutional_box_predictor', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='mask_rcnn_box_predictor', full_name='object_detection.protos.BoxPredictor.mask_rcnn_box_predictor', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='rfcn_box_predictor', full_name='object_detection.protos.BoxPredictor.rfcn_box_predictor', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='weight_shared_convolutional_box_predictor', full_name='object_detection.protos.BoxPredictor.weight_shared_convolutional_box_predictor', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + _descriptor.OneofDescriptor( + name='box_predictor_oneof', full_name='object_detection.protos.BoxPredictor.box_predictor_oneof', + index=0, containing_type=None, fields=[]), + ], + serialized_start=116, + serialized_end=516, +) + + +_CONVOLUTIONALBOXPREDICTOR = _descriptor.Descriptor( + name='ConvolutionalBoxPredictor', + full_name='object_detection.protos.ConvolutionalBoxPredictor', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='conv_hyperparams', full_name='object_detection.protos.ConvolutionalBoxPredictor.conv_hyperparams', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='min_depth', full_name='object_detection.protos.ConvolutionalBoxPredictor.min_depth', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_depth', full_name='object_detection.protos.ConvolutionalBoxPredictor.max_depth', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='num_layers_before_predictor', full_name='object_detection.protos.ConvolutionalBoxPredictor.num_layers_before_predictor', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='use_dropout', full_name='object_detection.protos.ConvolutionalBoxPredictor.use_dropout', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='dropout_keep_probability', full_name='object_detection.protos.ConvolutionalBoxPredictor.dropout_keep_probability', index=5, + number=6, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(0.8), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='kernel_size', full_name='object_detection.protos.ConvolutionalBoxPredictor.kernel_size', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='box_code_size', full_name='object_detection.protos.ConvolutionalBoxPredictor.box_code_size', index=7, + number=8, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=4, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='apply_sigmoid_to_scores', full_name='object_detection.protos.ConvolutionalBoxPredictor.apply_sigmoid_to_scores', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='class_prediction_bias_init', full_name='object_detection.protos.ConvolutionalBoxPredictor.class_prediction_bias_init', index=9, + number=10, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='use_depthwise', full_name='object_detection.protos.ConvolutionalBoxPredictor.use_depthwise', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=519, + serialized_end=919, +) + + +_WEIGHTSHAREDCONVOLUTIONALBOXPREDICTOR = _descriptor.Descriptor( + name='WeightSharedConvolutionalBoxPredictor', + full_name='object_detection.protos.WeightSharedConvolutionalBoxPredictor', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='conv_hyperparams', full_name='object_detection.protos.WeightSharedConvolutionalBoxPredictor.conv_hyperparams', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='num_layers_before_predictor', full_name='object_detection.protos.WeightSharedConvolutionalBoxPredictor.num_layers_before_predictor', index=1, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='depth', full_name='object_detection.protos.WeightSharedConvolutionalBoxPredictor.depth', index=2, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='kernel_size', full_name='object_detection.protos.WeightSharedConvolutionalBoxPredictor.kernel_size', index=3, + number=7, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=3, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='box_code_size', full_name='object_detection.protos.WeightSharedConvolutionalBoxPredictor.box_code_size', index=4, + number=8, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=4, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='class_prediction_bias_init', full_name='object_detection.protos.WeightSharedConvolutionalBoxPredictor.class_prediction_bias_init', index=5, + number=10, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='use_dropout', full_name='object_detection.protos.WeightSharedConvolutionalBoxPredictor.use_dropout', index=6, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='dropout_keep_probability', full_name='object_detection.protos.WeightSharedConvolutionalBoxPredictor.dropout_keep_probability', index=7, + number=12, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(0.8), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='share_prediction_tower', full_name='object_detection.protos.WeightSharedConvolutionalBoxPredictor.share_prediction_tower', index=8, + number=13, 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=922, + serialized_end=1278, +) + + +_MASKRCNNBOXPREDICTOR = _descriptor.Descriptor( + name='MaskRCNNBoxPredictor', + full_name='object_detection.protos.MaskRCNNBoxPredictor', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='fc_hyperparams', full_name='object_detection.protos.MaskRCNNBoxPredictor.fc_hyperparams', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='use_dropout', full_name='object_detection.protos.MaskRCNNBoxPredictor.use_dropout', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='dropout_keep_probability', full_name='object_detection.protos.MaskRCNNBoxPredictor.dropout_keep_probability', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='box_code_size', full_name='object_detection.protos.MaskRCNNBoxPredictor.box_code_size', index=3, + number=4, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=4, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='conv_hyperparams', full_name='object_detection.protos.MaskRCNNBoxPredictor.conv_hyperparams', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='predict_instance_masks', full_name='object_detection.protos.MaskRCNNBoxPredictor.predict_instance_masks', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='mask_prediction_conv_depth', full_name='object_detection.protos.MaskRCNNBoxPredictor.mask_prediction_conv_depth', index=6, + number=7, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=256, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='predict_keypoints', full_name='object_detection.protos.MaskRCNNBoxPredictor.predict_keypoints', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='mask_height', full_name='object_detection.protos.MaskRCNNBoxPredictor.mask_height', index=8, + number=9, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=15, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='mask_width', full_name='object_detection.protos.MaskRCNNBoxPredictor.mask_width', index=9, + number=10, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=15, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='mask_prediction_num_conv_layers', full_name='object_detection.protos.MaskRCNNBoxPredictor.mask_prediction_num_conv_layers', index=10, + number=11, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='masks_are_class_agnostic', full_name='object_detection.protos.MaskRCNNBoxPredictor.masks_are_class_agnostic', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='share_box_across_classes', full_name='object_detection.protos.MaskRCNNBoxPredictor.share_box_across_classes', index=12, + number=13, 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=1281, + serialized_end=1811, +) + + +_RFCNBOXPREDICTOR = _descriptor.Descriptor( + name='RfcnBoxPredictor', + full_name='object_detection.protos.RfcnBoxPredictor', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='conv_hyperparams', full_name='object_detection.protos.RfcnBoxPredictor.conv_hyperparams', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='num_spatial_bins_height', full_name='object_detection.protos.RfcnBoxPredictor.num_spatial_bins_height', index=1, + number=2, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=3, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='num_spatial_bins_width', full_name='object_detection.protos.RfcnBoxPredictor.num_spatial_bins_width', index=2, + number=3, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=3, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='depth', full_name='object_detection.protos.RfcnBoxPredictor.depth', index=3, + number=4, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=1024, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='box_code_size', full_name='object_detection.protos.RfcnBoxPredictor.box_code_size', index=4, + number=5, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=4, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='crop_height', full_name='object_detection.protos.RfcnBoxPredictor.crop_height', index=5, + number=6, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=12, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='crop_width', full_name='object_detection.protos.RfcnBoxPredictor.crop_width', index=6, + number=7, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=12, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=1814, + serialized_end=2063, +) + +_BOXPREDICTOR.fields_by_name['convolutional_box_predictor'].message_type = _CONVOLUTIONALBOXPREDICTOR +_BOXPREDICTOR.fields_by_name['mask_rcnn_box_predictor'].message_type = _MASKRCNNBOXPREDICTOR +_BOXPREDICTOR.fields_by_name['rfcn_box_predictor'].message_type = _RFCNBOXPREDICTOR +_BOXPREDICTOR.fields_by_name['weight_shared_convolutional_box_predictor'].message_type = _WEIGHTSHAREDCONVOLUTIONALBOXPREDICTOR +_BOXPREDICTOR.oneofs_by_name['box_predictor_oneof'].fields.append( + _BOXPREDICTOR.fields_by_name['convolutional_box_predictor']) +_BOXPREDICTOR.fields_by_name['convolutional_box_predictor'].containing_oneof = _BOXPREDICTOR.oneofs_by_name['box_predictor_oneof'] +_BOXPREDICTOR.oneofs_by_name['box_predictor_oneof'].fields.append( + _BOXPREDICTOR.fields_by_name['mask_rcnn_box_predictor']) +_BOXPREDICTOR.fields_by_name['mask_rcnn_box_predictor'].containing_oneof = _BOXPREDICTOR.oneofs_by_name['box_predictor_oneof'] +_BOXPREDICTOR.oneofs_by_name['box_predictor_oneof'].fields.append( + _BOXPREDICTOR.fields_by_name['rfcn_box_predictor']) +_BOXPREDICTOR.fields_by_name['rfcn_box_predictor'].containing_oneof = _BOXPREDICTOR.oneofs_by_name['box_predictor_oneof'] +_BOXPREDICTOR.oneofs_by_name['box_predictor_oneof'].fields.append( + _BOXPREDICTOR.fields_by_name['weight_shared_convolutional_box_predictor']) +_BOXPREDICTOR.fields_by_name['weight_shared_convolutional_box_predictor'].containing_oneof = _BOXPREDICTOR.oneofs_by_name['box_predictor_oneof'] +_CONVOLUTIONALBOXPREDICTOR.fields_by_name['conv_hyperparams'].message_type = object__detection_dot_protos_dot_hyperparams__pb2._HYPERPARAMS +_WEIGHTSHAREDCONVOLUTIONALBOXPREDICTOR.fields_by_name['conv_hyperparams'].message_type = object__detection_dot_protos_dot_hyperparams__pb2._HYPERPARAMS +_MASKRCNNBOXPREDICTOR.fields_by_name['fc_hyperparams'].message_type = object__detection_dot_protos_dot_hyperparams__pb2._HYPERPARAMS +_MASKRCNNBOXPREDICTOR.fields_by_name['conv_hyperparams'].message_type = object__detection_dot_protos_dot_hyperparams__pb2._HYPERPARAMS +_RFCNBOXPREDICTOR.fields_by_name['conv_hyperparams'].message_type = object__detection_dot_protos_dot_hyperparams__pb2._HYPERPARAMS +DESCRIPTOR.message_types_by_name['BoxPredictor'] = _BOXPREDICTOR +DESCRIPTOR.message_types_by_name['ConvolutionalBoxPredictor'] = _CONVOLUTIONALBOXPREDICTOR +DESCRIPTOR.message_types_by_name['WeightSharedConvolutionalBoxPredictor'] = _WEIGHTSHAREDCONVOLUTIONALBOXPREDICTOR +DESCRIPTOR.message_types_by_name['MaskRCNNBoxPredictor'] = _MASKRCNNBOXPREDICTOR +DESCRIPTOR.message_types_by_name['RfcnBoxPredictor'] = _RFCNBOXPREDICTOR +_sym_db.RegisterFileDescriptor(DESCRIPTOR) + +BoxPredictor = _reflection.GeneratedProtocolMessageType('BoxPredictor', (_message.Message,), dict( + DESCRIPTOR = _BOXPREDICTOR, + __module__ = 'object_detection.protos.box_predictor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.BoxPredictor) + )) +_sym_db.RegisterMessage(BoxPredictor) + +ConvolutionalBoxPredictor = _reflection.GeneratedProtocolMessageType('ConvolutionalBoxPredictor', (_message.Message,), dict( + DESCRIPTOR = _CONVOLUTIONALBOXPREDICTOR, + __module__ = 'object_detection.protos.box_predictor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.ConvolutionalBoxPredictor) + )) +_sym_db.RegisterMessage(ConvolutionalBoxPredictor) + +WeightSharedConvolutionalBoxPredictor = _reflection.GeneratedProtocolMessageType('WeightSharedConvolutionalBoxPredictor', (_message.Message,), dict( + DESCRIPTOR = _WEIGHTSHAREDCONVOLUTIONALBOXPREDICTOR, + __module__ = 'object_detection.protos.box_predictor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.WeightSharedConvolutionalBoxPredictor) + )) +_sym_db.RegisterMessage(WeightSharedConvolutionalBoxPredictor) + +MaskRCNNBoxPredictor = _reflection.GeneratedProtocolMessageType('MaskRCNNBoxPredictor', (_message.Message,), dict( + DESCRIPTOR = _MASKRCNNBOXPREDICTOR, + __module__ = 'object_detection.protos.box_predictor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.MaskRCNNBoxPredictor) + )) +_sym_db.RegisterMessage(MaskRCNNBoxPredictor) + +RfcnBoxPredictor = _reflection.GeneratedProtocolMessageType('RfcnBoxPredictor', (_message.Message,), dict( + DESCRIPTOR = _RFCNBOXPREDICTOR, + __module__ = 'object_detection.protos.box_predictor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.RfcnBoxPredictor) + )) +_sym_db.RegisterMessage(RfcnBoxPredictor) + + +# @@protoc_insertion_point(module_scope) diff --git a/research/object_detection/protos/eval.proto b/research/object_detection/object_detection/protos/eval.proto similarity index 100% rename from research/object_detection/protos/eval.proto rename to research/object_detection/object_detection/protos/eval.proto diff --git a/research/object_detection/object_detection/protos/eval_pb2.py b/research/object_detection/object_detection/protos/eval_pb2.py new file mode 100644 index 00000000000..89b5bf15984 --- /dev/null +++ b/research/object_detection/object_detection/protos/eval_pb2.py @@ -0,0 +1,209 @@ +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: object_detection/protos/eval.proto + +import sys +_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) +from google.protobuf import descriptor as _descriptor +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +from google.protobuf import descriptor_pb2 +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor.FileDescriptor( + name='object_detection/protos/eval.proto', + package='object_detection.protos', + syntax='proto2', + serialized_pb=_b('\n\"object_detection/protos/eval.proto\x12\x17object_detection.protos\"\xd8\x05\n\nEvalConfig\x12\x1e\n\x12num_visualizations\x18\x01 \x01(\r:\x02\x31\x30\x12\x1a\n\x0cnum_examples\x18\x02 \x01(\r:\x04\x35\x30\x30\x30\x12\x1f\n\x12\x65val_interval_secs\x18\x03 \x01(\r:\x03\x33\x30\x30\x12\x14\n\tmax_evals\x18\x04 \x01(\r:\x01\x30\x12\x19\n\nsave_graph\x18\x05 \x01(\x08:\x05\x66\x61lse\x12\"\n\x18visualization_export_dir\x18\x06 \x01(\t:\x00\x12\x15\n\x0b\x65val_master\x18\x07 \x01(\t:\x00\x12\x13\n\x0bmetrics_set\x18\x08 \x03(\t\x12\x15\n\x0b\x65xport_path\x18\t \x01(\t:\x00\x12!\n\x12ignore_groundtruth\x18\n \x01(\x08:\x05\x66\x61lse\x12\"\n\x13use_moving_averages\x18\x0b \x01(\x08:\x05\x66\x61lse\x12\"\n\x13\x65val_instance_masks\x18\x0c \x01(\x08:\x05\x66\x61lse\x12 \n\x13min_score_threshold\x18\r \x01(\x02:\x03\x30.5\x12&\n\x1amax_num_boxes_to_visualize\x18\x0e \x01(\x05:\x02\x32\x30\x12\x1a\n\x0bskip_scores\x18\x0f \x01(\x08:\x05\x66\x61lse\x12\x1a\n\x0bskip_labels\x18\x10 \x01(\x08:\x05\x66\x61lse\x12*\n\x1bvisualize_groundtruth_boxes\x18\x11 \x01(\x08:\x05\x66\x61lse\x12\x32\n#groundtruth_box_visualization_color\x18\x12 \x01(\t:\x05\x62lack\x12\x35\n&keep_image_id_for_visualization_export\x18\x13 \x01(\x08:\x05\x66\x61lse\x12$\n\x16retain_original_images\x18\x17 \x01(\x08:\x04true\x12+\n\x1cinclude_metrics_per_category\x18\x18 \x01(\x08:\x05\x66\x61lse') +) + + + + +_EVALCONFIG = _descriptor.Descriptor( + name='EvalConfig', + full_name='object_detection.protos.EvalConfig', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='num_visualizations', full_name='object_detection.protos.EvalConfig.num_visualizations', index=0, + number=1, type=13, cpp_type=3, label=1, + has_default_value=True, default_value=10, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='num_examples', full_name='object_detection.protos.EvalConfig.num_examples', index=1, + number=2, type=13, cpp_type=3, label=1, + has_default_value=True, default_value=5000, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='eval_interval_secs', full_name='object_detection.protos.EvalConfig.eval_interval_secs', index=2, + number=3, type=13, cpp_type=3, label=1, + has_default_value=True, default_value=300, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_evals', full_name='object_detection.protos.EvalConfig.max_evals', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='save_graph', full_name='object_detection.protos.EvalConfig.save_graph', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='visualization_export_dir', full_name='object_detection.protos.EvalConfig.visualization_export_dir', index=5, + number=6, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='eval_master', full_name='object_detection.protos.EvalConfig.eval_master', index=6, + number=7, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='metrics_set', full_name='object_detection.protos.EvalConfig.metrics_set', index=7, + number=8, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='export_path', full_name='object_detection.protos.EvalConfig.export_path', index=8, + number=9, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='ignore_groundtruth', full_name='object_detection.protos.EvalConfig.ignore_groundtruth', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='use_moving_averages', full_name='object_detection.protos.EvalConfig.use_moving_averages', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='eval_instance_masks', full_name='object_detection.protos.EvalConfig.eval_instance_masks', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='min_score_threshold', full_name='object_detection.protos.EvalConfig.min_score_threshold', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_num_boxes_to_visualize', full_name='object_detection.protos.EvalConfig.max_num_boxes_to_visualize', index=13, + number=14, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=20, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='skip_scores', full_name='object_detection.protos.EvalConfig.skip_scores', index=14, + number=15, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='skip_labels', full_name='object_detection.protos.EvalConfig.skip_labels', index=15, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='visualize_groundtruth_boxes', full_name='object_detection.protos.EvalConfig.visualize_groundtruth_boxes', index=16, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='groundtruth_box_visualization_color', full_name='object_detection.protos.EvalConfig.groundtruth_box_visualization_color', index=17, + number=18, type=9, cpp_type=9, label=1, + has_default_value=True, default_value=_b("black").decode('utf-8'), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='keep_image_id_for_visualization_export', full_name='object_detection.protos.EvalConfig.keep_image_id_for_visualization_export', index=18, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='retain_original_images', full_name='object_detection.protos.EvalConfig.retain_original_images', index=19, + number=23, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='include_metrics_per_category', full_name='object_detection.protos.EvalConfig.include_metrics_per_category', index=20, + number=24, 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=64, + serialized_end=792, +) + +DESCRIPTOR.message_types_by_name['EvalConfig'] = _EVALCONFIG +_sym_db.RegisterFileDescriptor(DESCRIPTOR) + +EvalConfig = _reflection.GeneratedProtocolMessageType('EvalConfig', (_message.Message,), dict( + DESCRIPTOR = _EVALCONFIG, + __module__ = 'object_detection.protos.eval_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.EvalConfig) + )) +_sym_db.RegisterMessage(EvalConfig) + + +# @@protoc_insertion_point(module_scope) diff --git a/research/object_detection/protos/faster_rcnn.proto b/research/object_detection/object_detection/protos/faster_rcnn.proto similarity index 100% rename from research/object_detection/protos/faster_rcnn.proto rename to research/object_detection/object_detection/protos/faster_rcnn.proto diff --git a/research/object_detection/protos/faster_rcnn_box_coder.proto b/research/object_detection/object_detection/protos/faster_rcnn_box_coder.proto similarity index 100% rename from research/object_detection/protos/faster_rcnn_box_coder.proto rename to research/object_detection/object_detection/protos/faster_rcnn_box_coder.proto diff --git a/research/object_detection/object_detection/protos/faster_rcnn_box_coder_pb2.py b/research/object_detection/object_detection/protos/faster_rcnn_box_coder_pb2.py new file mode 100644 index 00000000000..6faee77dc43 --- /dev/null +++ b/research/object_detection/object_detection/protos/faster_rcnn_box_coder_pb2.py @@ -0,0 +1,90 @@ +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: object_detection/protos/faster_rcnn_box_coder.proto + +import sys +_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) +from google.protobuf import descriptor as _descriptor +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +from google.protobuf import descriptor_pb2 +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor.FileDescriptor( + name='object_detection/protos/faster_rcnn_box_coder.proto', + package='object_detection.protos', + syntax='proto2', + serialized_pb=_b('\n3object_detection/protos/faster_rcnn_box_coder.proto\x12\x17object_detection.protos\"o\n\x12\x46\x61sterRcnnBoxCoder\x12\x13\n\x07y_scale\x18\x01 \x01(\x02:\x02\x31\x30\x12\x13\n\x07x_scale\x18\x02 \x01(\x02:\x02\x31\x30\x12\x17\n\x0cheight_scale\x18\x03 \x01(\x02:\x01\x35\x12\x16\n\x0bwidth_scale\x18\x04 \x01(\x02:\x01\x35') +) + + + + +_FASTERRCNNBOXCODER = _descriptor.Descriptor( + name='FasterRcnnBoxCoder', + full_name='object_detection.protos.FasterRcnnBoxCoder', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='y_scale', full_name='object_detection.protos.FasterRcnnBoxCoder.y_scale', index=0, + number=1, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(10), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='x_scale', full_name='object_detection.protos.FasterRcnnBoxCoder.x_scale', index=1, + number=2, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(10), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='height_scale', full_name='object_detection.protos.FasterRcnnBoxCoder.height_scale', index=2, + number=3, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(5), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='width_scale', full_name='object_detection.protos.FasterRcnnBoxCoder.width_scale', index=3, + number=4, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(5), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=80, + serialized_end=191, +) + +DESCRIPTOR.message_types_by_name['FasterRcnnBoxCoder'] = _FASTERRCNNBOXCODER +_sym_db.RegisterFileDescriptor(DESCRIPTOR) + +FasterRcnnBoxCoder = _reflection.GeneratedProtocolMessageType('FasterRcnnBoxCoder', (_message.Message,), dict( + DESCRIPTOR = _FASTERRCNNBOXCODER, + __module__ = 'object_detection.protos.faster_rcnn_box_coder_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.FasterRcnnBoxCoder) + )) +_sym_db.RegisterMessage(FasterRcnnBoxCoder) + + +# @@protoc_insertion_point(module_scope) diff --git a/research/object_detection/object_detection/protos/faster_rcnn_pb2.py b/research/object_detection/object_detection/protos/faster_rcnn_pb2.py new file mode 100644 index 00000000000..31c5f783442 --- /dev/null +++ b/research/object_detection/object_detection/protos/faster_rcnn_pb2.py @@ -0,0 +1,340 @@ +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: object_detection/protos/faster_rcnn.proto + +import sys +_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) +from google.protobuf import descriptor as _descriptor +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +from google.protobuf import descriptor_pb2 +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + +from object_detection.protos import anchor_generator_pb2 as object__detection_dot_protos_dot_anchor__generator__pb2 +from object_detection.protos import box_predictor_pb2 as object__detection_dot_protos_dot_box__predictor__pb2 +from object_detection.protos import hyperparams_pb2 as object__detection_dot_protos_dot_hyperparams__pb2 +from object_detection.protos import image_resizer_pb2 as object__detection_dot_protos_dot_image__resizer__pb2 +from object_detection.protos import losses_pb2 as object__detection_dot_protos_dot_losses__pb2 +from object_detection.protos import post_processing_pb2 as object__detection_dot_protos_dot_post__processing__pb2 + + +DESCRIPTOR = _descriptor.FileDescriptor( + name='object_detection/protos/faster_rcnn.proto', + package='object_detection.protos', + syntax='proto2', + serialized_pb=_b('\n)object_detection/protos/faster_rcnn.proto\x12\x17object_detection.protos\x1a.object_detection/protos/anchor_generator.proto\x1a+object_detection/protos/box_predictor.proto\x1a)object_detection/protos/hyperparams.proto\x1a+object_detection/protos/image_resizer.proto\x1a$object_detection/protos/losses.proto\x1a-object_detection/protos/post_processing.proto\"\x80\x0c\n\nFasterRcnn\x12\x1b\n\x10number_of_stages\x18\x01 \x01(\x05:\x01\x32\x12\x13\n\x0bnum_classes\x18\x03 \x01(\x05\x12<\n\rimage_resizer\x18\x04 \x01(\x0b\x32%.object_detection.protos.ImageResizer\x12N\n\x11\x66\x65\x61ture_extractor\x18\x05 \x01(\x0b\x32\x33.object_detection.protos.FasterRcnnFeatureExtractor\x12N\n\x1c\x66irst_stage_anchor_generator\x18\x06 \x01(\x0b\x32(.object_detection.protos.AnchorGenerator\x12\"\n\x17\x66irst_stage_atrous_rate\x18\x07 \x01(\x05:\x01\x31\x12X\n*first_stage_box_predictor_conv_hyperparams\x18\x08 \x01(\x0b\x32$.object_detection.protos.Hyperparams\x12\x30\n%first_stage_box_predictor_kernel_size\x18\t \x01(\x05:\x01\x33\x12,\n\x1f\x66irst_stage_box_predictor_depth\x18\n \x01(\x05:\x03\x35\x31\x32\x12\'\n\x1a\x66irst_stage_minibatch_size\x18\x0b \x01(\x05:\x03\x32\x35\x36\x12\x32\n%first_stage_positive_balance_fraction\x18\x0c \x01(\x02:\x03\x30.5\x12*\n\x1f\x66irst_stage_nms_score_threshold\x18\r \x01(\x02:\x01\x30\x12*\n\x1d\x66irst_stage_nms_iou_threshold\x18\x0e \x01(\x02:\x03\x30.7\x12&\n\x19\x66irst_stage_max_proposals\x18\x0f \x01(\x05:\x03\x33\x30\x30\x12/\n$first_stage_localization_loss_weight\x18\x10 \x01(\x02:\x01\x31\x12-\n\"first_stage_objectness_loss_weight\x18\x11 \x01(\x02:\x01\x31\x12\x19\n\x11initial_crop_size\x18\x12 \x01(\x05\x12\x1b\n\x13maxpool_kernel_size\x18\x13 \x01(\x05\x12\x16\n\x0emaxpool_stride\x18\x14 \x01(\x05\x12I\n\x1asecond_stage_box_predictor\x18\x15 \x01(\x0b\x32%.object_detection.protos.BoxPredictor\x12#\n\x17second_stage_batch_size\x18\x16 \x01(\x05:\x02\x36\x34\x12+\n\x1dsecond_stage_balance_fraction\x18\x17 \x01(\x02:\x04\x30.25\x12M\n\x1csecond_stage_post_processing\x18\x18 \x01(\x0b\x32\'.object_detection.protos.PostProcessing\x12\x30\n%second_stage_localization_loss_weight\x18\x19 \x01(\x02:\x01\x31\x12\x32\n\'second_stage_classification_loss_weight\x18\x1a \x01(\x02:\x01\x31\x12\x33\n(second_stage_mask_prediction_loss_weight\x18\x1b \x01(\x02:\x01\x31\x12\x45\n\x12hard_example_miner\x18\x1c \x01(\x0b\x32).object_detection.protos.HardExampleMiner\x12U\n second_stage_classification_loss\x18\x1d \x01(\x0b\x32+.object_detection.protos.ClassificationLoss\x12\'\n\x18inplace_batchnorm_update\x18\x1e \x01(\x08:\x05\x66\x61lse\x12)\n\x1ause_matmul_crop_and_resize\x18\x1f \x01(\x08:\x05\x66\x61lse\"x\n\x1a\x46\x61sterRcnnFeatureExtractor\x12\x0c\n\x04type\x18\x01 \x01(\t\x12\'\n\x1b\x66irst_stage_features_stride\x18\x02 \x01(\x05:\x02\x31\x36\x12#\n\x14\x62\x61tch_norm_trainable\x18\x03 \x01(\x08:\x05\x66\x61lse') + , + dependencies=[object__detection_dot_protos_dot_anchor__generator__pb2.DESCRIPTOR,object__detection_dot_protos_dot_box__predictor__pb2.DESCRIPTOR,object__detection_dot_protos_dot_hyperparams__pb2.DESCRIPTOR,object__detection_dot_protos_dot_image__resizer__pb2.DESCRIPTOR,object__detection_dot_protos_dot_losses__pb2.DESCRIPTOR,object__detection_dot_protos_dot_post__processing__pb2.DESCRIPTOR,]) + + + + +_FASTERRCNN = _descriptor.Descriptor( + name='FasterRcnn', + full_name='object_detection.protos.FasterRcnn', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='number_of_stages', full_name='object_detection.protos.FasterRcnn.number_of_stages', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='num_classes', full_name='object_detection.protos.FasterRcnn.num_classes', index=1, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='image_resizer', full_name='object_detection.protos.FasterRcnn.image_resizer', index=2, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='feature_extractor', full_name='object_detection.protos.FasterRcnn.feature_extractor', index=3, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='first_stage_anchor_generator', full_name='object_detection.protos.FasterRcnn.first_stage_anchor_generator', index=4, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='first_stage_atrous_rate', full_name='object_detection.protos.FasterRcnn.first_stage_atrous_rate', index=5, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='first_stage_box_predictor_conv_hyperparams', full_name='object_detection.protos.FasterRcnn.first_stage_box_predictor_conv_hyperparams', index=6, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='first_stage_box_predictor_kernel_size', full_name='object_detection.protos.FasterRcnn.first_stage_box_predictor_kernel_size', index=7, + number=9, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=3, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='first_stage_box_predictor_depth', full_name='object_detection.protos.FasterRcnn.first_stage_box_predictor_depth', index=8, + number=10, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=512, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='first_stage_minibatch_size', full_name='object_detection.protos.FasterRcnn.first_stage_minibatch_size', index=9, + number=11, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=256, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='first_stage_positive_balance_fraction', full_name='object_detection.protos.FasterRcnn.first_stage_positive_balance_fraction', index=10, + number=12, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='first_stage_nms_score_threshold', full_name='object_detection.protos.FasterRcnn.first_stage_nms_score_threshold', index=11, + number=13, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='first_stage_nms_iou_threshold', full_name='object_detection.protos.FasterRcnn.first_stage_nms_iou_threshold', index=12, + number=14, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(0.7), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='first_stage_max_proposals', full_name='object_detection.protos.FasterRcnn.first_stage_max_proposals', index=13, + number=15, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=300, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='first_stage_localization_loss_weight', full_name='object_detection.protos.FasterRcnn.first_stage_localization_loss_weight', index=14, + number=16, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='first_stage_objectness_loss_weight', full_name='object_detection.protos.FasterRcnn.first_stage_objectness_loss_weight', index=15, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='initial_crop_size', full_name='object_detection.protos.FasterRcnn.initial_crop_size', index=16, + number=18, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='maxpool_kernel_size', full_name='object_detection.protos.FasterRcnn.maxpool_kernel_size', index=17, + number=19, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='maxpool_stride', full_name='object_detection.protos.FasterRcnn.maxpool_stride', index=18, + number=20, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='second_stage_box_predictor', full_name='object_detection.protos.FasterRcnn.second_stage_box_predictor', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='second_stage_batch_size', full_name='object_detection.protos.FasterRcnn.second_stage_batch_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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='second_stage_balance_fraction', full_name='object_detection.protos.FasterRcnn.second_stage_balance_fraction', index=21, + number=23, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='second_stage_post_processing', full_name='object_detection.protos.FasterRcnn.second_stage_post_processing', index=22, + number=24, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='second_stage_localization_loss_weight', full_name='object_detection.protos.FasterRcnn.second_stage_localization_loss_weight', index=23, + number=25, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='second_stage_classification_loss_weight', full_name='object_detection.protos.FasterRcnn.second_stage_classification_loss_weight', index=24, + number=26, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='second_stage_mask_prediction_loss_weight', full_name='object_detection.protos.FasterRcnn.second_stage_mask_prediction_loss_weight', index=25, + number=27, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='hard_example_miner', full_name='object_detection.protos.FasterRcnn.hard_example_miner', index=26, + number=28, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='second_stage_classification_loss', full_name='object_detection.protos.FasterRcnn.second_stage_classification_loss', index=27, + number=29, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='inplace_batchnorm_update', full_name='object_detection.protos.FasterRcnn.inplace_batchnorm_update', index=28, + number=30, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='use_matmul_crop_and_resize', full_name='object_detection.protos.FasterRcnn.use_matmul_crop_and_resize', index=29, + number=31, 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=337, + serialized_end=1873, +) + + +_FASTERRCNNFEATUREEXTRACTOR = _descriptor.Descriptor( + name='FasterRcnnFeatureExtractor', + full_name='object_detection.protos.FasterRcnnFeatureExtractor', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='type', full_name='object_detection.protos.FasterRcnnFeatureExtractor.type', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='first_stage_features_stride', full_name='object_detection.protos.FasterRcnnFeatureExtractor.first_stage_features_stride', index=1, + number=2, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=16, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='batch_norm_trainable', full_name='object_detection.protos.FasterRcnnFeatureExtractor.batch_norm_trainable', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=1875, + serialized_end=1995, +) + +_FASTERRCNN.fields_by_name['image_resizer'].message_type = object__detection_dot_protos_dot_image__resizer__pb2._IMAGERESIZER +_FASTERRCNN.fields_by_name['feature_extractor'].message_type = _FASTERRCNNFEATUREEXTRACTOR +_FASTERRCNN.fields_by_name['first_stage_anchor_generator'].message_type = object__detection_dot_protos_dot_anchor__generator__pb2._ANCHORGENERATOR +_FASTERRCNN.fields_by_name['first_stage_box_predictor_conv_hyperparams'].message_type = object__detection_dot_protos_dot_hyperparams__pb2._HYPERPARAMS +_FASTERRCNN.fields_by_name['second_stage_box_predictor'].message_type = object__detection_dot_protos_dot_box__predictor__pb2._BOXPREDICTOR +_FASTERRCNN.fields_by_name['second_stage_post_processing'].message_type = object__detection_dot_protos_dot_post__processing__pb2._POSTPROCESSING +_FASTERRCNN.fields_by_name['hard_example_miner'].message_type = object__detection_dot_protos_dot_losses__pb2._HARDEXAMPLEMINER +_FASTERRCNN.fields_by_name['second_stage_classification_loss'].message_type = object__detection_dot_protos_dot_losses__pb2._CLASSIFICATIONLOSS +DESCRIPTOR.message_types_by_name['FasterRcnn'] = _FASTERRCNN +DESCRIPTOR.message_types_by_name['FasterRcnnFeatureExtractor'] = _FASTERRCNNFEATUREEXTRACTOR +_sym_db.RegisterFileDescriptor(DESCRIPTOR) + +FasterRcnn = _reflection.GeneratedProtocolMessageType('FasterRcnn', (_message.Message,), dict( + DESCRIPTOR = _FASTERRCNN, + __module__ = 'object_detection.protos.faster_rcnn_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.FasterRcnn) + )) +_sym_db.RegisterMessage(FasterRcnn) + +FasterRcnnFeatureExtractor = _reflection.GeneratedProtocolMessageType('FasterRcnnFeatureExtractor', (_message.Message,), dict( + DESCRIPTOR = _FASTERRCNNFEATUREEXTRACTOR, + __module__ = 'object_detection.protos.faster_rcnn_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.FasterRcnnFeatureExtractor) + )) +_sym_db.RegisterMessage(FasterRcnnFeatureExtractor) + + +# @@protoc_insertion_point(module_scope) diff --git a/research/object_detection/protos/graph_rewriter.proto b/research/object_detection/object_detection/protos/graph_rewriter.proto similarity index 100% rename from research/object_detection/protos/graph_rewriter.proto rename to research/object_detection/object_detection/protos/graph_rewriter.proto diff --git a/research/object_detection/object_detection/protos/graph_rewriter_pb2.py b/research/object_detection/object_detection/protos/graph_rewriter_pb2.py new file mode 100644 index 00000000000..aa8e5be3232 --- /dev/null +++ b/research/object_detection/object_detection/protos/graph_rewriter_pb2.py @@ -0,0 +1,123 @@ +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: object_detection/protos/graph_rewriter.proto + +import sys +_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) +from google.protobuf import descriptor as _descriptor +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +from google.protobuf import descriptor_pb2 +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor.FileDescriptor( + name='object_detection/protos/graph_rewriter.proto', + package='object_detection.protos', + syntax='proto2', + serialized_pb=_b('\n,object_detection/protos/graph_rewriter.proto\x12\x17object_detection.protos\"L\n\rGraphRewriter\x12;\n\x0cquantization\x18\x01 \x01(\x0b\x32%.object_detection.protos.Quantization\"Y\n\x0cQuantization\x12\x15\n\x05\x64\x65lay\x18\x01 \x01(\x05:\x06\x35\x30\x30\x30\x30\x30\x12\x16\n\x0bweight_bits\x18\x02 \x01(\x05:\x01\x38\x12\x1a\n\x0f\x61\x63tivation_bits\x18\x03 \x01(\x05:\x01\x38') +) + + + + +_GRAPHREWRITER = _descriptor.Descriptor( + name='GraphRewriter', + full_name='object_detection.protos.GraphRewriter', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='quantization', full_name='object_detection.protos.GraphRewriter.quantization', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=73, + serialized_end=149, +) + + +_QUANTIZATION = _descriptor.Descriptor( + name='Quantization', + full_name='object_detection.protos.Quantization', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='delay', full_name='object_detection.protos.Quantization.delay', index=0, + number=1, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=500000, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='weight_bits', full_name='object_detection.protos.Quantization.weight_bits', index=1, + number=2, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=8, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='activation_bits', full_name='object_detection.protos.Quantization.activation_bits', index=2, + number=3, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=8, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=151, + serialized_end=240, +) + +_GRAPHREWRITER.fields_by_name['quantization'].message_type = _QUANTIZATION +DESCRIPTOR.message_types_by_name['GraphRewriter'] = _GRAPHREWRITER +DESCRIPTOR.message_types_by_name['Quantization'] = _QUANTIZATION +_sym_db.RegisterFileDescriptor(DESCRIPTOR) + +GraphRewriter = _reflection.GeneratedProtocolMessageType('GraphRewriter', (_message.Message,), dict( + DESCRIPTOR = _GRAPHREWRITER, + __module__ = 'object_detection.protos.graph_rewriter_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.GraphRewriter) + )) +_sym_db.RegisterMessage(GraphRewriter) + +Quantization = _reflection.GeneratedProtocolMessageType('Quantization', (_message.Message,), dict( + DESCRIPTOR = _QUANTIZATION, + __module__ = 'object_detection.protos.graph_rewriter_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.Quantization) + )) +_sym_db.RegisterMessage(Quantization) + + +# @@protoc_insertion_point(module_scope) diff --git a/research/object_detection/protos/grid_anchor_generator.proto b/research/object_detection/object_detection/protos/grid_anchor_generator.proto similarity index 100% rename from research/object_detection/protos/grid_anchor_generator.proto rename to research/object_detection/object_detection/protos/grid_anchor_generator.proto diff --git a/research/object_detection/object_detection/protos/grid_anchor_generator_pb2.py b/research/object_detection/object_detection/protos/grid_anchor_generator_pb2.py new file mode 100644 index 00000000000..83b6011cd96 --- /dev/null +++ b/research/object_detection/object_detection/protos/grid_anchor_generator_pb2.py @@ -0,0 +1,118 @@ +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: object_detection/protos/grid_anchor_generator.proto + +import sys +_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) +from google.protobuf import descriptor as _descriptor +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +from google.protobuf import descriptor_pb2 +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor.FileDescriptor( + name='object_detection/protos/grid_anchor_generator.proto', + package='object_detection.protos', + syntax='proto2', + serialized_pb=_b('\n3object_detection/protos/grid_anchor_generator.proto\x12\x17object_detection.protos\"\xcd\x01\n\x13GridAnchorGenerator\x12\x13\n\x06height\x18\x01 \x01(\x05:\x03\x32\x35\x36\x12\x12\n\x05width\x18\x02 \x01(\x05:\x03\x32\x35\x36\x12\x19\n\rheight_stride\x18\x03 \x01(\x05:\x02\x31\x36\x12\x18\n\x0cwidth_stride\x18\x04 \x01(\x05:\x02\x31\x36\x12\x18\n\rheight_offset\x18\x05 \x01(\x05:\x01\x30\x12\x17\n\x0cwidth_offset\x18\x06 \x01(\x05:\x01\x30\x12\x0e\n\x06scales\x18\x07 \x03(\x02\x12\x15\n\raspect_ratios\x18\x08 \x03(\x02') +) + + + + +_GRIDANCHORGENERATOR = _descriptor.Descriptor( + name='GridAnchorGenerator', + full_name='object_detection.protos.GridAnchorGenerator', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='height', full_name='object_detection.protos.GridAnchorGenerator.height', index=0, + number=1, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=256, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='width', full_name='object_detection.protos.GridAnchorGenerator.width', index=1, + number=2, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=256, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='height_stride', full_name='object_detection.protos.GridAnchorGenerator.height_stride', index=2, + number=3, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=16, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='width_stride', full_name='object_detection.protos.GridAnchorGenerator.width_stride', index=3, + number=4, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=16, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='height_offset', full_name='object_detection.protos.GridAnchorGenerator.height_offset', index=4, + number=5, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='width_offset', full_name='object_detection.protos.GridAnchorGenerator.width_offset', index=5, + number=6, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='scales', full_name='object_detection.protos.GridAnchorGenerator.scales', index=6, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='aspect_ratios', full_name='object_detection.protos.GridAnchorGenerator.aspect_ratios', index=7, + 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=81, + serialized_end=286, +) + +DESCRIPTOR.message_types_by_name['GridAnchorGenerator'] = _GRIDANCHORGENERATOR +_sym_db.RegisterFileDescriptor(DESCRIPTOR) + +GridAnchorGenerator = _reflection.GeneratedProtocolMessageType('GridAnchorGenerator', (_message.Message,), dict( + DESCRIPTOR = _GRIDANCHORGENERATOR, + __module__ = 'object_detection.protos.grid_anchor_generator_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.GridAnchorGenerator) + )) +_sym_db.RegisterMessage(GridAnchorGenerator) + + +# @@protoc_insertion_point(module_scope) diff --git a/research/object_detection/protos/hyperparams.proto b/research/object_detection/object_detection/protos/hyperparams.proto similarity index 100% rename from research/object_detection/protos/hyperparams.proto rename to research/object_detection/object_detection/protos/hyperparams.proto diff --git a/research/object_detection/object_detection/protos/hyperparams_pb2.py b/research/object_detection/object_detection/protos/hyperparams_pb2.py new file mode 100644 index 00000000000..40494fd2be0 --- /dev/null +++ b/research/object_detection/object_detection/protos/hyperparams_pb2.py @@ -0,0 +1,605 @@ +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: object_detection/protos/hyperparams.proto + +import sys +_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) +from google.protobuf import descriptor as _descriptor +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +from google.protobuf import descriptor_pb2 +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor.FileDescriptor( + name='object_detection/protos/hyperparams.proto', + package='object_detection.protos', + syntax='proto2', + serialized_pb=_b('\n)object_detection/protos/hyperparams.proto\x12\x17object_detection.protos\"\xac\x03\n\x0bHyperparams\x12\x39\n\x02op\x18\x01 \x01(\x0e\x32\'.object_detection.protos.Hyperparams.Op:\x04\x43ONV\x12\x39\n\x0bregularizer\x18\x02 \x01(\x0b\x32$.object_detection.protos.Regularizer\x12\x39\n\x0binitializer\x18\x03 \x01(\x0b\x32$.object_detection.protos.Initializer\x12I\n\nactivation\x18\x04 \x01(\x0e\x32/.object_detection.protos.Hyperparams.Activation:\x04RELU\x12\x36\n\nbatch_norm\x18\x05 \x01(\x0b\x32\".object_detection.protos.BatchNorm\x12#\n\x14regularize_depthwise\x18\x06 \x01(\x08:\x05\x66\x61lse\"\x16\n\x02Op\x12\x08\n\x04\x43ONV\x10\x01\x12\x06\n\x02\x46\x43\x10\x02\",\n\nActivation\x12\x08\n\x04NONE\x10\x00\x12\x08\n\x04RELU\x10\x01\x12\n\n\x06RELU_6\x10\x02\"\xa6\x01\n\x0bRegularizer\x12@\n\x0el1_regularizer\x18\x01 \x01(\x0b\x32&.object_detection.protos.L1RegularizerH\x00\x12@\n\x0el2_regularizer\x18\x02 \x01(\x0b\x32&.object_detection.protos.L2RegularizerH\x00\x42\x13\n\x11regularizer_oneof\"\"\n\rL1Regularizer\x12\x11\n\x06weight\x18\x01 \x01(\x02:\x01\x31\"\"\n\rL2Regularizer\x12\x11\n\x06weight\x18\x01 \x01(\x02:\x01\x31\"\xb3\x02\n\x0bInitializer\x12[\n\x1ctruncated_normal_initializer\x18\x01 \x01(\x0b\x32\x33.object_detection.protos.TruncatedNormalInitializerH\x00\x12[\n\x1cvariance_scaling_initializer\x18\x02 \x01(\x0b\x32\x33.object_detection.protos.VarianceScalingInitializerH\x00\x12U\n\x19random_normal_initializer\x18\x03 \x01(\x0b\x32\x30.object_detection.protos.RandomNormalInitializerH\x00\x42\x13\n\x11initializer_oneof\"@\n\x1aTruncatedNormalInitializer\x12\x0f\n\x04mean\x18\x01 \x01(\x02:\x01\x30\x12\x11\n\x06stddev\x18\x02 \x01(\x02:\x01\x31\"\xc5\x01\n\x1aVarianceScalingInitializer\x12\x11\n\x06\x66\x61\x63tor\x18\x01 \x01(\x02:\x01\x32\x12\x16\n\x07uniform\x18\x02 \x01(\x08:\x05\x66\x61lse\x12N\n\x04mode\x18\x03 \x01(\x0e\x32\x38.object_detection.protos.VarianceScalingInitializer.Mode:\x06\x46\x41N_IN\",\n\x04Mode\x12\n\n\x06\x46\x41N_IN\x10\x00\x12\x0b\n\x07\x46\x41N_OUT\x10\x01\x12\x0b\n\x07\x46\x41N_AVG\x10\x02\"=\n\x17RandomNormalInitializer\x12\x0f\n\x04mean\x18\x01 \x01(\x02:\x01\x30\x12\x11\n\x06stddev\x18\x02 \x01(\x02:\x01\x31\"z\n\tBatchNorm\x12\x14\n\x05\x64\x65\x63\x61y\x18\x01 \x01(\x02:\x05\x30.999\x12\x14\n\x06\x63\x65nter\x18\x02 \x01(\x08:\x04true\x12\x14\n\x05scale\x18\x03 \x01(\x08:\x05\x66\x61lse\x12\x16\n\x07\x65psilon\x18\x04 \x01(\x02:\x05\x30.001\x12\x13\n\x05train\x18\x05 \x01(\x08:\x04true') +) + + + +_HYPERPARAMS_OP = _descriptor.EnumDescriptor( + name='Op', + full_name='object_detection.protos.Hyperparams.Op', + filename=None, + file=DESCRIPTOR, + values=[ + _descriptor.EnumValueDescriptor( + name='CONV', index=0, number=1, + options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='FC', index=1, number=2, + options=None, + type=None), + ], + containing_type=None, + options=None, + serialized_start=431, + serialized_end=453, +) +_sym_db.RegisterEnumDescriptor(_HYPERPARAMS_OP) + +_HYPERPARAMS_ACTIVATION = _descriptor.EnumDescriptor( + name='Activation', + full_name='object_detection.protos.Hyperparams.Activation', + filename=None, + file=DESCRIPTOR, + values=[ + _descriptor.EnumValueDescriptor( + name='NONE', index=0, number=0, + options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='RELU', index=1, number=1, + options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='RELU_6', index=2, number=2, + options=None, + type=None), + ], + containing_type=None, + options=None, + serialized_start=455, + serialized_end=499, +) +_sym_db.RegisterEnumDescriptor(_HYPERPARAMS_ACTIVATION) + +_VARIANCESCALINGINITIALIZER_MODE = _descriptor.EnumDescriptor( + name='Mode', + full_name='object_detection.protos.VarianceScalingInitializer.Mode', + filename=None, + file=DESCRIPTOR, + values=[ + _descriptor.EnumValueDescriptor( + name='FAN_IN', index=0, number=0, + options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='FAN_OUT', index=1, number=1, + options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='FAN_AVG', index=2, number=2, + options=None, + type=None), + ], + containing_type=None, + options=None, + serialized_start=1272, + serialized_end=1316, +) +_sym_db.RegisterEnumDescriptor(_VARIANCESCALINGINITIALIZER_MODE) + + +_HYPERPARAMS = _descriptor.Descriptor( + name='Hyperparams', + full_name='object_detection.protos.Hyperparams', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='op', full_name='object_detection.protos.Hyperparams.op', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='regularizer', full_name='object_detection.protos.Hyperparams.regularizer', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='initializer', full_name='object_detection.protos.Hyperparams.initializer', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='activation', full_name='object_detection.protos.Hyperparams.activation', index=3, + number=4, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='batch_norm', full_name='object_detection.protos.Hyperparams.batch_norm', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='regularize_depthwise', full_name='object_detection.protos.Hyperparams.regularize_depthwise', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + _HYPERPARAMS_OP, + _HYPERPARAMS_ACTIVATION, + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=71, + serialized_end=499, +) + + +_REGULARIZER = _descriptor.Descriptor( + name='Regularizer', + full_name='object_detection.protos.Regularizer', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='l1_regularizer', full_name='object_detection.protos.Regularizer.l1_regularizer', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='l2_regularizer', full_name='object_detection.protos.Regularizer.l2_regularizer', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + _descriptor.OneofDescriptor( + name='regularizer_oneof', full_name='object_detection.protos.Regularizer.regularizer_oneof', + index=0, containing_type=None, fields=[]), + ], + serialized_start=502, + serialized_end=668, +) + + +_L1REGULARIZER = _descriptor.Descriptor( + name='L1Regularizer', + full_name='object_detection.protos.L1Regularizer', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='weight', full_name='object_detection.protos.L1Regularizer.weight', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=670, + serialized_end=704, +) + + +_L2REGULARIZER = _descriptor.Descriptor( + name='L2Regularizer', + full_name='object_detection.protos.L2Regularizer', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='weight', full_name='object_detection.protos.L2Regularizer.weight', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=706, + serialized_end=740, +) + + +_INITIALIZER = _descriptor.Descriptor( + name='Initializer', + full_name='object_detection.protos.Initializer', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='truncated_normal_initializer', full_name='object_detection.protos.Initializer.truncated_normal_initializer', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='variance_scaling_initializer', full_name='object_detection.protos.Initializer.variance_scaling_initializer', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='random_normal_initializer', full_name='object_detection.protos.Initializer.random_normal_initializer', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + _descriptor.OneofDescriptor( + name='initializer_oneof', full_name='object_detection.protos.Initializer.initializer_oneof', + index=0, containing_type=None, fields=[]), + ], + serialized_start=743, + serialized_end=1050, +) + + +_TRUNCATEDNORMALINITIALIZER = _descriptor.Descriptor( + name='TruncatedNormalInitializer', + full_name='object_detection.protos.TruncatedNormalInitializer', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='mean', full_name='object_detection.protos.TruncatedNormalInitializer.mean', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='stddev', full_name='object_detection.protos.TruncatedNormalInitializer.stddev', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=1052, + serialized_end=1116, +) + + +_VARIANCESCALINGINITIALIZER = _descriptor.Descriptor( + name='VarianceScalingInitializer', + full_name='object_detection.protos.VarianceScalingInitializer', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='factor', full_name='object_detection.protos.VarianceScalingInitializer.factor', index=0, + number=1, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(2), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='uniform', full_name='object_detection.protos.VarianceScalingInitializer.uniform', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='mode', full_name='object_detection.protos.VarianceScalingInitializer.mode', index=2, + number=3, 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + _VARIANCESCALINGINITIALIZER_MODE, + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=1119, + serialized_end=1316, +) + + +_RANDOMNORMALINITIALIZER = _descriptor.Descriptor( + name='RandomNormalInitializer', + full_name='object_detection.protos.RandomNormalInitializer', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='mean', full_name='object_detection.protos.RandomNormalInitializer.mean', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='stddev', full_name='object_detection.protos.RandomNormalInitializer.stddev', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=1318, + serialized_end=1379, +) + + +_BATCHNORM = _descriptor.Descriptor( + name='BatchNorm', + full_name='object_detection.protos.BatchNorm', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='decay', full_name='object_detection.protos.BatchNorm.decay', index=0, + number=1, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='center', full_name='object_detection.protos.BatchNorm.center', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='scale', full_name='object_detection.protos.BatchNorm.scale', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='epsilon', full_name='object_detection.protos.BatchNorm.epsilon', index=3, + number=4, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(0.001), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='train', full_name='object_detection.protos.BatchNorm.train', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=1381, + serialized_end=1503, +) + +_HYPERPARAMS.fields_by_name['op'].enum_type = _HYPERPARAMS_OP +_HYPERPARAMS.fields_by_name['regularizer'].message_type = _REGULARIZER +_HYPERPARAMS.fields_by_name['initializer'].message_type = _INITIALIZER +_HYPERPARAMS.fields_by_name['activation'].enum_type = _HYPERPARAMS_ACTIVATION +_HYPERPARAMS.fields_by_name['batch_norm'].message_type = _BATCHNORM +_HYPERPARAMS_OP.containing_type = _HYPERPARAMS +_HYPERPARAMS_ACTIVATION.containing_type = _HYPERPARAMS +_REGULARIZER.fields_by_name['l1_regularizer'].message_type = _L1REGULARIZER +_REGULARIZER.fields_by_name['l2_regularizer'].message_type = _L2REGULARIZER +_REGULARIZER.oneofs_by_name['regularizer_oneof'].fields.append( + _REGULARIZER.fields_by_name['l1_regularizer']) +_REGULARIZER.fields_by_name['l1_regularizer'].containing_oneof = _REGULARIZER.oneofs_by_name['regularizer_oneof'] +_REGULARIZER.oneofs_by_name['regularizer_oneof'].fields.append( + _REGULARIZER.fields_by_name['l2_regularizer']) +_REGULARIZER.fields_by_name['l2_regularizer'].containing_oneof = _REGULARIZER.oneofs_by_name['regularizer_oneof'] +_INITIALIZER.fields_by_name['truncated_normal_initializer'].message_type = _TRUNCATEDNORMALINITIALIZER +_INITIALIZER.fields_by_name['variance_scaling_initializer'].message_type = _VARIANCESCALINGINITIALIZER +_INITIALIZER.fields_by_name['random_normal_initializer'].message_type = _RANDOMNORMALINITIALIZER +_INITIALIZER.oneofs_by_name['initializer_oneof'].fields.append( + _INITIALIZER.fields_by_name['truncated_normal_initializer']) +_INITIALIZER.fields_by_name['truncated_normal_initializer'].containing_oneof = _INITIALIZER.oneofs_by_name['initializer_oneof'] +_INITIALIZER.oneofs_by_name['initializer_oneof'].fields.append( + _INITIALIZER.fields_by_name['variance_scaling_initializer']) +_INITIALIZER.fields_by_name['variance_scaling_initializer'].containing_oneof = _INITIALIZER.oneofs_by_name['initializer_oneof'] +_INITIALIZER.oneofs_by_name['initializer_oneof'].fields.append( + _INITIALIZER.fields_by_name['random_normal_initializer']) +_INITIALIZER.fields_by_name['random_normal_initializer'].containing_oneof = _INITIALIZER.oneofs_by_name['initializer_oneof'] +_VARIANCESCALINGINITIALIZER.fields_by_name['mode'].enum_type = _VARIANCESCALINGINITIALIZER_MODE +_VARIANCESCALINGINITIALIZER_MODE.containing_type = _VARIANCESCALINGINITIALIZER +DESCRIPTOR.message_types_by_name['Hyperparams'] = _HYPERPARAMS +DESCRIPTOR.message_types_by_name['Regularizer'] = _REGULARIZER +DESCRIPTOR.message_types_by_name['L1Regularizer'] = _L1REGULARIZER +DESCRIPTOR.message_types_by_name['L2Regularizer'] = _L2REGULARIZER +DESCRIPTOR.message_types_by_name['Initializer'] = _INITIALIZER +DESCRIPTOR.message_types_by_name['TruncatedNormalInitializer'] = _TRUNCATEDNORMALINITIALIZER +DESCRIPTOR.message_types_by_name['VarianceScalingInitializer'] = _VARIANCESCALINGINITIALIZER +DESCRIPTOR.message_types_by_name['RandomNormalInitializer'] = _RANDOMNORMALINITIALIZER +DESCRIPTOR.message_types_by_name['BatchNorm'] = _BATCHNORM +_sym_db.RegisterFileDescriptor(DESCRIPTOR) + +Hyperparams = _reflection.GeneratedProtocolMessageType('Hyperparams', (_message.Message,), dict( + DESCRIPTOR = _HYPERPARAMS, + __module__ = 'object_detection.protos.hyperparams_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.Hyperparams) + )) +_sym_db.RegisterMessage(Hyperparams) + +Regularizer = _reflection.GeneratedProtocolMessageType('Regularizer', (_message.Message,), dict( + DESCRIPTOR = _REGULARIZER, + __module__ = 'object_detection.protos.hyperparams_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.Regularizer) + )) +_sym_db.RegisterMessage(Regularizer) + +L1Regularizer = _reflection.GeneratedProtocolMessageType('L1Regularizer', (_message.Message,), dict( + DESCRIPTOR = _L1REGULARIZER, + __module__ = 'object_detection.protos.hyperparams_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.L1Regularizer) + )) +_sym_db.RegisterMessage(L1Regularizer) + +L2Regularizer = _reflection.GeneratedProtocolMessageType('L2Regularizer', (_message.Message,), dict( + DESCRIPTOR = _L2REGULARIZER, + __module__ = 'object_detection.protos.hyperparams_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.L2Regularizer) + )) +_sym_db.RegisterMessage(L2Regularizer) + +Initializer = _reflection.GeneratedProtocolMessageType('Initializer', (_message.Message,), dict( + DESCRIPTOR = _INITIALIZER, + __module__ = 'object_detection.protos.hyperparams_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.Initializer) + )) +_sym_db.RegisterMessage(Initializer) + +TruncatedNormalInitializer = _reflection.GeneratedProtocolMessageType('TruncatedNormalInitializer', (_message.Message,), dict( + DESCRIPTOR = _TRUNCATEDNORMALINITIALIZER, + __module__ = 'object_detection.protos.hyperparams_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.TruncatedNormalInitializer) + )) +_sym_db.RegisterMessage(TruncatedNormalInitializer) + +VarianceScalingInitializer = _reflection.GeneratedProtocolMessageType('VarianceScalingInitializer', (_message.Message,), dict( + DESCRIPTOR = _VARIANCESCALINGINITIALIZER, + __module__ = 'object_detection.protos.hyperparams_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.VarianceScalingInitializer) + )) +_sym_db.RegisterMessage(VarianceScalingInitializer) + +RandomNormalInitializer = _reflection.GeneratedProtocolMessageType('RandomNormalInitializer', (_message.Message,), dict( + DESCRIPTOR = _RANDOMNORMALINITIALIZER, + __module__ = 'object_detection.protos.hyperparams_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.RandomNormalInitializer) + )) +_sym_db.RegisterMessage(RandomNormalInitializer) + +BatchNorm = _reflection.GeneratedProtocolMessageType('BatchNorm', (_message.Message,), dict( + DESCRIPTOR = _BATCHNORM, + __module__ = 'object_detection.protos.hyperparams_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.BatchNorm) + )) +_sym_db.RegisterMessage(BatchNorm) + + +# @@protoc_insertion_point(module_scope) diff --git a/research/object_detection/protos/image_resizer.proto b/research/object_detection/object_detection/protos/image_resizer.proto similarity index 100% rename from research/object_detection/protos/image_resizer.proto rename to research/object_detection/object_detection/protos/image_resizer.proto diff --git a/research/object_detection/object_detection/protos/image_resizer_pb2.py b/research/object_detection/object_detection/protos/image_resizer_pb2.py new file mode 100644 index 00000000000..80c169ff8c4 --- /dev/null +++ b/research/object_detection/object_detection/protos/image_resizer_pb2.py @@ -0,0 +1,260 @@ +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: object_detection/protos/image_resizer.proto + +import sys +_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) +from google.protobuf.internal import enum_type_wrapper +from google.protobuf import descriptor as _descriptor +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +from google.protobuf import descriptor_pb2 +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor.FileDescriptor( + name='object_detection/protos/image_resizer.proto', + package='object_detection.protos', + syntax='proto2', + serialized_pb=_b('\n+object_detection/protos/image_resizer.proto\x12\x17object_detection.protos\"\xc6\x01\n\x0cImageResizer\x12T\n\x19keep_aspect_ratio_resizer\x18\x01 \x01(\x0b\x32/.object_detection.protos.KeepAspectRatioResizerH\x00\x12I\n\x13\x66ixed_shape_resizer\x18\x02 \x01(\x0b\x32*.object_detection.protos.FixedShapeResizerH\x00\x42\x15\n\x13image_resizer_oneof\"\x80\x02\n\x16KeepAspectRatioResizer\x12\x1a\n\rmin_dimension\x18\x01 \x01(\x05:\x03\x36\x30\x30\x12\x1b\n\rmax_dimension\x18\x02 \x01(\x05:\x04\x31\x30\x32\x34\x12\x44\n\rresize_method\x18\x03 \x01(\x0e\x32#.object_detection.protos.ResizeType:\x08\x42ILINEAR\x12#\n\x14pad_to_max_dimension\x18\x04 \x01(\x08:\x05\x66\x61lse\x12#\n\x14\x63onvert_to_grayscale\x18\x05 \x01(\x08:\x05\x66\x61lse\x12\x1d\n\x15per_channel_pad_value\x18\x06 \x03(\x02\"\xa7\x01\n\x11\x46ixedShapeResizer\x12\x13\n\x06height\x18\x01 \x01(\x05:\x03\x33\x30\x30\x12\x12\n\x05width\x18\x02 \x01(\x05:\x03\x33\x30\x30\x12\x44\n\rresize_method\x18\x03 \x01(\x0e\x32#.object_detection.protos.ResizeType:\x08\x42ILINEAR\x12#\n\x14\x63onvert_to_grayscale\x18\x04 \x01(\x08:\x05\x66\x61lse*G\n\nResizeType\x12\x0c\n\x08\x42ILINEAR\x10\x00\x12\x14\n\x10NEAREST_NEIGHBOR\x10\x01\x12\x0b\n\x07\x42ICUBIC\x10\x02\x12\x08\n\x04\x41REA\x10\x03') +) + +_RESIZETYPE = _descriptor.EnumDescriptor( + name='ResizeType', + full_name='object_detection.protos.ResizeType', + filename=None, + file=DESCRIPTOR, + values=[ + _descriptor.EnumValueDescriptor( + name='BILINEAR', index=0, number=0, + options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='NEAREST_NEIGHBOR', index=1, number=1, + options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='BICUBIC', index=2, number=2, + options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='AREA', index=3, number=3, + options=None, + type=None), + ], + containing_type=None, + options=None, + serialized_start=702, + serialized_end=773, +) +_sym_db.RegisterEnumDescriptor(_RESIZETYPE) + +ResizeType = enum_type_wrapper.EnumTypeWrapper(_RESIZETYPE) +BILINEAR = 0 +NEAREST_NEIGHBOR = 1 +BICUBIC = 2 +AREA = 3 + + + +_IMAGERESIZER = _descriptor.Descriptor( + name='ImageResizer', + full_name='object_detection.protos.ImageResizer', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='keep_aspect_ratio_resizer', full_name='object_detection.protos.ImageResizer.keep_aspect_ratio_resizer', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='fixed_shape_resizer', full_name='object_detection.protos.ImageResizer.fixed_shape_resizer', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + _descriptor.OneofDescriptor( + name='image_resizer_oneof', full_name='object_detection.protos.ImageResizer.image_resizer_oneof', + index=0, containing_type=None, fields=[]), + ], + serialized_start=73, + serialized_end=271, +) + + +_KEEPASPECTRATIORESIZER = _descriptor.Descriptor( + name='KeepAspectRatioResizer', + full_name='object_detection.protos.KeepAspectRatioResizer', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='min_dimension', full_name='object_detection.protos.KeepAspectRatioResizer.min_dimension', index=0, + number=1, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=600, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_dimension', full_name='object_detection.protos.KeepAspectRatioResizer.max_dimension', index=1, + number=2, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=1024, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='resize_method', full_name='object_detection.protos.KeepAspectRatioResizer.resize_method', index=2, + number=3, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='pad_to_max_dimension', full_name='object_detection.protos.KeepAspectRatioResizer.pad_to_max_dimension', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='convert_to_grayscale', full_name='object_detection.protos.KeepAspectRatioResizer.convert_to_grayscale', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='per_channel_pad_value', full_name='object_detection.protos.KeepAspectRatioResizer.per_channel_pad_value', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=274, + serialized_end=530, +) + + +_FIXEDSHAPERESIZER = _descriptor.Descriptor( + name='FixedShapeResizer', + full_name='object_detection.protos.FixedShapeResizer', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='height', full_name='object_detection.protos.FixedShapeResizer.height', index=0, + number=1, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=300, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='width', full_name='object_detection.protos.FixedShapeResizer.width', index=1, + number=2, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=300, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='resize_method', full_name='object_detection.protos.FixedShapeResizer.resize_method', index=2, + number=3, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='convert_to_grayscale', full_name='object_detection.protos.FixedShapeResizer.convert_to_grayscale', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=533, + serialized_end=700, +) + +_IMAGERESIZER.fields_by_name['keep_aspect_ratio_resizer'].message_type = _KEEPASPECTRATIORESIZER +_IMAGERESIZER.fields_by_name['fixed_shape_resizer'].message_type = _FIXEDSHAPERESIZER +_IMAGERESIZER.oneofs_by_name['image_resizer_oneof'].fields.append( + _IMAGERESIZER.fields_by_name['keep_aspect_ratio_resizer']) +_IMAGERESIZER.fields_by_name['keep_aspect_ratio_resizer'].containing_oneof = _IMAGERESIZER.oneofs_by_name['image_resizer_oneof'] +_IMAGERESIZER.oneofs_by_name['image_resizer_oneof'].fields.append( + _IMAGERESIZER.fields_by_name['fixed_shape_resizer']) +_IMAGERESIZER.fields_by_name['fixed_shape_resizer'].containing_oneof = _IMAGERESIZER.oneofs_by_name['image_resizer_oneof'] +_KEEPASPECTRATIORESIZER.fields_by_name['resize_method'].enum_type = _RESIZETYPE +_FIXEDSHAPERESIZER.fields_by_name['resize_method'].enum_type = _RESIZETYPE +DESCRIPTOR.message_types_by_name['ImageResizer'] = _IMAGERESIZER +DESCRIPTOR.message_types_by_name['KeepAspectRatioResizer'] = _KEEPASPECTRATIORESIZER +DESCRIPTOR.message_types_by_name['FixedShapeResizer'] = _FIXEDSHAPERESIZER +DESCRIPTOR.enum_types_by_name['ResizeType'] = _RESIZETYPE +_sym_db.RegisterFileDescriptor(DESCRIPTOR) + +ImageResizer = _reflection.GeneratedProtocolMessageType('ImageResizer', (_message.Message,), dict( + DESCRIPTOR = _IMAGERESIZER, + __module__ = 'object_detection.protos.image_resizer_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.ImageResizer) + )) +_sym_db.RegisterMessage(ImageResizer) + +KeepAspectRatioResizer = _reflection.GeneratedProtocolMessageType('KeepAspectRatioResizer', (_message.Message,), dict( + DESCRIPTOR = _KEEPASPECTRATIORESIZER, + __module__ = 'object_detection.protos.image_resizer_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.KeepAspectRatioResizer) + )) +_sym_db.RegisterMessage(KeepAspectRatioResizer) + +FixedShapeResizer = _reflection.GeneratedProtocolMessageType('FixedShapeResizer', (_message.Message,), dict( + DESCRIPTOR = _FIXEDSHAPERESIZER, + __module__ = 'object_detection.protos.image_resizer_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.FixedShapeResizer) + )) +_sym_db.RegisterMessage(FixedShapeResizer) + + +# @@protoc_insertion_point(module_scope) diff --git a/research/object_detection/protos/input_reader.proto b/research/object_detection/object_detection/protos/input_reader.proto similarity index 100% rename from research/object_detection/protos/input_reader.proto rename to research/object_detection/object_detection/protos/input_reader.proto diff --git a/research/object_detection/object_detection/protos/input_reader_pb2.py b/research/object_detection/object_detection/protos/input_reader_pb2.py new file mode 100644 index 00000000000..d69bc6d2022 --- /dev/null +++ b/research/object_detection/object_detection/protos/input_reader_pb2.py @@ -0,0 +1,332 @@ +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: object_detection/protos/input_reader.proto + +import sys +_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) +from google.protobuf.internal import enum_type_wrapper +from google.protobuf import descriptor as _descriptor +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +from google.protobuf import descriptor_pb2 +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor.FileDescriptor( + name='object_detection/protos/input_reader.proto', + package='object_detection.protos', + syntax='proto2', + serialized_pb=_b('\n*object_detection/protos/input_reader.proto\x12\x17object_detection.protos\"\xd0\x06\n\x0bInputReader\x12\x18\n\x0elabel_map_path\x18\x01 \x01(\t:\x00\x12\x15\n\x07shuffle\x18\x02 \x01(\x08:\x04true\x12!\n\x13shuffle_buffer_size\x18\x0b \x01(\r:\x04\x32\x30\x34\x38\x12*\n\x1d\x66ilenames_shuffle_buffer_size\x18\x0c \x01(\r:\x03\x31\x30\x30\x12\x15\n\nnum_epochs\x18\x05 \x01(\r:\x01\x30\x12\x17\n\x0bnum_readers\x18\x06 \x01(\r:\x02\x36\x34\x12\x1f\n\x14num_parallel_batches\x18\x13 \x01(\r:\x01\x38\x12\x1f\n\x14num_prefetch_batches\x18\x14 \x01(\x05:\x01\x32\x12 \n\x0equeue_capacity\x18\x03 \x01(\r:\x04\x32\x30\x30\x30\x42\x02\x18\x01\x12#\n\x11min_after_dequeue\x18\x04 \x01(\r:\x04\x31\x30\x30\x30\x42\x02\x18\x01\x12\x1d\n\x11read_block_length\x18\x0f \x01(\r:\x02\x33\x32\x12\x1e\n\rprefetch_size\x18\r \x01(\r:\x03\x35\x31\x32\x42\x02\x18\x01\x12&\n\x16num_parallel_map_calls\x18\x0e \x01(\r:\x02\x36\x34\x42\x02\x18\x01\x12\"\n\x17num_additional_channels\x18\x12 \x01(\x05:\x01\x30\x12\x18\n\rnum_keypoints\x18\x10 \x01(\r:\x01\x30\x12 \n\x13max_number_of_boxes\x18\x15 \x01(\x05:\x03\x31\x30\x30\x12\"\n\x13load_instance_masks\x18\x07 \x01(\x08:\x05\x66\x61lse\x12M\n\tmask_type\x18\n \x01(\x0e\x32).object_detection.protos.InstanceMaskType:\x0fNUMERICAL_MASKS\x12\x1f\n\x10use_display_name\x18\x11 \x01(\x08:\x05\x66\x61lse\x12N\n\x16tf_record_input_reader\x18\x08 \x01(\x0b\x32,.object_detection.protos.TFRecordInputReaderH\x00\x12M\n\x15\x65xternal_input_reader\x18\t \x01(\x0b\x32,.object_detection.protos.ExternalInputReaderH\x00\x42\x0e\n\x0cinput_reader\")\n\x13TFRecordInputReader\x12\x12\n\ninput_path\x18\x01 \x03(\t\"\x1c\n\x13\x45xternalInputReader*\x05\x08\x01\x10\xe8\x07*C\n\x10InstanceMaskType\x12\x0b\n\x07\x44\x45\x46\x41ULT\x10\x00\x12\x13\n\x0fNUMERICAL_MASKS\x10\x01\x12\r\n\tPNG_MASKS\x10\x02') +) + +_INSTANCEMASKTYPE = _descriptor.EnumDescriptor( + name='InstanceMaskType', + full_name='object_detection.protos.InstanceMaskType', + filename=None, + file=DESCRIPTOR, + values=[ + _descriptor.EnumValueDescriptor( + name='DEFAULT', index=0, number=0, + options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='NUMERICAL_MASKS', index=1, number=1, + options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='PNG_MASKS', index=2, number=2, + options=None, + type=None), + ], + containing_type=None, + options=None, + serialized_start=995, + serialized_end=1062, +) +_sym_db.RegisterEnumDescriptor(_INSTANCEMASKTYPE) + +InstanceMaskType = enum_type_wrapper.EnumTypeWrapper(_INSTANCEMASKTYPE) +DEFAULT = 0 +NUMERICAL_MASKS = 1 +PNG_MASKS = 2 + + + +_INPUTREADER = _descriptor.Descriptor( + name='InputReader', + full_name='object_detection.protos.InputReader', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='label_map_path', full_name='object_detection.protos.InputReader.label_map_path', index=0, + number=1, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='shuffle', full_name='object_detection.protos.InputReader.shuffle', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='shuffle_buffer_size', full_name='object_detection.protos.InputReader.shuffle_buffer_size', index=2, + number=11, type=13, cpp_type=3, label=1, + has_default_value=True, default_value=2048, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='filenames_shuffle_buffer_size', full_name='object_detection.protos.InputReader.filenames_shuffle_buffer_size', index=3, + number=12, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='num_epochs', full_name='object_detection.protos.InputReader.num_epochs', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='num_readers', full_name='object_detection.protos.InputReader.num_readers', index=5, + number=6, type=13, cpp_type=3, label=1, + has_default_value=True, default_value=64, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='num_parallel_batches', full_name='object_detection.protos.InputReader.num_parallel_batches', index=6, + number=19, type=13, cpp_type=3, label=1, + has_default_value=True, default_value=8, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='num_prefetch_batches', full_name='object_detection.protos.InputReader.num_prefetch_batches', index=7, + number=20, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='queue_capacity', full_name='object_detection.protos.InputReader.queue_capacity', index=8, + number=3, type=13, cpp_type=3, label=1, + has_default_value=True, default_value=2000, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=_descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\030\001')), file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='min_after_dequeue', full_name='object_detection.protos.InputReader.min_after_dequeue', index=9, + number=4, type=13, cpp_type=3, label=1, + has_default_value=True, default_value=1000, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=_descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\030\001')), file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='read_block_length', full_name='object_detection.protos.InputReader.read_block_length', index=10, + number=15, type=13, cpp_type=3, label=1, + has_default_value=True, default_value=32, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='prefetch_size', full_name='object_detection.protos.InputReader.prefetch_size', index=11, + number=13, type=13, cpp_type=3, label=1, + has_default_value=True, default_value=512, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=_descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\030\001')), file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='num_parallel_map_calls', full_name='object_detection.protos.InputReader.num_parallel_map_calls', index=12, + number=14, type=13, cpp_type=3, label=1, + has_default_value=True, default_value=64, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=_descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\030\001')), file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='num_additional_channels', full_name='object_detection.protos.InputReader.num_additional_channels', index=13, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='num_keypoints', full_name='object_detection.protos.InputReader.num_keypoints', index=14, + number=16, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_number_of_boxes', full_name='object_detection.protos.InputReader.max_number_of_boxes', index=15, + number=21, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=100, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='load_instance_masks', full_name='object_detection.protos.InputReader.load_instance_masks', index=16, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='mask_type', full_name='object_detection.protos.InputReader.mask_type', index=17, + number=10, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='use_display_name', full_name='object_detection.protos.InputReader.use_display_name', index=18, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='tf_record_input_reader', full_name='object_detection.protos.InputReader.tf_record_input_reader', index=19, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='external_input_reader', full_name='object_detection.protos.InputReader.external_input_reader', index=20, + number=9, 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + _descriptor.OneofDescriptor( + name='input_reader', full_name='object_detection.protos.InputReader.input_reader', + index=0, containing_type=None, fields=[]), + ], + serialized_start=72, + serialized_end=920, +) + + +_TFRECORDINPUTREADER = _descriptor.Descriptor( + name='TFRecordInputReader', + full_name='object_detection.protos.TFRecordInputReader', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='input_path', full_name='object_detection.protos.TFRecordInputReader.input_path', index=0, + number=1, 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=922, + serialized_end=963, +) + + +_EXTERNALINPUTREADER = _descriptor.Descriptor( + name='ExternalInputReader', + full_name='object_detection.protos.ExternalInputReader', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=True, + syntax='proto2', + extension_ranges=[(1, 1000), ], + oneofs=[ + ], + serialized_start=965, + serialized_end=993, +) + +_INPUTREADER.fields_by_name['mask_type'].enum_type = _INSTANCEMASKTYPE +_INPUTREADER.fields_by_name['tf_record_input_reader'].message_type = _TFRECORDINPUTREADER +_INPUTREADER.fields_by_name['external_input_reader'].message_type = _EXTERNALINPUTREADER +_INPUTREADER.oneofs_by_name['input_reader'].fields.append( + _INPUTREADER.fields_by_name['tf_record_input_reader']) +_INPUTREADER.fields_by_name['tf_record_input_reader'].containing_oneof = _INPUTREADER.oneofs_by_name['input_reader'] +_INPUTREADER.oneofs_by_name['input_reader'].fields.append( + _INPUTREADER.fields_by_name['external_input_reader']) +_INPUTREADER.fields_by_name['external_input_reader'].containing_oneof = _INPUTREADER.oneofs_by_name['input_reader'] +DESCRIPTOR.message_types_by_name['InputReader'] = _INPUTREADER +DESCRIPTOR.message_types_by_name['TFRecordInputReader'] = _TFRECORDINPUTREADER +DESCRIPTOR.message_types_by_name['ExternalInputReader'] = _EXTERNALINPUTREADER +DESCRIPTOR.enum_types_by_name['InstanceMaskType'] = _INSTANCEMASKTYPE +_sym_db.RegisterFileDescriptor(DESCRIPTOR) + +InputReader = _reflection.GeneratedProtocolMessageType('InputReader', (_message.Message,), dict( + DESCRIPTOR = _INPUTREADER, + __module__ = 'object_detection.protos.input_reader_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.InputReader) + )) +_sym_db.RegisterMessage(InputReader) + +TFRecordInputReader = _reflection.GeneratedProtocolMessageType('TFRecordInputReader', (_message.Message,), dict( + DESCRIPTOR = _TFRECORDINPUTREADER, + __module__ = 'object_detection.protos.input_reader_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.TFRecordInputReader) + )) +_sym_db.RegisterMessage(TFRecordInputReader) + +ExternalInputReader = _reflection.GeneratedProtocolMessageType('ExternalInputReader', (_message.Message,), dict( + DESCRIPTOR = _EXTERNALINPUTREADER, + __module__ = 'object_detection.protos.input_reader_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.ExternalInputReader) + )) +_sym_db.RegisterMessage(ExternalInputReader) + + +_INPUTREADER.fields_by_name['queue_capacity'].has_options = True +_INPUTREADER.fields_by_name['queue_capacity']._options = _descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\030\001')) +_INPUTREADER.fields_by_name['min_after_dequeue'].has_options = True +_INPUTREADER.fields_by_name['min_after_dequeue']._options = _descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\030\001')) +_INPUTREADER.fields_by_name['prefetch_size'].has_options = True +_INPUTREADER.fields_by_name['prefetch_size']._options = _descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\030\001')) +_INPUTREADER.fields_by_name['num_parallel_map_calls'].has_options = True +_INPUTREADER.fields_by_name['num_parallel_map_calls']._options = _descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\030\001')) +# @@protoc_insertion_point(module_scope) diff --git a/research/object_detection/protos/keypoint_box_coder.proto b/research/object_detection/object_detection/protos/keypoint_box_coder.proto similarity index 100% rename from research/object_detection/protos/keypoint_box_coder.proto rename to research/object_detection/object_detection/protos/keypoint_box_coder.proto diff --git a/research/object_detection/object_detection/protos/keypoint_box_coder_pb2.py b/research/object_detection/object_detection/protos/keypoint_box_coder_pb2.py new file mode 100644 index 00000000000..d473bc9d060 --- /dev/null +++ b/research/object_detection/object_detection/protos/keypoint_box_coder_pb2.py @@ -0,0 +1,97 @@ +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: object_detection/protos/keypoint_box_coder.proto + +import sys +_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) +from google.protobuf import descriptor as _descriptor +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +from google.protobuf import descriptor_pb2 +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor.FileDescriptor( + name='object_detection/protos/keypoint_box_coder.proto', + package='object_detection.protos', + syntax='proto2', + serialized_pb=_b('\n0object_detection/protos/keypoint_box_coder.proto\x12\x17object_detection.protos\"\x84\x01\n\x10KeypointBoxCoder\x12\x15\n\rnum_keypoints\x18\x01 \x01(\x05\x12\x13\n\x07y_scale\x18\x02 \x01(\x02:\x02\x31\x30\x12\x13\n\x07x_scale\x18\x03 \x01(\x02:\x02\x31\x30\x12\x17\n\x0cheight_scale\x18\x04 \x01(\x02:\x01\x35\x12\x16\n\x0bwidth_scale\x18\x05 \x01(\x02:\x01\x35') +) + + + + +_KEYPOINTBOXCODER = _descriptor.Descriptor( + name='KeypointBoxCoder', + full_name='object_detection.protos.KeypointBoxCoder', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='num_keypoints', full_name='object_detection.protos.KeypointBoxCoder.num_keypoints', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='y_scale', full_name='object_detection.protos.KeypointBoxCoder.y_scale', index=1, + number=2, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(10), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='x_scale', full_name='object_detection.protos.KeypointBoxCoder.x_scale', index=2, + number=3, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(10), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='height_scale', full_name='object_detection.protos.KeypointBoxCoder.height_scale', index=3, + number=4, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(5), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='width_scale', full_name='object_detection.protos.KeypointBoxCoder.width_scale', index=4, + number=5, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(5), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=78, + serialized_end=210, +) + +DESCRIPTOR.message_types_by_name['KeypointBoxCoder'] = _KEYPOINTBOXCODER +_sym_db.RegisterFileDescriptor(DESCRIPTOR) + +KeypointBoxCoder = _reflection.GeneratedProtocolMessageType('KeypointBoxCoder', (_message.Message,), dict( + DESCRIPTOR = _KEYPOINTBOXCODER, + __module__ = 'object_detection.protos.keypoint_box_coder_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.KeypointBoxCoder) + )) +_sym_db.RegisterMessage(KeypointBoxCoder) + + +# @@protoc_insertion_point(module_scope) diff --git a/research/object_detection/protos/losses.proto b/research/object_detection/object_detection/protos/losses.proto similarity index 100% rename from research/object_detection/protos/losses.proto rename to research/object_detection/object_detection/protos/losses.proto diff --git a/research/object_detection/object_detection/protos/losses_pb2.py b/research/object_detection/object_detection/protos/losses_pb2.py new file mode 100644 index 00000000000..219c2652b7e --- /dev/null +++ b/research/object_detection/object_detection/protos/losses_pb2.py @@ -0,0 +1,755 @@ +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: object_detection/protos/losses.proto + +import sys +_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) +from google.protobuf import descriptor as _descriptor +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +from google.protobuf import descriptor_pb2 +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor.FileDescriptor( + name='object_detection/protos/losses.proto', + package='object_detection.protos', + syntax='proto2', + serialized_pb=_b('\n$object_detection/protos/losses.proto\x12\x17object_detection.protos\"\xee\x02\n\x04Loss\x12\x44\n\x11localization_loss\x18\x01 \x01(\x0b\x32).object_detection.protos.LocalizationLoss\x12H\n\x13\x63lassification_loss\x18\x02 \x01(\x0b\x32+.object_detection.protos.ClassificationLoss\x12\x45\n\x12hard_example_miner\x18\x03 \x01(\x0b\x32).object_detection.protos.HardExampleMiner\x12 \n\x15\x63lassification_weight\x18\x04 \x01(\x02:\x01\x31\x12\x1e\n\x13localization_weight\x18\x05 \x01(\x02:\x01\x31\x12M\n\x16random_example_sampler\x18\x06 \x01(\x0b\x32-.object_detection.protos.RandomExampleSampler\"\x9a\x02\n\x10LocalizationLoss\x12J\n\x0bweighted_l2\x18\x01 \x01(\x0b\x32\x33.object_detection.protos.WeightedL2LocalizationLossH\x00\x12W\n\x12weighted_smooth_l1\x18\x02 \x01(\x0b\x32\x39.object_detection.protos.WeightedSmoothL1LocalizationLossH\x00\x12L\n\x0cweighted_iou\x18\x03 \x01(\x0b\x32\x34.object_detection.protos.WeightedIOULocalizationLossH\x00\x42\x13\n\x11localization_loss\">\n\x1aWeightedL2LocalizationLoss\x12 \n\x11\x61nchorwise_output\x18\x01 \x01(\x08:\x05\x66\x61lse\"V\n WeightedSmoothL1LocalizationLoss\x12 \n\x11\x61nchorwise_output\x18\x01 \x01(\x08:\x05\x66\x61lse\x12\x10\n\x05\x64\x65lta\x18\x02 \x01(\x02:\x01\x31\"\x1d\n\x1bWeightedIOULocalizationLoss\"\x82\x04\n\x12\x43lassificationLoss\x12V\n\x10weighted_sigmoid\x18\x01 \x01(\x0b\x32:.object_detection.protos.WeightedSigmoidClassificationLossH\x00\x12V\n\x10weighted_softmax\x18\x02 \x01(\x0b\x32:.object_detection.protos.WeightedSoftmaxClassificationLossH\x00\x12j\n\x17weighted_logits_softmax\x18\x05 \x01(\x0b\x32G.object_detection.protos.WeightedSoftmaxClassificationAgainstLogitsLossH\x00\x12^\n\x14\x62ootstrapped_sigmoid\x18\x03 \x01(\x0b\x32>.object_detection.protos.BootstrappedSigmoidClassificationLossH\x00\x12Y\n\x16weighted_sigmoid_focal\x18\x04 \x01(\x0b\x32\x37.object_detection.protos.SigmoidFocalClassificationLossH\x00\x42\x15\n\x13\x63lassification_loss\"E\n!WeightedSigmoidClassificationLoss\x12 \n\x11\x61nchorwise_output\x18\x01 \x01(\x08:\x05\x66\x61lse\"c\n\x1eSigmoidFocalClassificationLoss\x12 \n\x11\x61nchorwise_output\x18\x01 \x01(\x08:\x05\x66\x61lse\x12\x10\n\x05gamma\x18\x02 \x01(\x02:\x01\x32\x12\r\n\x05\x61lpha\x18\x03 \x01(\x02\"]\n!WeightedSoftmaxClassificationLoss\x12 \n\x11\x61nchorwise_output\x18\x01 \x01(\x08:\x05\x66\x61lse\x12\x16\n\x0blogit_scale\x18\x02 \x01(\x02:\x01\x31\"j\n.WeightedSoftmaxClassificationAgainstLogitsLoss\x12 \n\x11\x61nchorwise_output\x18\x01 \x01(\x08:\x05\x66\x61lse\x12\x16\n\x0blogit_scale\x18\x02 \x01(\x02:\x01\x31\"w\n%BootstrappedSigmoidClassificationLoss\x12\r\n\x05\x61lpha\x18\x01 \x01(\x02\x12\x1d\n\x0ehard_bootstrap\x18\x02 \x01(\x08:\x05\x66\x61lse\x12 \n\x11\x61nchorwise_output\x18\x03 \x01(\x08:\x05\x66\x61lse\"\xa1\x02\n\x10HardExampleMiner\x12\x1d\n\x11num_hard_examples\x18\x01 \x01(\x05:\x02\x36\x34\x12\x1a\n\riou_threshold\x18\x02 \x01(\x02:\x03\x30.7\x12K\n\tloss_type\x18\x03 \x01(\x0e\x32\x32.object_detection.protos.HardExampleMiner.LossType:\x04\x42OTH\x12%\n\x1amax_negatives_per_positive\x18\x04 \x01(\x05:\x01\x30\x12\"\n\x17min_negatives_per_image\x18\x05 \x01(\x05:\x01\x30\":\n\x08LossType\x12\x08\n\x04\x42OTH\x10\x00\x12\x12\n\x0e\x43LASSIFICATION\x10\x01\x12\x10\n\x0cLOCALIZATION\x10\x02\">\n\x14RandomExampleSampler\x12&\n\x18positive_sample_fraction\x18\x01 \x01(\x02:\x04\x30.01') +) + + + +_HARDEXAMPLEMINER_LOSSTYPE = _descriptor.EnumDescriptor( + name='LossType', + full_name='object_detection.protos.HardExampleMiner.LossType', + filename=None, + file=DESCRIPTOR, + values=[ + _descriptor.EnumValueDescriptor( + name='BOTH', index=0, number=0, + options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='CLASSIFICATION', index=1, number=1, + options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='LOCALIZATION', index=2, number=2, + options=None, + type=None), + ], + containing_type=None, + options=None, + serialized_start=2147, + serialized_end=2205, +) +_sym_db.RegisterEnumDescriptor(_HARDEXAMPLEMINER_LOSSTYPE) + + +_LOSS = _descriptor.Descriptor( + name='Loss', + full_name='object_detection.protos.Loss', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='localization_loss', full_name='object_detection.protos.Loss.localization_loss', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='classification_loss', full_name='object_detection.protos.Loss.classification_loss', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='hard_example_miner', full_name='object_detection.protos.Loss.hard_example_miner', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='classification_weight', full_name='object_detection.protos.Loss.classification_weight', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='localization_weight', full_name='object_detection.protos.Loss.localization_weight', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='random_example_sampler', full_name='object_detection.protos.Loss.random_example_sampler', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=66, + serialized_end=432, +) + + +_LOCALIZATIONLOSS = _descriptor.Descriptor( + name='LocalizationLoss', + full_name='object_detection.protos.LocalizationLoss', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='weighted_l2', full_name='object_detection.protos.LocalizationLoss.weighted_l2', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='weighted_smooth_l1', full_name='object_detection.protos.LocalizationLoss.weighted_smooth_l1', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='weighted_iou', full_name='object_detection.protos.LocalizationLoss.weighted_iou', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + _descriptor.OneofDescriptor( + name='localization_loss', full_name='object_detection.protos.LocalizationLoss.localization_loss', + index=0, containing_type=None, fields=[]), + ], + serialized_start=435, + serialized_end=717, +) + + +_WEIGHTEDL2LOCALIZATIONLOSS = _descriptor.Descriptor( + name='WeightedL2LocalizationLoss', + full_name='object_detection.protos.WeightedL2LocalizationLoss', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='anchorwise_output', full_name='object_detection.protos.WeightedL2LocalizationLoss.anchorwise_output', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=719, + serialized_end=781, +) + + +_WEIGHTEDSMOOTHL1LOCALIZATIONLOSS = _descriptor.Descriptor( + name='WeightedSmoothL1LocalizationLoss', + full_name='object_detection.protos.WeightedSmoothL1LocalizationLoss', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='anchorwise_output', full_name='object_detection.protos.WeightedSmoothL1LocalizationLoss.anchorwise_output', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='delta', full_name='object_detection.protos.WeightedSmoothL1LocalizationLoss.delta', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=783, + serialized_end=869, +) + + +_WEIGHTEDIOULOCALIZATIONLOSS = _descriptor.Descriptor( + name='WeightedIOULocalizationLoss', + full_name='object_detection.protos.WeightedIOULocalizationLoss', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=871, + serialized_end=900, +) + + +_CLASSIFICATIONLOSS = _descriptor.Descriptor( + name='ClassificationLoss', + full_name='object_detection.protos.ClassificationLoss', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='weighted_sigmoid', full_name='object_detection.protos.ClassificationLoss.weighted_sigmoid', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='weighted_softmax', full_name='object_detection.protos.ClassificationLoss.weighted_softmax', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='weighted_logits_softmax', full_name='object_detection.protos.ClassificationLoss.weighted_logits_softmax', index=2, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='bootstrapped_sigmoid', full_name='object_detection.protos.ClassificationLoss.bootstrapped_sigmoid', index=3, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='weighted_sigmoid_focal', full_name='object_detection.protos.ClassificationLoss.weighted_sigmoid_focal', index=4, + 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + _descriptor.OneofDescriptor( + name='classification_loss', full_name='object_detection.protos.ClassificationLoss.classification_loss', + index=0, containing_type=None, fields=[]), + ], + serialized_start=903, + serialized_end=1417, +) + + +_WEIGHTEDSIGMOIDCLASSIFICATIONLOSS = _descriptor.Descriptor( + name='WeightedSigmoidClassificationLoss', + full_name='object_detection.protos.WeightedSigmoidClassificationLoss', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='anchorwise_output', full_name='object_detection.protos.WeightedSigmoidClassificationLoss.anchorwise_output', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=1419, + serialized_end=1488, +) + + +_SIGMOIDFOCALCLASSIFICATIONLOSS = _descriptor.Descriptor( + name='SigmoidFocalClassificationLoss', + full_name='object_detection.protos.SigmoidFocalClassificationLoss', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='anchorwise_output', full_name='object_detection.protos.SigmoidFocalClassificationLoss.anchorwise_output', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='gamma', full_name='object_detection.protos.SigmoidFocalClassificationLoss.gamma', index=1, + number=2, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(2), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='alpha', full_name='object_detection.protos.SigmoidFocalClassificationLoss.alpha', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=1490, + serialized_end=1589, +) + + +_WEIGHTEDSOFTMAXCLASSIFICATIONLOSS = _descriptor.Descriptor( + name='WeightedSoftmaxClassificationLoss', + full_name='object_detection.protos.WeightedSoftmaxClassificationLoss', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='anchorwise_output', full_name='object_detection.protos.WeightedSoftmaxClassificationLoss.anchorwise_output', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='logit_scale', full_name='object_detection.protos.WeightedSoftmaxClassificationLoss.logit_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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=1591, + serialized_end=1684, +) + + +_WEIGHTEDSOFTMAXCLASSIFICATIONAGAINSTLOGITSLOSS = _descriptor.Descriptor( + name='WeightedSoftmaxClassificationAgainstLogitsLoss', + full_name='object_detection.protos.WeightedSoftmaxClassificationAgainstLogitsLoss', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='anchorwise_output', full_name='object_detection.protos.WeightedSoftmaxClassificationAgainstLogitsLoss.anchorwise_output', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='logit_scale', full_name='object_detection.protos.WeightedSoftmaxClassificationAgainstLogitsLoss.logit_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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=1686, + serialized_end=1792, +) + + +_BOOTSTRAPPEDSIGMOIDCLASSIFICATIONLOSS = _descriptor.Descriptor( + name='BootstrappedSigmoidClassificationLoss', + full_name='object_detection.protos.BootstrappedSigmoidClassificationLoss', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='alpha', full_name='object_detection.protos.BootstrappedSigmoidClassificationLoss.alpha', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='hard_bootstrap', full_name='object_detection.protos.BootstrappedSigmoidClassificationLoss.hard_bootstrap', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='anchorwise_output', full_name='object_detection.protos.BootstrappedSigmoidClassificationLoss.anchorwise_output', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=1794, + serialized_end=1913, +) + + +_HARDEXAMPLEMINER = _descriptor.Descriptor( + name='HardExampleMiner', + full_name='object_detection.protos.HardExampleMiner', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='num_hard_examples', full_name='object_detection.protos.HardExampleMiner.num_hard_examples', index=0, + number=1, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='iou_threshold', full_name='object_detection.protos.HardExampleMiner.iou_threshold', index=1, + number=2, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(0.7), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='loss_type', full_name='object_detection.protos.HardExampleMiner.loss_type', index=2, + number=3, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_negatives_per_positive', full_name='object_detection.protos.HardExampleMiner.max_negatives_per_positive', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='min_negatives_per_image', full_name='object_detection.protos.HardExampleMiner.min_negatives_per_image', index=4, + number=5, 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + _HARDEXAMPLEMINER_LOSSTYPE, + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=1916, + serialized_end=2205, +) + + +_RANDOMEXAMPLESAMPLER = _descriptor.Descriptor( + name='RandomExampleSampler', + full_name='object_detection.protos.RandomExampleSampler', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='positive_sample_fraction', full_name='object_detection.protos.RandomExampleSampler.positive_sample_fraction', index=0, + number=1, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(0.01), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=2207, + serialized_end=2269, +) + +_LOSS.fields_by_name['localization_loss'].message_type = _LOCALIZATIONLOSS +_LOSS.fields_by_name['classification_loss'].message_type = _CLASSIFICATIONLOSS +_LOSS.fields_by_name['hard_example_miner'].message_type = _HARDEXAMPLEMINER +_LOSS.fields_by_name['random_example_sampler'].message_type = _RANDOMEXAMPLESAMPLER +_LOCALIZATIONLOSS.fields_by_name['weighted_l2'].message_type = _WEIGHTEDL2LOCALIZATIONLOSS +_LOCALIZATIONLOSS.fields_by_name['weighted_smooth_l1'].message_type = _WEIGHTEDSMOOTHL1LOCALIZATIONLOSS +_LOCALIZATIONLOSS.fields_by_name['weighted_iou'].message_type = _WEIGHTEDIOULOCALIZATIONLOSS +_LOCALIZATIONLOSS.oneofs_by_name['localization_loss'].fields.append( + _LOCALIZATIONLOSS.fields_by_name['weighted_l2']) +_LOCALIZATIONLOSS.fields_by_name['weighted_l2'].containing_oneof = _LOCALIZATIONLOSS.oneofs_by_name['localization_loss'] +_LOCALIZATIONLOSS.oneofs_by_name['localization_loss'].fields.append( + _LOCALIZATIONLOSS.fields_by_name['weighted_smooth_l1']) +_LOCALIZATIONLOSS.fields_by_name['weighted_smooth_l1'].containing_oneof = _LOCALIZATIONLOSS.oneofs_by_name['localization_loss'] +_LOCALIZATIONLOSS.oneofs_by_name['localization_loss'].fields.append( + _LOCALIZATIONLOSS.fields_by_name['weighted_iou']) +_LOCALIZATIONLOSS.fields_by_name['weighted_iou'].containing_oneof = _LOCALIZATIONLOSS.oneofs_by_name['localization_loss'] +_CLASSIFICATIONLOSS.fields_by_name['weighted_sigmoid'].message_type = _WEIGHTEDSIGMOIDCLASSIFICATIONLOSS +_CLASSIFICATIONLOSS.fields_by_name['weighted_softmax'].message_type = _WEIGHTEDSOFTMAXCLASSIFICATIONLOSS +_CLASSIFICATIONLOSS.fields_by_name['weighted_logits_softmax'].message_type = _WEIGHTEDSOFTMAXCLASSIFICATIONAGAINSTLOGITSLOSS +_CLASSIFICATIONLOSS.fields_by_name['bootstrapped_sigmoid'].message_type = _BOOTSTRAPPEDSIGMOIDCLASSIFICATIONLOSS +_CLASSIFICATIONLOSS.fields_by_name['weighted_sigmoid_focal'].message_type = _SIGMOIDFOCALCLASSIFICATIONLOSS +_CLASSIFICATIONLOSS.oneofs_by_name['classification_loss'].fields.append( + _CLASSIFICATIONLOSS.fields_by_name['weighted_sigmoid']) +_CLASSIFICATIONLOSS.fields_by_name['weighted_sigmoid'].containing_oneof = _CLASSIFICATIONLOSS.oneofs_by_name['classification_loss'] +_CLASSIFICATIONLOSS.oneofs_by_name['classification_loss'].fields.append( + _CLASSIFICATIONLOSS.fields_by_name['weighted_softmax']) +_CLASSIFICATIONLOSS.fields_by_name['weighted_softmax'].containing_oneof = _CLASSIFICATIONLOSS.oneofs_by_name['classification_loss'] +_CLASSIFICATIONLOSS.oneofs_by_name['classification_loss'].fields.append( + _CLASSIFICATIONLOSS.fields_by_name['weighted_logits_softmax']) +_CLASSIFICATIONLOSS.fields_by_name['weighted_logits_softmax'].containing_oneof = _CLASSIFICATIONLOSS.oneofs_by_name['classification_loss'] +_CLASSIFICATIONLOSS.oneofs_by_name['classification_loss'].fields.append( + _CLASSIFICATIONLOSS.fields_by_name['bootstrapped_sigmoid']) +_CLASSIFICATIONLOSS.fields_by_name['bootstrapped_sigmoid'].containing_oneof = _CLASSIFICATIONLOSS.oneofs_by_name['classification_loss'] +_CLASSIFICATIONLOSS.oneofs_by_name['classification_loss'].fields.append( + _CLASSIFICATIONLOSS.fields_by_name['weighted_sigmoid_focal']) +_CLASSIFICATIONLOSS.fields_by_name['weighted_sigmoid_focal'].containing_oneof = _CLASSIFICATIONLOSS.oneofs_by_name['classification_loss'] +_HARDEXAMPLEMINER.fields_by_name['loss_type'].enum_type = _HARDEXAMPLEMINER_LOSSTYPE +_HARDEXAMPLEMINER_LOSSTYPE.containing_type = _HARDEXAMPLEMINER +DESCRIPTOR.message_types_by_name['Loss'] = _LOSS +DESCRIPTOR.message_types_by_name['LocalizationLoss'] = _LOCALIZATIONLOSS +DESCRIPTOR.message_types_by_name['WeightedL2LocalizationLoss'] = _WEIGHTEDL2LOCALIZATIONLOSS +DESCRIPTOR.message_types_by_name['WeightedSmoothL1LocalizationLoss'] = _WEIGHTEDSMOOTHL1LOCALIZATIONLOSS +DESCRIPTOR.message_types_by_name['WeightedIOULocalizationLoss'] = _WEIGHTEDIOULOCALIZATIONLOSS +DESCRIPTOR.message_types_by_name['ClassificationLoss'] = _CLASSIFICATIONLOSS +DESCRIPTOR.message_types_by_name['WeightedSigmoidClassificationLoss'] = _WEIGHTEDSIGMOIDCLASSIFICATIONLOSS +DESCRIPTOR.message_types_by_name['SigmoidFocalClassificationLoss'] = _SIGMOIDFOCALCLASSIFICATIONLOSS +DESCRIPTOR.message_types_by_name['WeightedSoftmaxClassificationLoss'] = _WEIGHTEDSOFTMAXCLASSIFICATIONLOSS +DESCRIPTOR.message_types_by_name['WeightedSoftmaxClassificationAgainstLogitsLoss'] = _WEIGHTEDSOFTMAXCLASSIFICATIONAGAINSTLOGITSLOSS +DESCRIPTOR.message_types_by_name['BootstrappedSigmoidClassificationLoss'] = _BOOTSTRAPPEDSIGMOIDCLASSIFICATIONLOSS +DESCRIPTOR.message_types_by_name['HardExampleMiner'] = _HARDEXAMPLEMINER +DESCRIPTOR.message_types_by_name['RandomExampleSampler'] = _RANDOMEXAMPLESAMPLER +_sym_db.RegisterFileDescriptor(DESCRIPTOR) + +Loss = _reflection.GeneratedProtocolMessageType('Loss', (_message.Message,), dict( + DESCRIPTOR = _LOSS, + __module__ = 'object_detection.protos.losses_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.Loss) + )) +_sym_db.RegisterMessage(Loss) + +LocalizationLoss = _reflection.GeneratedProtocolMessageType('LocalizationLoss', (_message.Message,), dict( + DESCRIPTOR = _LOCALIZATIONLOSS, + __module__ = 'object_detection.protos.losses_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.LocalizationLoss) + )) +_sym_db.RegisterMessage(LocalizationLoss) + +WeightedL2LocalizationLoss = _reflection.GeneratedProtocolMessageType('WeightedL2LocalizationLoss', (_message.Message,), dict( + DESCRIPTOR = _WEIGHTEDL2LOCALIZATIONLOSS, + __module__ = 'object_detection.protos.losses_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.WeightedL2LocalizationLoss) + )) +_sym_db.RegisterMessage(WeightedL2LocalizationLoss) + +WeightedSmoothL1LocalizationLoss = _reflection.GeneratedProtocolMessageType('WeightedSmoothL1LocalizationLoss', (_message.Message,), dict( + DESCRIPTOR = _WEIGHTEDSMOOTHL1LOCALIZATIONLOSS, + __module__ = 'object_detection.protos.losses_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.WeightedSmoothL1LocalizationLoss) + )) +_sym_db.RegisterMessage(WeightedSmoothL1LocalizationLoss) + +WeightedIOULocalizationLoss = _reflection.GeneratedProtocolMessageType('WeightedIOULocalizationLoss', (_message.Message,), dict( + DESCRIPTOR = _WEIGHTEDIOULOCALIZATIONLOSS, + __module__ = 'object_detection.protos.losses_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.WeightedIOULocalizationLoss) + )) +_sym_db.RegisterMessage(WeightedIOULocalizationLoss) + +ClassificationLoss = _reflection.GeneratedProtocolMessageType('ClassificationLoss', (_message.Message,), dict( + DESCRIPTOR = _CLASSIFICATIONLOSS, + __module__ = 'object_detection.protos.losses_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.ClassificationLoss) + )) +_sym_db.RegisterMessage(ClassificationLoss) + +WeightedSigmoidClassificationLoss = _reflection.GeneratedProtocolMessageType('WeightedSigmoidClassificationLoss', (_message.Message,), dict( + DESCRIPTOR = _WEIGHTEDSIGMOIDCLASSIFICATIONLOSS, + __module__ = 'object_detection.protos.losses_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.WeightedSigmoidClassificationLoss) + )) +_sym_db.RegisterMessage(WeightedSigmoidClassificationLoss) + +SigmoidFocalClassificationLoss = _reflection.GeneratedProtocolMessageType('SigmoidFocalClassificationLoss', (_message.Message,), dict( + DESCRIPTOR = _SIGMOIDFOCALCLASSIFICATIONLOSS, + __module__ = 'object_detection.protos.losses_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.SigmoidFocalClassificationLoss) + )) +_sym_db.RegisterMessage(SigmoidFocalClassificationLoss) + +WeightedSoftmaxClassificationLoss = _reflection.GeneratedProtocolMessageType('WeightedSoftmaxClassificationLoss', (_message.Message,), dict( + DESCRIPTOR = _WEIGHTEDSOFTMAXCLASSIFICATIONLOSS, + __module__ = 'object_detection.protos.losses_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.WeightedSoftmaxClassificationLoss) + )) +_sym_db.RegisterMessage(WeightedSoftmaxClassificationLoss) + +WeightedSoftmaxClassificationAgainstLogitsLoss = _reflection.GeneratedProtocolMessageType('WeightedSoftmaxClassificationAgainstLogitsLoss', (_message.Message,), dict( + DESCRIPTOR = _WEIGHTEDSOFTMAXCLASSIFICATIONAGAINSTLOGITSLOSS, + __module__ = 'object_detection.protos.losses_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.WeightedSoftmaxClassificationAgainstLogitsLoss) + )) +_sym_db.RegisterMessage(WeightedSoftmaxClassificationAgainstLogitsLoss) + +BootstrappedSigmoidClassificationLoss = _reflection.GeneratedProtocolMessageType('BootstrappedSigmoidClassificationLoss', (_message.Message,), dict( + DESCRIPTOR = _BOOTSTRAPPEDSIGMOIDCLASSIFICATIONLOSS, + __module__ = 'object_detection.protos.losses_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.BootstrappedSigmoidClassificationLoss) + )) +_sym_db.RegisterMessage(BootstrappedSigmoidClassificationLoss) + +HardExampleMiner = _reflection.GeneratedProtocolMessageType('HardExampleMiner', (_message.Message,), dict( + DESCRIPTOR = _HARDEXAMPLEMINER, + __module__ = 'object_detection.protos.losses_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.HardExampleMiner) + )) +_sym_db.RegisterMessage(HardExampleMiner) + +RandomExampleSampler = _reflection.GeneratedProtocolMessageType('RandomExampleSampler', (_message.Message,), dict( + DESCRIPTOR = _RANDOMEXAMPLESAMPLER, + __module__ = 'object_detection.protos.losses_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.RandomExampleSampler) + )) +_sym_db.RegisterMessage(RandomExampleSampler) + + +# @@protoc_insertion_point(module_scope) diff --git a/research/object_detection/protos/matcher.proto b/research/object_detection/object_detection/protos/matcher.proto similarity index 100% rename from research/object_detection/protos/matcher.proto rename to research/object_detection/object_detection/protos/matcher.proto diff --git a/research/object_detection/object_detection/protos/matcher_pb2.py b/research/object_detection/object_detection/protos/matcher_pb2.py new file mode 100644 index 00000000000..c422bb810a1 --- /dev/null +++ b/research/object_detection/object_detection/protos/matcher_pb2.py @@ -0,0 +1,90 @@ +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: object_detection/protos/matcher.proto + +import sys +_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) +from google.protobuf import descriptor as _descriptor +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +from google.protobuf import descriptor_pb2 +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + +from object_detection.protos import argmax_matcher_pb2 as object__detection_dot_protos_dot_argmax__matcher__pb2 +from object_detection.protos import bipartite_matcher_pb2 as object__detection_dot_protos_dot_bipartite__matcher__pb2 + + +DESCRIPTOR = _descriptor.FileDescriptor( + name='object_detection/protos/matcher.proto', + package='object_detection.protos', + syntax='proto2', + serialized_pb=_b('\n%object_detection/protos/matcher.proto\x12\x17object_detection.protos\x1a,object_detection/protos/argmax_matcher.proto\x1a/object_detection/protos/bipartite_matcher.proto\"\xa4\x01\n\x07Matcher\x12@\n\x0e\x61rgmax_matcher\x18\x01 \x01(\x0b\x32&.object_detection.protos.ArgMaxMatcherH\x00\x12\x46\n\x11\x62ipartite_matcher\x18\x02 \x01(\x0b\x32).object_detection.protos.BipartiteMatcherH\x00\x42\x0f\n\rmatcher_oneof') + , + dependencies=[object__detection_dot_protos_dot_argmax__matcher__pb2.DESCRIPTOR,object__detection_dot_protos_dot_bipartite__matcher__pb2.DESCRIPTOR,]) + + + + +_MATCHER = _descriptor.Descriptor( + name='Matcher', + full_name='object_detection.protos.Matcher', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='argmax_matcher', full_name='object_detection.protos.Matcher.argmax_matcher', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='bipartite_matcher', full_name='object_detection.protos.Matcher.bipartite_matcher', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + _descriptor.OneofDescriptor( + name='matcher_oneof', full_name='object_detection.protos.Matcher.matcher_oneof', + index=0, containing_type=None, fields=[]), + ], + serialized_start=162, + serialized_end=326, +) + +_MATCHER.fields_by_name['argmax_matcher'].message_type = object__detection_dot_protos_dot_argmax__matcher__pb2._ARGMAXMATCHER +_MATCHER.fields_by_name['bipartite_matcher'].message_type = object__detection_dot_protos_dot_bipartite__matcher__pb2._BIPARTITEMATCHER +_MATCHER.oneofs_by_name['matcher_oneof'].fields.append( + _MATCHER.fields_by_name['argmax_matcher']) +_MATCHER.fields_by_name['argmax_matcher'].containing_oneof = _MATCHER.oneofs_by_name['matcher_oneof'] +_MATCHER.oneofs_by_name['matcher_oneof'].fields.append( + _MATCHER.fields_by_name['bipartite_matcher']) +_MATCHER.fields_by_name['bipartite_matcher'].containing_oneof = _MATCHER.oneofs_by_name['matcher_oneof'] +DESCRIPTOR.message_types_by_name['Matcher'] = _MATCHER +_sym_db.RegisterFileDescriptor(DESCRIPTOR) + +Matcher = _reflection.GeneratedProtocolMessageType('Matcher', (_message.Message,), dict( + DESCRIPTOR = _MATCHER, + __module__ = 'object_detection.protos.matcher_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.Matcher) + )) +_sym_db.RegisterMessage(Matcher) + + +# @@protoc_insertion_point(module_scope) diff --git a/research/object_detection/protos/mean_stddev_box_coder.proto b/research/object_detection/object_detection/protos/mean_stddev_box_coder.proto similarity index 100% rename from research/object_detection/protos/mean_stddev_box_coder.proto rename to research/object_detection/object_detection/protos/mean_stddev_box_coder.proto diff --git a/research/object_detection/object_detection/protos/mean_stddev_box_coder_pb2.py b/research/object_detection/object_detection/protos/mean_stddev_box_coder_pb2.py new file mode 100644 index 00000000000..9c34456034e --- /dev/null +++ b/research/object_detection/object_detection/protos/mean_stddev_box_coder_pb2.py @@ -0,0 +1,69 @@ +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: object_detection/protos/mean_stddev_box_coder.proto + +import sys +_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) +from google.protobuf import descriptor as _descriptor +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +from google.protobuf import descriptor_pb2 +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor.FileDescriptor( + name='object_detection/protos/mean_stddev_box_coder.proto', + package='object_detection.protos', + syntax='proto2', + serialized_pb=_b('\n3object_detection/protos/mean_stddev_box_coder.proto\x12\x17object_detection.protos\"*\n\x12MeanStddevBoxCoder\x12\x14\n\x06stddev\x18\x01 \x01(\x02:\x04\x30.01') +) + + + + +_MEANSTDDEVBOXCODER = _descriptor.Descriptor( + name='MeanStddevBoxCoder', + full_name='object_detection.protos.MeanStddevBoxCoder', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='stddev', full_name='object_detection.protos.MeanStddevBoxCoder.stddev', index=0, + number=1, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(0.01), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=80, + serialized_end=122, +) + +DESCRIPTOR.message_types_by_name['MeanStddevBoxCoder'] = _MEANSTDDEVBOXCODER +_sym_db.RegisterFileDescriptor(DESCRIPTOR) + +MeanStddevBoxCoder = _reflection.GeneratedProtocolMessageType('MeanStddevBoxCoder', (_message.Message,), dict( + DESCRIPTOR = _MEANSTDDEVBOXCODER, + __module__ = 'object_detection.protos.mean_stddev_box_coder_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.MeanStddevBoxCoder) + )) +_sym_db.RegisterMessage(MeanStddevBoxCoder) + + +# @@protoc_insertion_point(module_scope) diff --git a/research/object_detection/protos/model.proto b/research/object_detection/object_detection/protos/model.proto similarity index 100% rename from research/object_detection/protos/model.proto rename to research/object_detection/object_detection/protos/model.proto diff --git a/research/object_detection/object_detection/protos/model_pb2.py b/research/object_detection/object_detection/protos/model_pb2.py new file mode 100644 index 00000000000..54ec4673bb6 --- /dev/null +++ b/research/object_detection/object_detection/protos/model_pb2.py @@ -0,0 +1,90 @@ +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: object_detection/protos/model.proto + +import sys +_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) +from google.protobuf import descriptor as _descriptor +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +from google.protobuf import descriptor_pb2 +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + +from object_detection.protos import faster_rcnn_pb2 as object__detection_dot_protos_dot_faster__rcnn__pb2 +from object_detection.protos import ssd_pb2 as object__detection_dot_protos_dot_ssd__pb2 + + +DESCRIPTOR = _descriptor.FileDescriptor( + name='object_detection/protos/model.proto', + package='object_detection.protos', + syntax='proto2', + serialized_pb=_b('\n#object_detection/protos/model.proto\x12\x17object_detection.protos\x1a)object_detection/protos/faster_rcnn.proto\x1a!object_detection/protos/ssd.proto\"\x82\x01\n\x0e\x44\x65tectionModel\x12:\n\x0b\x66\x61ster_rcnn\x18\x01 \x01(\x0b\x32#.object_detection.protos.FasterRcnnH\x00\x12+\n\x03ssd\x18\x02 \x01(\x0b\x32\x1c.object_detection.protos.SsdH\x00\x42\x07\n\x05model') + , + dependencies=[object__detection_dot_protos_dot_faster__rcnn__pb2.DESCRIPTOR,object__detection_dot_protos_dot_ssd__pb2.DESCRIPTOR,]) + + + + +_DETECTIONMODEL = _descriptor.Descriptor( + name='DetectionModel', + full_name='object_detection.protos.DetectionModel', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='faster_rcnn', full_name='object_detection.protos.DetectionModel.faster_rcnn', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='ssd', full_name='object_detection.protos.DetectionModel.ssd', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + _descriptor.OneofDescriptor( + name='model', full_name='object_detection.protos.DetectionModel.model', + index=0, containing_type=None, fields=[]), + ], + serialized_start=143, + serialized_end=273, +) + +_DETECTIONMODEL.fields_by_name['faster_rcnn'].message_type = object__detection_dot_protos_dot_faster__rcnn__pb2._FASTERRCNN +_DETECTIONMODEL.fields_by_name['ssd'].message_type = object__detection_dot_protos_dot_ssd__pb2._SSD +_DETECTIONMODEL.oneofs_by_name['model'].fields.append( + _DETECTIONMODEL.fields_by_name['faster_rcnn']) +_DETECTIONMODEL.fields_by_name['faster_rcnn'].containing_oneof = _DETECTIONMODEL.oneofs_by_name['model'] +_DETECTIONMODEL.oneofs_by_name['model'].fields.append( + _DETECTIONMODEL.fields_by_name['ssd']) +_DETECTIONMODEL.fields_by_name['ssd'].containing_oneof = _DETECTIONMODEL.oneofs_by_name['model'] +DESCRIPTOR.message_types_by_name['DetectionModel'] = _DETECTIONMODEL +_sym_db.RegisterFileDescriptor(DESCRIPTOR) + +DetectionModel = _reflection.GeneratedProtocolMessageType('DetectionModel', (_message.Message,), dict( + DESCRIPTOR = _DETECTIONMODEL, + __module__ = 'object_detection.protos.model_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.DetectionModel) + )) +_sym_db.RegisterMessage(DetectionModel) + + +# @@protoc_insertion_point(module_scope) diff --git a/research/object_detection/protos/multiscale_anchor_generator.proto b/research/object_detection/object_detection/protos/multiscale_anchor_generator.proto similarity index 100% rename from research/object_detection/protos/multiscale_anchor_generator.proto rename to research/object_detection/object_detection/protos/multiscale_anchor_generator.proto diff --git a/research/object_detection/object_detection/protos/multiscale_anchor_generator_pb2.py b/research/object_detection/object_detection/protos/multiscale_anchor_generator_pb2.py new file mode 100644 index 00000000000..875c5a80041 --- /dev/null +++ b/research/object_detection/object_detection/protos/multiscale_anchor_generator_pb2.py @@ -0,0 +1,104 @@ +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: object_detection/protos/multiscale_anchor_generator.proto + +import sys +_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) +from google.protobuf import descriptor as _descriptor +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +from google.protobuf import descriptor_pb2 +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor.FileDescriptor( + name='object_detection/protos/multiscale_anchor_generator.proto', + package='object_detection.protos', + syntax='proto2', + serialized_pb=_b('\n9object_detection/protos/multiscale_anchor_generator.proto\x12\x17object_detection.protos\"\xba\x01\n\x19MultiscaleAnchorGenerator\x12\x14\n\tmin_level\x18\x01 \x01(\x05:\x01\x33\x12\x14\n\tmax_level\x18\x02 \x01(\x05:\x01\x37\x12\x17\n\x0c\x61nchor_scale\x18\x03 \x01(\x02:\x01\x34\x12\x15\n\raspect_ratios\x18\x04 \x03(\x02\x12\x1c\n\x11scales_per_octave\x18\x05 \x01(\x05:\x01\x32\x12#\n\x15normalize_coordinates\x18\x06 \x01(\x08:\x04true') +) + + + + +_MULTISCALEANCHORGENERATOR = _descriptor.Descriptor( + name='MultiscaleAnchorGenerator', + full_name='object_detection.protos.MultiscaleAnchorGenerator', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='min_level', full_name='object_detection.protos.MultiscaleAnchorGenerator.min_level', index=0, + number=1, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=3, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_level', full_name='object_detection.protos.MultiscaleAnchorGenerator.max_level', index=1, + number=2, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=7, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='anchor_scale', full_name='object_detection.protos.MultiscaleAnchorGenerator.anchor_scale', index=2, + number=3, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(4), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='aspect_ratios', full_name='object_detection.protos.MultiscaleAnchorGenerator.aspect_ratios', index=3, + number=4, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='scales_per_octave', full_name='object_detection.protos.MultiscaleAnchorGenerator.scales_per_octave', index=4, + number=5, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='normalize_coordinates', full_name='object_detection.protos.MultiscaleAnchorGenerator.normalize_coordinates', index=5, + number=6, 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=87, + serialized_end=273, +) + +DESCRIPTOR.message_types_by_name['MultiscaleAnchorGenerator'] = _MULTISCALEANCHORGENERATOR +_sym_db.RegisterFileDescriptor(DESCRIPTOR) + +MultiscaleAnchorGenerator = _reflection.GeneratedProtocolMessageType('MultiscaleAnchorGenerator', (_message.Message,), dict( + DESCRIPTOR = _MULTISCALEANCHORGENERATOR, + __module__ = 'object_detection.protos.multiscale_anchor_generator_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.MultiscaleAnchorGenerator) + )) +_sym_db.RegisterMessage(MultiscaleAnchorGenerator) + + +# @@protoc_insertion_point(module_scope) diff --git a/research/object_detection/protos/optimizer.proto b/research/object_detection/object_detection/protos/optimizer.proto similarity index 100% rename from research/object_detection/protos/optimizer.proto rename to research/object_detection/object_detection/protos/optimizer.proto diff --git a/research/object_detection/object_detection/protos/optimizer_pb2.py b/research/object_detection/object_detection/protos/optimizer_pb2.py new file mode 100644 index 00000000000..00beb94e176 --- /dev/null +++ b/research/object_detection/object_detection/protos/optimizer_pb2.py @@ -0,0 +1,626 @@ +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: object_detection/protos/optimizer.proto + +import sys +_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) +from google.protobuf import descriptor as _descriptor +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +from google.protobuf import descriptor_pb2 +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor.FileDescriptor( + name='object_detection/protos/optimizer.proto', + package='object_detection.protos', + syntax='proto2', + serialized_pb=_b('\n\'object_detection/protos/optimizer.proto\x12\x17object_detection.protos\"\xb5\x02\n\tOptimizer\x12G\n\x12rms_prop_optimizer\x18\x01 \x01(\x0b\x32).object_detection.protos.RMSPropOptimizerH\x00\x12H\n\x12momentum_optimizer\x18\x02 \x01(\x0b\x32*.object_detection.protos.MomentumOptimizerH\x00\x12@\n\x0e\x61\x64\x61m_optimizer\x18\x03 \x01(\x0b\x32&.object_detection.protos.AdamOptimizerH\x00\x12 \n\x12use_moving_average\x18\x04 \x01(\x08:\x04true\x12$\n\x14moving_average_decay\x18\x05 \x01(\x02:\x06\x30.9999B\x0b\n\toptimizer\"\x9f\x01\n\x10RMSPropOptimizer\x12<\n\rlearning_rate\x18\x01 \x01(\x0b\x32%.object_detection.protos.LearningRate\x12%\n\x18momentum_optimizer_value\x18\x02 \x01(\x02:\x03\x30.9\x12\x12\n\x05\x64\x65\x63\x61y\x18\x03 \x01(\x02:\x03\x30.9\x12\x12\n\x07\x65psilon\x18\x04 \x01(\x02:\x01\x31\"x\n\x11MomentumOptimizer\x12<\n\rlearning_rate\x18\x01 \x01(\x0b\x32%.object_detection.protos.LearningRate\x12%\n\x18momentum_optimizer_value\x18\x02 \x01(\x02:\x03\x30.9\"M\n\rAdamOptimizer\x12<\n\rlearning_rate\x18\x01 \x01(\x0b\x32%.object_detection.protos.LearningRate\"\x80\x03\n\x0cLearningRate\x12O\n\x16\x63onstant_learning_rate\x18\x01 \x01(\x0b\x32-.object_detection.protos.ConstantLearningRateH\x00\x12`\n\x1f\x65xponential_decay_learning_rate\x18\x02 \x01(\x0b\x32\x35.object_detection.protos.ExponentialDecayLearningRateH\x00\x12T\n\x19manual_step_learning_rate\x18\x03 \x01(\x0b\x32/.object_detection.protos.ManualStepLearningRateH\x00\x12V\n\x1a\x63osine_decay_learning_rate\x18\x04 \x01(\x0b\x32\x30.object_detection.protos.CosineDecayLearningRateH\x00\x42\x0f\n\rlearning_rate\"4\n\x14\x43onstantLearningRate\x12\x1c\n\rlearning_rate\x18\x01 \x01(\x02:\x05\x30.002\"\xef\x01\n\x1c\x45xponentialDecayLearningRate\x12$\n\x15initial_learning_rate\x18\x01 \x01(\x02:\x05\x30.002\x12\x1c\n\x0b\x64\x65\x63\x61y_steps\x18\x02 \x01(\r:\x07\x34\x30\x30\x30\x30\x30\x30\x12\x1a\n\x0c\x64\x65\x63\x61y_factor\x18\x03 \x01(\x02:\x04\x30.95\x12\x17\n\tstaircase\x18\x04 \x01(\x08:\x04true\x12\x1f\n\x14\x62urnin_learning_rate\x18\x05 \x01(\x02:\x01\x30\x12\x17\n\x0c\x62urnin_steps\x18\x06 \x01(\r:\x01\x30\x12\x1c\n\x11min_learning_rate\x18\x07 \x01(\x02:\x01\x30\"\xf1\x01\n\x16ManualStepLearningRate\x12$\n\x15initial_learning_rate\x18\x01 \x01(\x02:\x05\x30.002\x12V\n\x08schedule\x18\x02 \x03(\x0b\x32\x44.object_detection.protos.ManualStepLearningRate.LearningRateSchedule\x12\x15\n\x06warmup\x18\x03 \x01(\x08:\x05\x66\x61lse\x1a\x42\n\x14LearningRateSchedule\x12\x0c\n\x04step\x18\x01 \x01(\r\x12\x1c\n\rlearning_rate\x18\x02 \x01(\x02:\x05\x30.002\"\xbe\x01\n\x17\x43osineDecayLearningRate\x12!\n\x12learning_rate_base\x18\x01 \x01(\x02:\x05\x30.002\x12\x1c\n\x0btotal_steps\x18\x02 \x01(\r:\x07\x34\x30\x30\x30\x30\x30\x30\x12$\n\x14warmup_learning_rate\x18\x03 \x01(\x02:\x06\x30.0002\x12\x1b\n\x0cwarmup_steps\x18\x04 \x01(\r:\x05\x31\x30\x30\x30\x30\x12\x1f\n\x14hold_base_rate_steps\x18\x05 \x01(\r:\x01\x30') +) + + + + +_OPTIMIZER = _descriptor.Descriptor( + name='Optimizer', + full_name='object_detection.protos.Optimizer', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='rms_prop_optimizer', full_name='object_detection.protos.Optimizer.rms_prop_optimizer', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='momentum_optimizer', full_name='object_detection.protos.Optimizer.momentum_optimizer', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='adam_optimizer', full_name='object_detection.protos.Optimizer.adam_optimizer', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='use_moving_average', full_name='object_detection.protos.Optimizer.use_moving_average', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='moving_average_decay', full_name='object_detection.protos.Optimizer.moving_average_decay', index=4, + number=5, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(0.9999), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + _descriptor.OneofDescriptor( + name='optimizer', full_name='object_detection.protos.Optimizer.optimizer', + index=0, containing_type=None, fields=[]), + ], + serialized_start=69, + serialized_end=378, +) + + +_RMSPROPOPTIMIZER = _descriptor.Descriptor( + name='RMSPropOptimizer', + full_name='object_detection.protos.RMSPropOptimizer', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='learning_rate', full_name='object_detection.protos.RMSPropOptimizer.learning_rate', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='momentum_optimizer_value', full_name='object_detection.protos.RMSPropOptimizer.momentum_optimizer_value', index=1, + number=2, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(0.9), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='decay', full_name='object_detection.protos.RMSPropOptimizer.decay', index=2, + number=3, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(0.9), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='epsilon', full_name='object_detection.protos.RMSPropOptimizer.epsilon', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=381, + serialized_end=540, +) + + +_MOMENTUMOPTIMIZER = _descriptor.Descriptor( + name='MomentumOptimizer', + full_name='object_detection.protos.MomentumOptimizer', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='learning_rate', full_name='object_detection.protos.MomentumOptimizer.learning_rate', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='momentum_optimizer_value', full_name='object_detection.protos.MomentumOptimizer.momentum_optimizer_value', index=1, + number=2, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(0.9), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=542, + serialized_end=662, +) + + +_ADAMOPTIMIZER = _descriptor.Descriptor( + name='AdamOptimizer', + full_name='object_detection.protos.AdamOptimizer', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='learning_rate', full_name='object_detection.protos.AdamOptimizer.learning_rate', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=664, + serialized_end=741, +) + + +_LEARNINGRATE = _descriptor.Descriptor( + name='LearningRate', + full_name='object_detection.protos.LearningRate', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='constant_learning_rate', full_name='object_detection.protos.LearningRate.constant_learning_rate', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='exponential_decay_learning_rate', full_name='object_detection.protos.LearningRate.exponential_decay_learning_rate', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='manual_step_learning_rate', full_name='object_detection.protos.LearningRate.manual_step_learning_rate', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='cosine_decay_learning_rate', full_name='object_detection.protos.LearningRate.cosine_decay_learning_rate', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + _descriptor.OneofDescriptor( + name='learning_rate', full_name='object_detection.protos.LearningRate.learning_rate', + index=0, containing_type=None, fields=[]), + ], + serialized_start=744, + serialized_end=1128, +) + + +_CONSTANTLEARNINGRATE = _descriptor.Descriptor( + name='ConstantLearningRate', + full_name='object_detection.protos.ConstantLearningRate', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='learning_rate', full_name='object_detection.protos.ConstantLearningRate.learning_rate', index=0, + number=1, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(0.002), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=1130, + serialized_end=1182, +) + + +_EXPONENTIALDECAYLEARNINGRATE = _descriptor.Descriptor( + name='ExponentialDecayLearningRate', + full_name='object_detection.protos.ExponentialDecayLearningRate', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='initial_learning_rate', full_name='object_detection.protos.ExponentialDecayLearningRate.initial_learning_rate', index=0, + number=1, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(0.002), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='decay_steps', full_name='object_detection.protos.ExponentialDecayLearningRate.decay_steps', index=1, + number=2, type=13, cpp_type=3, label=1, + has_default_value=True, default_value=4000000, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='decay_factor', full_name='object_detection.protos.ExponentialDecayLearningRate.decay_factor', index=2, + number=3, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(0.95), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='staircase', full_name='object_detection.protos.ExponentialDecayLearningRate.staircase', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='burnin_learning_rate', full_name='object_detection.protos.ExponentialDecayLearningRate.burnin_learning_rate', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='burnin_steps', full_name='object_detection.protos.ExponentialDecayLearningRate.burnin_steps', index=5, + number=6, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='min_learning_rate', full_name='object_detection.protos.ExponentialDecayLearningRate.min_learning_rate', index=6, + number=7, 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=1185, + serialized_end=1424, +) + + +_MANUALSTEPLEARNINGRATE_LEARNINGRATESCHEDULE = _descriptor.Descriptor( + name='LearningRateSchedule', + full_name='object_detection.protos.ManualStepLearningRate.LearningRateSchedule', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='step', full_name='object_detection.protos.ManualStepLearningRate.LearningRateSchedule.step', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='learning_rate', full_name='object_detection.protos.ManualStepLearningRate.LearningRateSchedule.learning_rate', index=1, + number=2, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(0.002), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=1602, + serialized_end=1668, +) + +_MANUALSTEPLEARNINGRATE = _descriptor.Descriptor( + name='ManualStepLearningRate', + full_name='object_detection.protos.ManualStepLearningRate', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='initial_learning_rate', full_name='object_detection.protos.ManualStepLearningRate.initial_learning_rate', index=0, + number=1, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(0.002), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='schedule', full_name='object_detection.protos.ManualStepLearningRate.schedule', index=1, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='warmup', full_name='object_detection.protos.ManualStepLearningRate.warmup', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[_MANUALSTEPLEARNINGRATE_LEARNINGRATESCHEDULE, ], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=1427, + serialized_end=1668, +) + + +_COSINEDECAYLEARNINGRATE = _descriptor.Descriptor( + name='CosineDecayLearningRate', + full_name='object_detection.protos.CosineDecayLearningRate', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='learning_rate_base', full_name='object_detection.protos.CosineDecayLearningRate.learning_rate_base', index=0, + number=1, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(0.002), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='total_steps', full_name='object_detection.protos.CosineDecayLearningRate.total_steps', index=1, + number=2, type=13, cpp_type=3, label=1, + has_default_value=True, default_value=4000000, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='warmup_learning_rate', full_name='object_detection.protos.CosineDecayLearningRate.warmup_learning_rate', index=2, + number=3, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(0.0002), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='warmup_steps', full_name='object_detection.protos.CosineDecayLearningRate.warmup_steps', index=3, + number=4, type=13, cpp_type=3, label=1, + has_default_value=True, default_value=10000, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='hold_base_rate_steps', full_name='object_detection.protos.CosineDecayLearningRate.hold_base_rate_steps', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=1671, + serialized_end=1861, +) + +_OPTIMIZER.fields_by_name['rms_prop_optimizer'].message_type = _RMSPROPOPTIMIZER +_OPTIMIZER.fields_by_name['momentum_optimizer'].message_type = _MOMENTUMOPTIMIZER +_OPTIMIZER.fields_by_name['adam_optimizer'].message_type = _ADAMOPTIMIZER +_OPTIMIZER.oneofs_by_name['optimizer'].fields.append( + _OPTIMIZER.fields_by_name['rms_prop_optimizer']) +_OPTIMIZER.fields_by_name['rms_prop_optimizer'].containing_oneof = _OPTIMIZER.oneofs_by_name['optimizer'] +_OPTIMIZER.oneofs_by_name['optimizer'].fields.append( + _OPTIMIZER.fields_by_name['momentum_optimizer']) +_OPTIMIZER.fields_by_name['momentum_optimizer'].containing_oneof = _OPTIMIZER.oneofs_by_name['optimizer'] +_OPTIMIZER.oneofs_by_name['optimizer'].fields.append( + _OPTIMIZER.fields_by_name['adam_optimizer']) +_OPTIMIZER.fields_by_name['adam_optimizer'].containing_oneof = _OPTIMIZER.oneofs_by_name['optimizer'] +_RMSPROPOPTIMIZER.fields_by_name['learning_rate'].message_type = _LEARNINGRATE +_MOMENTUMOPTIMIZER.fields_by_name['learning_rate'].message_type = _LEARNINGRATE +_ADAMOPTIMIZER.fields_by_name['learning_rate'].message_type = _LEARNINGRATE +_LEARNINGRATE.fields_by_name['constant_learning_rate'].message_type = _CONSTANTLEARNINGRATE +_LEARNINGRATE.fields_by_name['exponential_decay_learning_rate'].message_type = _EXPONENTIALDECAYLEARNINGRATE +_LEARNINGRATE.fields_by_name['manual_step_learning_rate'].message_type = _MANUALSTEPLEARNINGRATE +_LEARNINGRATE.fields_by_name['cosine_decay_learning_rate'].message_type = _COSINEDECAYLEARNINGRATE +_LEARNINGRATE.oneofs_by_name['learning_rate'].fields.append( + _LEARNINGRATE.fields_by_name['constant_learning_rate']) +_LEARNINGRATE.fields_by_name['constant_learning_rate'].containing_oneof = _LEARNINGRATE.oneofs_by_name['learning_rate'] +_LEARNINGRATE.oneofs_by_name['learning_rate'].fields.append( + _LEARNINGRATE.fields_by_name['exponential_decay_learning_rate']) +_LEARNINGRATE.fields_by_name['exponential_decay_learning_rate'].containing_oneof = _LEARNINGRATE.oneofs_by_name['learning_rate'] +_LEARNINGRATE.oneofs_by_name['learning_rate'].fields.append( + _LEARNINGRATE.fields_by_name['manual_step_learning_rate']) +_LEARNINGRATE.fields_by_name['manual_step_learning_rate'].containing_oneof = _LEARNINGRATE.oneofs_by_name['learning_rate'] +_LEARNINGRATE.oneofs_by_name['learning_rate'].fields.append( + _LEARNINGRATE.fields_by_name['cosine_decay_learning_rate']) +_LEARNINGRATE.fields_by_name['cosine_decay_learning_rate'].containing_oneof = _LEARNINGRATE.oneofs_by_name['learning_rate'] +_MANUALSTEPLEARNINGRATE_LEARNINGRATESCHEDULE.containing_type = _MANUALSTEPLEARNINGRATE +_MANUALSTEPLEARNINGRATE.fields_by_name['schedule'].message_type = _MANUALSTEPLEARNINGRATE_LEARNINGRATESCHEDULE +DESCRIPTOR.message_types_by_name['Optimizer'] = _OPTIMIZER +DESCRIPTOR.message_types_by_name['RMSPropOptimizer'] = _RMSPROPOPTIMIZER +DESCRIPTOR.message_types_by_name['MomentumOptimizer'] = _MOMENTUMOPTIMIZER +DESCRIPTOR.message_types_by_name['AdamOptimizer'] = _ADAMOPTIMIZER +DESCRIPTOR.message_types_by_name['LearningRate'] = _LEARNINGRATE +DESCRIPTOR.message_types_by_name['ConstantLearningRate'] = _CONSTANTLEARNINGRATE +DESCRIPTOR.message_types_by_name['ExponentialDecayLearningRate'] = _EXPONENTIALDECAYLEARNINGRATE +DESCRIPTOR.message_types_by_name['ManualStepLearningRate'] = _MANUALSTEPLEARNINGRATE +DESCRIPTOR.message_types_by_name['CosineDecayLearningRate'] = _COSINEDECAYLEARNINGRATE +_sym_db.RegisterFileDescriptor(DESCRIPTOR) + +Optimizer = _reflection.GeneratedProtocolMessageType('Optimizer', (_message.Message,), dict( + DESCRIPTOR = _OPTIMIZER, + __module__ = 'object_detection.protos.optimizer_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.Optimizer) + )) +_sym_db.RegisterMessage(Optimizer) + +RMSPropOptimizer = _reflection.GeneratedProtocolMessageType('RMSPropOptimizer', (_message.Message,), dict( + DESCRIPTOR = _RMSPROPOPTIMIZER, + __module__ = 'object_detection.protos.optimizer_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.RMSPropOptimizer) + )) +_sym_db.RegisterMessage(RMSPropOptimizer) + +MomentumOptimizer = _reflection.GeneratedProtocolMessageType('MomentumOptimizer', (_message.Message,), dict( + DESCRIPTOR = _MOMENTUMOPTIMIZER, + __module__ = 'object_detection.protos.optimizer_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.MomentumOptimizer) + )) +_sym_db.RegisterMessage(MomentumOptimizer) + +AdamOptimizer = _reflection.GeneratedProtocolMessageType('AdamOptimizer', (_message.Message,), dict( + DESCRIPTOR = _ADAMOPTIMIZER, + __module__ = 'object_detection.protos.optimizer_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.AdamOptimizer) + )) +_sym_db.RegisterMessage(AdamOptimizer) + +LearningRate = _reflection.GeneratedProtocolMessageType('LearningRate', (_message.Message,), dict( + DESCRIPTOR = _LEARNINGRATE, + __module__ = 'object_detection.protos.optimizer_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.LearningRate) + )) +_sym_db.RegisterMessage(LearningRate) + +ConstantLearningRate = _reflection.GeneratedProtocolMessageType('ConstantLearningRate', (_message.Message,), dict( + DESCRIPTOR = _CONSTANTLEARNINGRATE, + __module__ = 'object_detection.protos.optimizer_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.ConstantLearningRate) + )) +_sym_db.RegisterMessage(ConstantLearningRate) + +ExponentialDecayLearningRate = _reflection.GeneratedProtocolMessageType('ExponentialDecayLearningRate', (_message.Message,), dict( + DESCRIPTOR = _EXPONENTIALDECAYLEARNINGRATE, + __module__ = 'object_detection.protos.optimizer_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.ExponentialDecayLearningRate) + )) +_sym_db.RegisterMessage(ExponentialDecayLearningRate) + +ManualStepLearningRate = _reflection.GeneratedProtocolMessageType('ManualStepLearningRate', (_message.Message,), dict( + + LearningRateSchedule = _reflection.GeneratedProtocolMessageType('LearningRateSchedule', (_message.Message,), dict( + DESCRIPTOR = _MANUALSTEPLEARNINGRATE_LEARNINGRATESCHEDULE, + __module__ = 'object_detection.protos.optimizer_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.ManualStepLearningRate.LearningRateSchedule) + )) + , + DESCRIPTOR = _MANUALSTEPLEARNINGRATE, + __module__ = 'object_detection.protos.optimizer_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.ManualStepLearningRate) + )) +_sym_db.RegisterMessage(ManualStepLearningRate) +_sym_db.RegisterMessage(ManualStepLearningRate.LearningRateSchedule) + +CosineDecayLearningRate = _reflection.GeneratedProtocolMessageType('CosineDecayLearningRate', (_message.Message,), dict( + DESCRIPTOR = _COSINEDECAYLEARNINGRATE, + __module__ = 'object_detection.protos.optimizer_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.CosineDecayLearningRate) + )) +_sym_db.RegisterMessage(CosineDecayLearningRate) + + +# @@protoc_insertion_point(module_scope) diff --git a/research/object_detection/protos/pipeline.proto b/research/object_detection/object_detection/protos/pipeline.proto similarity index 100% rename from research/object_detection/protos/pipeline.proto rename to research/object_detection/object_detection/protos/pipeline.proto diff --git a/research/object_detection/object_detection/protos/pipeline_pb2.py b/research/object_detection/object_detection/protos/pipeline_pb2.py new file mode 100644 index 00000000000..977148fa0ba --- /dev/null +++ b/research/object_detection/object_detection/protos/pipeline_pb2.py @@ -0,0 +1,116 @@ +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: object_detection/protos/pipeline.proto + +import sys +_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) +from google.protobuf import descriptor as _descriptor +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +from google.protobuf import descriptor_pb2 +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + +from object_detection.protos import eval_pb2 as object__detection_dot_protos_dot_eval__pb2 +from object_detection.protos import graph_rewriter_pb2 as object__detection_dot_protos_dot_graph__rewriter__pb2 +from object_detection.protos import input_reader_pb2 as object__detection_dot_protos_dot_input__reader__pb2 +from object_detection.protos import model_pb2 as object__detection_dot_protos_dot_model__pb2 +from object_detection.protos import train_pb2 as object__detection_dot_protos_dot_train__pb2 + + +DESCRIPTOR = _descriptor.FileDescriptor( + name='object_detection/protos/pipeline.proto', + package='object_detection.protos', + syntax='proto2', + serialized_pb=_b('\n&object_detection/protos/pipeline.proto\x12\x17object_detection.protos\x1a\"object_detection/protos/eval.proto\x1a,object_detection/protos/graph_rewriter.proto\x1a*object_detection/protos/input_reader.proto\x1a#object_detection/protos/model.proto\x1a#object_detection/protos/train.proto\"\x95\x03\n\x17TrainEvalPipelineConfig\x12\x36\n\x05model\x18\x01 \x01(\x0b\x32\'.object_detection.protos.DetectionModel\x12:\n\x0ctrain_config\x18\x02 \x01(\x0b\x32$.object_detection.protos.TrainConfig\x12@\n\x12train_input_reader\x18\x03 \x01(\x0b\x32$.object_detection.protos.InputReader\x12\x38\n\x0b\x65val_config\x18\x04 \x01(\x0b\x32#.object_detection.protos.EvalConfig\x12?\n\x11\x65val_input_reader\x18\x05 \x01(\x0b\x32$.object_detection.protos.InputReader\x12>\n\x0egraph_rewriter\x18\x06 \x01(\x0b\x32&.object_detection.protos.GraphRewriter*\t\x08\xe8\x07\x10\x80\x80\x80\x80\x02') + , + dependencies=[object__detection_dot_protos_dot_eval__pb2.DESCRIPTOR,object__detection_dot_protos_dot_graph__rewriter__pb2.DESCRIPTOR,object__detection_dot_protos_dot_input__reader__pb2.DESCRIPTOR,object__detection_dot_protos_dot_model__pb2.DESCRIPTOR,object__detection_dot_protos_dot_train__pb2.DESCRIPTOR,]) + + + + +_TRAINEVALPIPELINECONFIG = _descriptor.Descriptor( + name='TrainEvalPipelineConfig', + full_name='object_detection.protos.TrainEvalPipelineConfig', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='model', full_name='object_detection.protos.TrainEvalPipelineConfig.model', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='train_config', full_name='object_detection.protos.TrainEvalPipelineConfig.train_config', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='train_input_reader', full_name='object_detection.protos.TrainEvalPipelineConfig.train_input_reader', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='eval_config', full_name='object_detection.protos.TrainEvalPipelineConfig.eval_config', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='eval_input_reader', full_name='object_detection.protos.TrainEvalPipelineConfig.eval_input_reader', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='graph_rewriter', full_name='object_detection.protos.TrainEvalPipelineConfig.graph_rewriter', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=True, + syntax='proto2', + extension_ranges=[(1000, 536870912), ], + oneofs=[ + ], + serialized_start=268, + serialized_end=673, +) + +_TRAINEVALPIPELINECONFIG.fields_by_name['model'].message_type = object__detection_dot_protos_dot_model__pb2._DETECTIONMODEL +_TRAINEVALPIPELINECONFIG.fields_by_name['train_config'].message_type = object__detection_dot_protos_dot_train__pb2._TRAINCONFIG +_TRAINEVALPIPELINECONFIG.fields_by_name['train_input_reader'].message_type = object__detection_dot_protos_dot_input__reader__pb2._INPUTREADER +_TRAINEVALPIPELINECONFIG.fields_by_name['eval_config'].message_type = object__detection_dot_protos_dot_eval__pb2._EVALCONFIG +_TRAINEVALPIPELINECONFIG.fields_by_name['eval_input_reader'].message_type = object__detection_dot_protos_dot_input__reader__pb2._INPUTREADER +_TRAINEVALPIPELINECONFIG.fields_by_name['graph_rewriter'].message_type = object__detection_dot_protos_dot_graph__rewriter__pb2._GRAPHREWRITER +DESCRIPTOR.message_types_by_name['TrainEvalPipelineConfig'] = _TRAINEVALPIPELINECONFIG +_sym_db.RegisterFileDescriptor(DESCRIPTOR) + +TrainEvalPipelineConfig = _reflection.GeneratedProtocolMessageType('TrainEvalPipelineConfig', (_message.Message,), dict( + DESCRIPTOR = _TRAINEVALPIPELINECONFIG, + __module__ = 'object_detection.protos.pipeline_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.TrainEvalPipelineConfig) + )) +_sym_db.RegisterMessage(TrainEvalPipelineConfig) + + +# @@protoc_insertion_point(module_scope) diff --git a/research/object_detection/protos/post_processing.proto b/research/object_detection/object_detection/protos/post_processing.proto similarity index 100% rename from research/object_detection/protos/post_processing.proto rename to research/object_detection/object_detection/protos/post_processing.proto diff --git a/research/object_detection/object_detection/protos/post_processing_pb2.py b/research/object_detection/object_detection/protos/post_processing_pb2.py new file mode 100644 index 00000000000..4001956ad07 --- /dev/null +++ b/research/object_detection/object_detection/protos/post_processing_pb2.py @@ -0,0 +1,173 @@ +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: object_detection/protos/post_processing.proto + +import sys +_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) +from google.protobuf import descriptor as _descriptor +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +from google.protobuf import descriptor_pb2 +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor.FileDescriptor( + name='object_detection/protos/post_processing.proto', + package='object_detection.protos', + syntax='proto2', + serialized_pb=_b('\n-object_detection/protos/post_processing.proto\x12\x17object_detection.protos\"\x9a\x01\n\x16\x42\x61tchNonMaxSuppression\x12\x1a\n\x0fscore_threshold\x18\x01 \x01(\x02:\x01\x30\x12\x1a\n\riou_threshold\x18\x02 \x01(\x02:\x03\x30.6\x12%\n\x18max_detections_per_class\x18\x03 \x01(\x05:\x03\x31\x30\x30\x12!\n\x14max_total_detections\x18\x05 \x01(\x05:\x03\x31\x30\x30\"\x91\x02\n\x0ePostProcessing\x12R\n\x19\x62\x61tch_non_max_suppression\x18\x01 \x01(\x0b\x32/.object_detection.protos.BatchNonMaxSuppression\x12Y\n\x0fscore_converter\x18\x02 \x01(\x0e\x32\x36.object_detection.protos.PostProcessing.ScoreConverter:\x08IDENTITY\x12\x16\n\x0blogit_scale\x18\x03 \x01(\x02:\x01\x31\"8\n\x0eScoreConverter\x12\x0c\n\x08IDENTITY\x10\x00\x12\x0b\n\x07SIGMOID\x10\x01\x12\x0b\n\x07SOFTMAX\x10\x02') +) + + + +_POSTPROCESSING_SCORECONVERTER = _descriptor.EnumDescriptor( + name='ScoreConverter', + full_name='object_detection.protos.PostProcessing.ScoreConverter', + filename=None, + file=DESCRIPTOR, + values=[ + _descriptor.EnumValueDescriptor( + name='IDENTITY', index=0, number=0, + options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='SIGMOID', index=1, number=1, + options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='SOFTMAX', index=2, number=2, + options=None, + type=None), + ], + containing_type=None, + options=None, + serialized_start=449, + serialized_end=505, +) +_sym_db.RegisterEnumDescriptor(_POSTPROCESSING_SCORECONVERTER) + + +_BATCHNONMAXSUPPRESSION = _descriptor.Descriptor( + name='BatchNonMaxSuppression', + full_name='object_detection.protos.BatchNonMaxSuppression', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='score_threshold', full_name='object_detection.protos.BatchNonMaxSuppression.score_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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='iou_threshold', full_name='object_detection.protos.BatchNonMaxSuppression.iou_threshold', index=1, + number=2, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(0.6), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_detections_per_class', full_name='object_detection.protos.BatchNonMaxSuppression.max_detections_per_class', index=2, + number=3, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=100, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_total_detections', full_name='object_detection.protos.BatchNonMaxSuppression.max_total_detections', index=3, + number=5, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=100, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=75, + serialized_end=229, +) + + +_POSTPROCESSING = _descriptor.Descriptor( + name='PostProcessing', + full_name='object_detection.protos.PostProcessing', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='batch_non_max_suppression', full_name='object_detection.protos.PostProcessing.batch_non_max_suppression', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='score_converter', full_name='object_detection.protos.PostProcessing.score_converter', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='logit_scale', full_name='object_detection.protos.PostProcessing.logit_scale', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + _POSTPROCESSING_SCORECONVERTER, + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=232, + serialized_end=505, +) + +_POSTPROCESSING.fields_by_name['batch_non_max_suppression'].message_type = _BATCHNONMAXSUPPRESSION +_POSTPROCESSING.fields_by_name['score_converter'].enum_type = _POSTPROCESSING_SCORECONVERTER +_POSTPROCESSING_SCORECONVERTER.containing_type = _POSTPROCESSING +DESCRIPTOR.message_types_by_name['BatchNonMaxSuppression'] = _BATCHNONMAXSUPPRESSION +DESCRIPTOR.message_types_by_name['PostProcessing'] = _POSTPROCESSING +_sym_db.RegisterFileDescriptor(DESCRIPTOR) + +BatchNonMaxSuppression = _reflection.GeneratedProtocolMessageType('BatchNonMaxSuppression', (_message.Message,), dict( + DESCRIPTOR = _BATCHNONMAXSUPPRESSION, + __module__ = 'object_detection.protos.post_processing_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.BatchNonMaxSuppression) + )) +_sym_db.RegisterMessage(BatchNonMaxSuppression) + +PostProcessing = _reflection.GeneratedProtocolMessageType('PostProcessing', (_message.Message,), dict( + DESCRIPTOR = _POSTPROCESSING, + __module__ = 'object_detection.protos.post_processing_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.PostProcessing) + )) +_sym_db.RegisterMessage(PostProcessing) + + +# @@protoc_insertion_point(module_scope) diff --git a/research/object_detection/protos/preprocessor.proto b/research/object_detection/object_detection/protos/preprocessor.proto similarity index 100% rename from research/object_detection/protos/preprocessor.proto rename to research/object_detection/object_detection/protos/preprocessor.proto diff --git a/research/object_detection/object_detection/protos/preprocessor_pb2.py b/research/object_detection/object_detection/protos/preprocessor_pb2.py new file mode 100644 index 00000000000..d4cebb64750 --- /dev/null +++ b/research/object_detection/object_detection/protos/preprocessor_pb2.py @@ -0,0 +1,2028 @@ +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: object_detection/protos/preprocessor.proto + +import sys +_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) +from google.protobuf import descriptor as _descriptor +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +from google.protobuf import descriptor_pb2 +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor.FileDescriptor( + name='object_detection/protos/preprocessor.proto', + package='object_detection.protos', + syntax='proto2', + serialized_pb=_b('\n*object_detection/protos/preprocessor.proto\x12\x17object_detection.protos\"\xea\x10\n\x11PreprocessingStep\x12\x42\n\x0fnormalize_image\x18\x01 \x01(\x0b\x32\'.object_detection.protos.NormalizeImageH\x00\x12O\n\x16random_horizontal_flip\x18\x02 \x01(\x0b\x32-.object_detection.protos.RandomHorizontalFlipH\x00\x12R\n\x18random_pixel_value_scale\x18\x03 \x01(\x0b\x32..object_detection.protos.RandomPixelValueScaleH\x00\x12G\n\x12random_image_scale\x18\x04 \x01(\x0b\x32).object_detection.protos.RandomImageScaleH\x00\x12\x46\n\x12random_rgb_to_gray\x18\x05 \x01(\x0b\x32(.object_detection.protos.RandomRGBtoGrayH\x00\x12S\n\x18random_adjust_brightness\x18\x06 \x01(\x0b\x32/.object_detection.protos.RandomAdjustBrightnessH\x00\x12O\n\x16random_adjust_contrast\x18\x07 \x01(\x0b\x32-.object_detection.protos.RandomAdjustContrastH\x00\x12\x45\n\x11random_adjust_hue\x18\x08 \x01(\x0b\x32(.object_detection.protos.RandomAdjustHueH\x00\x12S\n\x18random_adjust_saturation\x18\t \x01(\x0b\x32/.object_detection.protos.RandomAdjustSaturationH\x00\x12K\n\x14random_distort_color\x18\n \x01(\x0b\x32+.object_detection.protos.RandomDistortColorH\x00\x12I\n\x13random_jitter_boxes\x18\x0b \x01(\x0b\x32*.object_detection.protos.RandomJitterBoxesH\x00\x12\x45\n\x11random_crop_image\x18\x0c \x01(\x0b\x32(.object_detection.protos.RandomCropImageH\x00\x12\x43\n\x10random_pad_image\x18\r \x01(\x0b\x32\'.object_detection.protos.RandomPadImageH\x00\x12L\n\x15random_crop_pad_image\x18\x0e \x01(\x0b\x32+.object_detection.protos.RandomCropPadImageH\x00\x12W\n\x1brandom_crop_to_aspect_ratio\x18\x0f \x01(\x0b\x32\x30.object_detection.protos.RandomCropToAspectRatioH\x00\x12K\n\x14random_black_patches\x18\x10 \x01(\x0b\x32+.object_detection.protos.RandomBlackPatchesH\x00\x12K\n\x14random_resize_method\x18\x11 \x01(\x0b\x32+.object_detection.protos.RandomResizeMethodH\x00\x12\x61\n scale_boxes_to_pixel_coordinates\x18\x12 \x01(\x0b\x32\x35.object_detection.protos.ScaleBoxesToPixelCoordinatesH\x00\x12<\n\x0cresize_image\x18\x13 \x01(\x0b\x32$.object_detection.protos.ResizeImageH\x00\x12M\n\x15subtract_channel_mean\x18\x14 \x01(\x0b\x32,.object_detection.protos.SubtractChannelMeanH\x00\x12\x41\n\x0fssd_random_crop\x18\x15 \x01(\x0b\x32&.object_detection.protos.SSDRandomCropH\x00\x12H\n\x13ssd_random_crop_pad\x18\x16 \x01(\x0b\x32).object_detection.protos.SSDRandomCropPadH\x00\x12\x64\n\"ssd_random_crop_fixed_aspect_ratio\x18\x17 \x01(\x0b\x32\x36.object_detection.protos.SSDRandomCropFixedAspectRatioH\x00\x12k\n&ssd_random_crop_pad_fixed_aspect_ratio\x18\x18 \x01(\x0b\x32\x39.object_detection.protos.SSDRandomCropPadFixedAspectRatioH\x00\x12K\n\x14random_vertical_flip\x18\x19 \x01(\x0b\x32+.object_detection.protos.RandomVerticalFlipH\x00\x12\x46\n\x11random_rotation90\x18\x1a \x01(\x0b\x32).object_detection.protos.RandomRotation90H\x00\x12\x39\n\x0brgb_to_gray\x18\x1b \x01(\x0b\x32\".object_detection.protos.RGBtoGrayH\x00\x42\x14\n\x12preprocessing_step\"v\n\x0eNormalizeImage\x12\x17\n\x0foriginal_minval\x18\x01 \x01(\x02\x12\x17\n\x0foriginal_maxval\x18\x02 \x01(\x02\x12\x18\n\rtarget_minval\x18\x03 \x01(\x02:\x01\x30\x12\x18\n\rtarget_maxval\x18\x04 \x01(\x02:\x01\x31\"9\n\x14RandomHorizontalFlip\x12!\n\x19keypoint_flip_permutation\x18\x01 \x03(\x05\"7\n\x12RandomVerticalFlip\x12!\n\x19keypoint_flip_permutation\x18\x01 \x03(\x05\"\x12\n\x10RandomRotation90\"A\n\x15RandomPixelValueScale\x12\x13\n\x06minval\x18\x01 \x01(\x02:\x03\x30.9\x12\x13\n\x06maxval\x18\x02 \x01(\x02:\x03\x31.1\"L\n\x10RandomImageScale\x12\x1c\n\x0fmin_scale_ratio\x18\x01 \x01(\x02:\x03\x30.5\x12\x1a\n\x0fmax_scale_ratio\x18\x02 \x01(\x02:\x01\x32\"+\n\x0fRandomRGBtoGray\x12\x18\n\x0bprobability\x18\x01 \x01(\x02:\x03\x30.1\"0\n\x16RandomAdjustBrightness\x12\x16\n\tmax_delta\x18\x01 \x01(\x02:\x03\x30.2\"G\n\x14RandomAdjustContrast\x12\x16\n\tmin_delta\x18\x01 \x01(\x02:\x03\x30.8\x12\x17\n\tmax_delta\x18\x02 \x01(\x02:\x04\x31.25\"*\n\x0fRandomAdjustHue\x12\x17\n\tmax_delta\x18\x01 \x01(\x02:\x04\x30.02\"I\n\x16RandomAdjustSaturation\x12\x16\n\tmin_delta\x18\x01 \x01(\x02:\x03\x30.8\x12\x17\n\tmax_delta\x18\x02 \x01(\x02:\x04\x31.25\",\n\x12RandomDistortColor\x12\x16\n\x0e\x63olor_ordering\x18\x01 \x01(\x05\"(\n\x11RandomJitterBoxes\x12\x13\n\x05ratio\x18\x01 \x01(\x02:\x04\x30.05\"\xd1\x01\n\x0fRandomCropImage\x12\x1d\n\x12min_object_covered\x18\x01 \x01(\x02:\x01\x31\x12\x1e\n\x10min_aspect_ratio\x18\x02 \x01(\x02:\x04\x30.75\x12\x1e\n\x10max_aspect_ratio\x18\x03 \x01(\x02:\x04\x31.33\x12\x15\n\x08min_area\x18\x04 \x01(\x02:\x03\x30.1\x12\x13\n\x08max_area\x18\x05 \x01(\x02:\x01\x31\x12\x1b\n\x0eoverlap_thresh\x18\x06 \x01(\x02:\x03\x30.3\x12\x16\n\x0brandom_coef\x18\x07 \x01(\x02:\x01\x30\"\x89\x01\n\x0eRandomPadImage\x12\x18\n\x10min_image_height\x18\x01 \x01(\x02\x12\x17\n\x0fmin_image_width\x18\x02 \x01(\x02\x12\x18\n\x10max_image_height\x18\x03 \x01(\x02\x12\x17\n\x0fmax_image_width\x18\x04 \x01(\x02\x12\x11\n\tpad_color\x18\x05 \x03(\x02\"\xa5\x02\n\x12RandomCropPadImage\x12\x1d\n\x12min_object_covered\x18\x01 \x01(\x02:\x01\x31\x12\x1e\n\x10min_aspect_ratio\x18\x02 \x01(\x02:\x04\x30.75\x12\x1e\n\x10max_aspect_ratio\x18\x03 \x01(\x02:\x04\x31.33\x12\x15\n\x08min_area\x18\x04 \x01(\x02:\x03\x30.1\x12\x13\n\x08max_area\x18\x05 \x01(\x02:\x01\x31\x12\x1b\n\x0eoverlap_thresh\x18\x06 \x01(\x02:\x03\x30.3\x12\x16\n\x0brandom_coef\x18\x07 \x01(\x02:\x01\x30\x12\x1d\n\x15min_padded_size_ratio\x18\x08 \x03(\x02\x12\x1d\n\x15max_padded_size_ratio\x18\t \x03(\x02\x12\x11\n\tpad_color\x18\n \x03(\x02\"O\n\x17RandomCropToAspectRatio\x12\x17\n\x0c\x61spect_ratio\x18\x01 \x01(\x02:\x01\x31\x12\x1b\n\x0eoverlap_thresh\x18\x02 \x01(\x02:\x03\x30.3\"o\n\x12RandomBlackPatches\x12\x1d\n\x11max_black_patches\x18\x01 \x01(\x05:\x02\x31\x30\x12\x18\n\x0bprobability\x18\x02 \x01(\x02:\x03\x30.5\x12 \n\x13size_to_image_ratio\x18\x03 \x01(\x02:\x03\x30.1\"A\n\x12RandomResizeMethod\x12\x15\n\rtarget_height\x18\x01 \x01(\x02\x12\x14\n\x0ctarget_width\x18\x02 \x01(\x02\"\x0b\n\tRGBtoGray\"\x1e\n\x1cScaleBoxesToPixelCoordinates\"\xc0\x01\n\x0bResizeImage\x12\x12\n\nnew_height\x18\x01 \x01(\x05\x12\x11\n\tnew_width\x18\x02 \x01(\x05\x12\x45\n\x06method\x18\x03 \x01(\x0e\x32+.object_detection.protos.ResizeImage.Method:\x08\x42ILINEAR\"C\n\x06Method\x12\x08\n\x04\x41REA\x10\x01\x12\x0b\n\x07\x42ICUBIC\x10\x02\x12\x0c\n\x08\x42ILINEAR\x10\x03\x12\x14\n\x10NEAREST_NEIGHBOR\x10\x04\"$\n\x13SubtractChannelMean\x12\r\n\x05means\x18\x01 \x03(\x02\"\xb9\x01\n\x16SSDRandomCropOperation\x12\x1a\n\x12min_object_covered\x18\x01 \x01(\x02\x12\x18\n\x10min_aspect_ratio\x18\x02 \x01(\x02\x12\x18\n\x10max_aspect_ratio\x18\x03 \x01(\x02\x12\x10\n\x08min_area\x18\x04 \x01(\x02\x12\x10\n\x08max_area\x18\x05 \x01(\x02\x12\x16\n\x0eoverlap_thresh\x18\x06 \x01(\x02\x12\x13\n\x0brandom_coef\x18\x07 \x01(\x02\"T\n\rSSDRandomCrop\x12\x43\n\noperations\x18\x01 \x03(\x0b\x32/.object_detection.protos.SSDRandomCropOperation\"\xb9\x02\n\x19SSDRandomCropPadOperation\x12\x1a\n\x12min_object_covered\x18\x01 \x01(\x02\x12\x18\n\x10min_aspect_ratio\x18\x02 \x01(\x02\x12\x18\n\x10max_aspect_ratio\x18\x03 \x01(\x02\x12\x10\n\x08min_area\x18\x04 \x01(\x02\x12\x10\n\x08max_area\x18\x05 \x01(\x02\x12\x16\n\x0eoverlap_thresh\x18\x06 \x01(\x02\x12\x13\n\x0brandom_coef\x18\x07 \x01(\x02\x12\x1d\n\x15min_padded_size_ratio\x18\x08 \x03(\x02\x12\x1d\n\x15max_padded_size_ratio\x18\t \x03(\x02\x12\x13\n\x0bpad_color_r\x18\n \x01(\x02\x12\x13\n\x0bpad_color_g\x18\x0b \x01(\x02\x12\x13\n\x0bpad_color_b\x18\x0c \x01(\x02\"Z\n\x10SSDRandomCropPad\x12\x46\n\noperations\x18\x01 \x03(\x0b\x32\x32.object_detection.protos.SSDRandomCropPadOperation\"\x95\x01\n&SSDRandomCropFixedAspectRatioOperation\x12\x1a\n\x12min_object_covered\x18\x01 \x01(\x02\x12\x10\n\x08min_area\x18\x04 \x01(\x02\x12\x10\n\x08max_area\x18\x05 \x01(\x02\x12\x16\n\x0eoverlap_thresh\x18\x06 \x01(\x02\x12\x13\n\x0brandom_coef\x18\x07 \x01(\x02\"\x8d\x01\n\x1dSSDRandomCropFixedAspectRatio\x12S\n\noperations\x18\x01 \x03(\x0b\x32?.object_detection.protos.SSDRandomCropFixedAspectRatioOperation\x12\x17\n\x0c\x61spect_ratio\x18\x02 \x01(\x02:\x01\x31\"\xcc\x01\n)SSDRandomCropPadFixedAspectRatioOperation\x12\x1a\n\x12min_object_covered\x18\x01 \x01(\x02\x12\x18\n\x10min_aspect_ratio\x18\x02 \x01(\x02\x12\x18\n\x10max_aspect_ratio\x18\x03 \x01(\x02\x12\x10\n\x08min_area\x18\x04 \x01(\x02\x12\x10\n\x08max_area\x18\x05 \x01(\x02\x12\x16\n\x0eoverlap_thresh\x18\x06 \x01(\x02\x12\x13\n\x0brandom_coef\x18\x07 \x01(\x02\"\xd1\x01\n SSDRandomCropPadFixedAspectRatio\x12V\n\noperations\x18\x01 \x03(\x0b\x32\x42.object_detection.protos.SSDRandomCropPadFixedAspectRatioOperation\x12\x17\n\x0c\x61spect_ratio\x18\x02 \x01(\x02:\x01\x31\x12\x1d\n\x15min_padded_size_ratio\x18\x03 \x03(\x02\x12\x1d\n\x15max_padded_size_ratio\x18\x04 \x03(\x02') +) + + + +_RESIZEIMAGE_METHOD = _descriptor.EnumDescriptor( + name='Method', + full_name='object_detection.protos.ResizeImage.Method', + filename=None, + file=DESCRIPTOR, + values=[ + _descriptor.EnumValueDescriptor( + name='AREA', index=0, number=1, + options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='BICUBIC', index=1, number=2, + options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='BILINEAR', index=2, number=3, + options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='NEAREST_NEIGHBOR', index=3, number=4, + options=None, + type=None), + ], + containing_type=None, + options=None, + serialized_start=4084, + serialized_end=4151, +) +_sym_db.RegisterEnumDescriptor(_RESIZEIMAGE_METHOD) + + +_PREPROCESSINGSTEP = _descriptor.Descriptor( + name='PreprocessingStep', + full_name='object_detection.protos.PreprocessingStep', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='normalize_image', full_name='object_detection.protos.PreprocessingStep.normalize_image', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='random_horizontal_flip', full_name='object_detection.protos.PreprocessingStep.random_horizontal_flip', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='random_pixel_value_scale', full_name='object_detection.protos.PreprocessingStep.random_pixel_value_scale', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='random_image_scale', full_name='object_detection.protos.PreprocessingStep.random_image_scale', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='random_rgb_to_gray', full_name='object_detection.protos.PreprocessingStep.random_rgb_to_gray', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='random_adjust_brightness', full_name='object_detection.protos.PreprocessingStep.random_adjust_brightness', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='random_adjust_contrast', full_name='object_detection.protos.PreprocessingStep.random_adjust_contrast', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='random_adjust_hue', full_name='object_detection.protos.PreprocessingStep.random_adjust_hue', index=7, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='random_adjust_saturation', full_name='object_detection.protos.PreprocessingStep.random_adjust_saturation', index=8, + number=9, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='random_distort_color', full_name='object_detection.protos.PreprocessingStep.random_distort_color', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='random_jitter_boxes', full_name='object_detection.protos.PreprocessingStep.random_jitter_boxes', index=10, + number=11, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='random_crop_image', full_name='object_detection.protos.PreprocessingStep.random_crop_image', index=11, + number=12, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='random_pad_image', full_name='object_detection.protos.PreprocessingStep.random_pad_image', index=12, + number=13, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='random_crop_pad_image', full_name='object_detection.protos.PreprocessingStep.random_crop_pad_image', index=13, + number=14, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='random_crop_to_aspect_ratio', full_name='object_detection.protos.PreprocessingStep.random_crop_to_aspect_ratio', index=14, + number=15, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='random_black_patches', full_name='object_detection.protos.PreprocessingStep.random_black_patches', index=15, + number=16, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='random_resize_method', full_name='object_detection.protos.PreprocessingStep.random_resize_method', index=16, + number=17, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='scale_boxes_to_pixel_coordinates', full_name='object_detection.protos.PreprocessingStep.scale_boxes_to_pixel_coordinates', index=17, + number=18, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='resize_image', full_name='object_detection.protos.PreprocessingStep.resize_image', index=18, + number=19, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='subtract_channel_mean', full_name='object_detection.protos.PreprocessingStep.subtract_channel_mean', index=19, + number=20, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='ssd_random_crop', full_name='object_detection.protos.PreprocessingStep.ssd_random_crop', index=20, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='ssd_random_crop_pad', full_name='object_detection.protos.PreprocessingStep.ssd_random_crop_pad', index=21, + number=22, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='ssd_random_crop_fixed_aspect_ratio', full_name='object_detection.protos.PreprocessingStep.ssd_random_crop_fixed_aspect_ratio', index=22, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='ssd_random_crop_pad_fixed_aspect_ratio', full_name='object_detection.protos.PreprocessingStep.ssd_random_crop_pad_fixed_aspect_ratio', index=23, + number=24, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='random_vertical_flip', full_name='object_detection.protos.PreprocessingStep.random_vertical_flip', index=24, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='random_rotation90', full_name='object_detection.protos.PreprocessingStep.random_rotation90', index=25, + number=26, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='rgb_to_gray', full_name='object_detection.protos.PreprocessingStep.rgb_to_gray', index=26, + 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + _descriptor.OneofDescriptor( + name='preprocessing_step', full_name='object_detection.protos.PreprocessingStep.preprocessing_step', + index=0, containing_type=None, fields=[]), + ], + serialized_start=72, + serialized_end=2226, +) + + +_NORMALIZEIMAGE = _descriptor.Descriptor( + name='NormalizeImage', + full_name='object_detection.protos.NormalizeImage', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='original_minval', full_name='object_detection.protos.NormalizeImage.original_minval', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='original_maxval', full_name='object_detection.protos.NormalizeImage.original_maxval', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='target_minval', full_name='object_detection.protos.NormalizeImage.target_minval', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='target_maxval', full_name='object_detection.protos.NormalizeImage.target_maxval', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=2228, + serialized_end=2346, +) + + +_RANDOMHORIZONTALFLIP = _descriptor.Descriptor( + name='RandomHorizontalFlip', + full_name='object_detection.protos.RandomHorizontalFlip', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='keypoint_flip_permutation', full_name='object_detection.protos.RandomHorizontalFlip.keypoint_flip_permutation', index=0, + number=1, 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=2348, + serialized_end=2405, +) + + +_RANDOMVERTICALFLIP = _descriptor.Descriptor( + name='RandomVerticalFlip', + full_name='object_detection.protos.RandomVerticalFlip', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='keypoint_flip_permutation', full_name='object_detection.protos.RandomVerticalFlip.keypoint_flip_permutation', index=0, + number=1, 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=2407, + serialized_end=2462, +) + + +_RANDOMROTATION90 = _descriptor.Descriptor( + name='RandomRotation90', + full_name='object_detection.protos.RandomRotation90', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=2464, + serialized_end=2482, +) + + +_RANDOMPIXELVALUESCALE = _descriptor.Descriptor( + name='RandomPixelValueScale', + full_name='object_detection.protos.RandomPixelValueScale', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='minval', full_name='object_detection.protos.RandomPixelValueScale.minval', index=0, + number=1, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(0.9), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='maxval', full_name='object_detection.protos.RandomPixelValueScale.maxval', index=1, + number=2, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(1.1), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=2484, + serialized_end=2549, +) + + +_RANDOMIMAGESCALE = _descriptor.Descriptor( + name='RandomImageScale', + full_name='object_detection.protos.RandomImageScale', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='min_scale_ratio', full_name='object_detection.protos.RandomImageScale.min_scale_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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_scale_ratio', full_name='object_detection.protos.RandomImageScale.max_scale_ratio', index=1, + number=2, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(2), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=2551, + serialized_end=2627, +) + + +_RANDOMRGBTOGRAY = _descriptor.Descriptor( + name='RandomRGBtoGray', + full_name='object_detection.protos.RandomRGBtoGray', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='probability', full_name='object_detection.protos.RandomRGBtoGray.probability', index=0, + number=1, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(0.1), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=2629, + serialized_end=2672, +) + + +_RANDOMADJUSTBRIGHTNESS = _descriptor.Descriptor( + name='RandomAdjustBrightness', + full_name='object_detection.protos.RandomAdjustBrightness', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='max_delta', full_name='object_detection.protos.RandomAdjustBrightness.max_delta', index=0, + number=1, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(0.2), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=2674, + serialized_end=2722, +) + + +_RANDOMADJUSTCONTRAST = _descriptor.Descriptor( + name='RandomAdjustContrast', + full_name='object_detection.protos.RandomAdjustContrast', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='min_delta', full_name='object_detection.protos.RandomAdjustContrast.min_delta', index=0, + number=1, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(0.8), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_delta', full_name='object_detection.protos.RandomAdjustContrast.max_delta', index=1, + number=2, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(1.25), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=2724, + serialized_end=2795, +) + + +_RANDOMADJUSTHUE = _descriptor.Descriptor( + name='RandomAdjustHue', + full_name='object_detection.protos.RandomAdjustHue', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='max_delta', full_name='object_detection.protos.RandomAdjustHue.max_delta', index=0, + number=1, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(0.02), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=2797, + serialized_end=2839, +) + + +_RANDOMADJUSTSATURATION = _descriptor.Descriptor( + name='RandomAdjustSaturation', + full_name='object_detection.protos.RandomAdjustSaturation', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='min_delta', full_name='object_detection.protos.RandomAdjustSaturation.min_delta', index=0, + number=1, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(0.8), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_delta', full_name='object_detection.protos.RandomAdjustSaturation.max_delta', index=1, + number=2, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(1.25), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=2841, + serialized_end=2914, +) + + +_RANDOMDISTORTCOLOR = _descriptor.Descriptor( + name='RandomDistortColor', + full_name='object_detection.protos.RandomDistortColor', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='color_ordering', full_name='object_detection.protos.RandomDistortColor.color_ordering', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=2916, + serialized_end=2960, +) + + +_RANDOMJITTERBOXES = _descriptor.Descriptor( + name='RandomJitterBoxes', + full_name='object_detection.protos.RandomJitterBoxes', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='ratio', full_name='object_detection.protos.RandomJitterBoxes.ratio', index=0, + number=1, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(0.05), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=2962, + serialized_end=3002, +) + + +_RANDOMCROPIMAGE = _descriptor.Descriptor( + name='RandomCropImage', + full_name='object_detection.protos.RandomCropImage', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='min_object_covered', full_name='object_detection.protos.RandomCropImage.min_object_covered', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='min_aspect_ratio', full_name='object_detection.protos.RandomCropImage.min_aspect_ratio', index=1, + number=2, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_aspect_ratio', full_name='object_detection.protos.RandomCropImage.max_aspect_ratio', index=2, + number=3, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(1.33), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='min_area', full_name='object_detection.protos.RandomCropImage.min_area', index=3, + number=4, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(0.1), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_area', full_name='object_detection.protos.RandomCropImage.max_area', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='overlap_thresh', full_name='object_detection.protos.RandomCropImage.overlap_thresh', index=5, + number=6, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='random_coef', full_name='object_detection.protos.RandomCropImage.random_coef', index=6, + number=7, 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=3005, + serialized_end=3214, +) + + +_RANDOMPADIMAGE = _descriptor.Descriptor( + name='RandomPadImage', + full_name='object_detection.protos.RandomPadImage', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='min_image_height', full_name='object_detection.protos.RandomPadImage.min_image_height', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='min_image_width', full_name='object_detection.protos.RandomPadImage.min_image_width', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_image_height', full_name='object_detection.protos.RandomPadImage.max_image_height', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_image_width', full_name='object_detection.protos.RandomPadImage.max_image_width', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='pad_color', full_name='object_detection.protos.RandomPadImage.pad_color', index=4, + 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=3217, + serialized_end=3354, +) + + +_RANDOMCROPPADIMAGE = _descriptor.Descriptor( + name='RandomCropPadImage', + full_name='object_detection.protos.RandomCropPadImage', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='min_object_covered', full_name='object_detection.protos.RandomCropPadImage.min_object_covered', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='min_aspect_ratio', full_name='object_detection.protos.RandomCropPadImage.min_aspect_ratio', index=1, + number=2, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_aspect_ratio', full_name='object_detection.protos.RandomCropPadImage.max_aspect_ratio', index=2, + number=3, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(1.33), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='min_area', full_name='object_detection.protos.RandomCropPadImage.min_area', index=3, + number=4, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(0.1), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_area', full_name='object_detection.protos.RandomCropPadImage.max_area', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='overlap_thresh', full_name='object_detection.protos.RandomCropPadImage.overlap_thresh', index=5, + number=6, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='random_coef', full_name='object_detection.protos.RandomCropPadImage.random_coef', index=6, + number=7, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='min_padded_size_ratio', full_name='object_detection.protos.RandomCropPadImage.min_padded_size_ratio', index=7, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_padded_size_ratio', full_name='object_detection.protos.RandomCropPadImage.max_padded_size_ratio', index=8, + number=9, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='pad_color', full_name='object_detection.protos.RandomCropPadImage.pad_color', index=9, + number=10, 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=3357, + serialized_end=3650, +) + + +_RANDOMCROPTOASPECTRATIO = _descriptor.Descriptor( + name='RandomCropToAspectRatio', + full_name='object_detection.protos.RandomCropToAspectRatio', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='aspect_ratio', full_name='object_detection.protos.RandomCropToAspectRatio.aspect_ratio', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='overlap_thresh', full_name='object_detection.protos.RandomCropToAspectRatio.overlap_thresh', index=1, + number=2, 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=3652, + serialized_end=3731, +) + + +_RANDOMBLACKPATCHES = _descriptor.Descriptor( + name='RandomBlackPatches', + full_name='object_detection.protos.RandomBlackPatches', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='max_black_patches', full_name='object_detection.protos.RandomBlackPatches.max_black_patches', index=0, + number=1, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=10, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='probability', full_name='object_detection.protos.RandomBlackPatches.probability', index=1, + number=2, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='size_to_image_ratio', full_name='object_detection.protos.RandomBlackPatches.size_to_image_ratio', index=2, + number=3, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(0.1), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=3733, + serialized_end=3844, +) + + +_RANDOMRESIZEMETHOD = _descriptor.Descriptor( + name='RandomResizeMethod', + full_name='object_detection.protos.RandomResizeMethod', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='target_height', full_name='object_detection.protos.RandomResizeMethod.target_height', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='target_width', full_name='object_detection.protos.RandomResizeMethod.target_width', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=3846, + serialized_end=3911, +) + + +_RGBTOGRAY = _descriptor.Descriptor( + name='RGBtoGray', + full_name='object_detection.protos.RGBtoGray', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=3913, + serialized_end=3924, +) + + +_SCALEBOXESTOPIXELCOORDINATES = _descriptor.Descriptor( + name='ScaleBoxesToPixelCoordinates', + full_name='object_detection.protos.ScaleBoxesToPixelCoordinates', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=3926, + serialized_end=3956, +) + + +_RESIZEIMAGE = _descriptor.Descriptor( + name='ResizeImage', + full_name='object_detection.protos.ResizeImage', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='new_height', full_name='object_detection.protos.ResizeImage.new_height', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='new_width', full_name='object_detection.protos.ResizeImage.new_width', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='method', full_name='object_detection.protos.ResizeImage.method', index=2, + number=3, type=14, cpp_type=8, label=1, + has_default_value=True, default_value=3, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + _RESIZEIMAGE_METHOD, + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=3959, + serialized_end=4151, +) + + +_SUBTRACTCHANNELMEAN = _descriptor.Descriptor( + name='SubtractChannelMean', + full_name='object_detection.protos.SubtractChannelMean', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='means', full_name='object_detection.protos.SubtractChannelMean.means', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=4153, + serialized_end=4189, +) + + +_SSDRANDOMCROPOPERATION = _descriptor.Descriptor( + name='SSDRandomCropOperation', + full_name='object_detection.protos.SSDRandomCropOperation', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='min_object_covered', full_name='object_detection.protos.SSDRandomCropOperation.min_object_covered', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='min_aspect_ratio', full_name='object_detection.protos.SSDRandomCropOperation.min_aspect_ratio', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_aspect_ratio', full_name='object_detection.protos.SSDRandomCropOperation.max_aspect_ratio', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='min_area', full_name='object_detection.protos.SSDRandomCropOperation.min_area', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_area', full_name='object_detection.protos.SSDRandomCropOperation.max_area', index=4, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='overlap_thresh', full_name='object_detection.protos.SSDRandomCropOperation.overlap_thresh', index=5, + number=6, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='random_coef', full_name='object_detection.protos.SSDRandomCropOperation.random_coef', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=4192, + serialized_end=4377, +) + + +_SSDRANDOMCROP = _descriptor.Descriptor( + name='SSDRandomCrop', + full_name='object_detection.protos.SSDRandomCrop', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='operations', full_name='object_detection.protos.SSDRandomCrop.operations', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=4379, + serialized_end=4463, +) + + +_SSDRANDOMCROPPADOPERATION = _descriptor.Descriptor( + name='SSDRandomCropPadOperation', + full_name='object_detection.protos.SSDRandomCropPadOperation', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='min_object_covered', full_name='object_detection.protos.SSDRandomCropPadOperation.min_object_covered', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='min_aspect_ratio', full_name='object_detection.protos.SSDRandomCropPadOperation.min_aspect_ratio', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_aspect_ratio', full_name='object_detection.protos.SSDRandomCropPadOperation.max_aspect_ratio', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='min_area', full_name='object_detection.protos.SSDRandomCropPadOperation.min_area', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_area', full_name='object_detection.protos.SSDRandomCropPadOperation.max_area', index=4, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='overlap_thresh', full_name='object_detection.protos.SSDRandomCropPadOperation.overlap_thresh', index=5, + number=6, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='random_coef', full_name='object_detection.protos.SSDRandomCropPadOperation.random_coef', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='min_padded_size_ratio', full_name='object_detection.protos.SSDRandomCropPadOperation.min_padded_size_ratio', index=7, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_padded_size_ratio', full_name='object_detection.protos.SSDRandomCropPadOperation.max_padded_size_ratio', index=8, + number=9, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='pad_color_r', full_name='object_detection.protos.SSDRandomCropPadOperation.pad_color_r', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='pad_color_g', full_name='object_detection.protos.SSDRandomCropPadOperation.pad_color_g', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='pad_color_b', full_name='object_detection.protos.SSDRandomCropPadOperation.pad_color_b', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=4466, + serialized_end=4779, +) + + +_SSDRANDOMCROPPAD = _descriptor.Descriptor( + name='SSDRandomCropPad', + full_name='object_detection.protos.SSDRandomCropPad', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='operations', full_name='object_detection.protos.SSDRandomCropPad.operations', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=4781, + serialized_end=4871, +) + + +_SSDRANDOMCROPFIXEDASPECTRATIOOPERATION = _descriptor.Descriptor( + name='SSDRandomCropFixedAspectRatioOperation', + full_name='object_detection.protos.SSDRandomCropFixedAspectRatioOperation', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='min_object_covered', full_name='object_detection.protos.SSDRandomCropFixedAspectRatioOperation.min_object_covered', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='min_area', full_name='object_detection.protos.SSDRandomCropFixedAspectRatioOperation.min_area', index=1, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_area', full_name='object_detection.protos.SSDRandomCropFixedAspectRatioOperation.max_area', index=2, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='overlap_thresh', full_name='object_detection.protos.SSDRandomCropFixedAspectRatioOperation.overlap_thresh', index=3, + number=6, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='random_coef', full_name='object_detection.protos.SSDRandomCropFixedAspectRatioOperation.random_coef', index=4, + 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=4874, + serialized_end=5023, +) + + +_SSDRANDOMCROPFIXEDASPECTRATIO = _descriptor.Descriptor( + name='SSDRandomCropFixedAspectRatio', + full_name='object_detection.protos.SSDRandomCropFixedAspectRatio', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='operations', full_name='object_detection.protos.SSDRandomCropFixedAspectRatio.operations', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='aspect_ratio', full_name='object_detection.protos.SSDRandomCropFixedAspectRatio.aspect_ratio', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=5026, + serialized_end=5167, +) + + +_SSDRANDOMCROPPADFIXEDASPECTRATIOOPERATION = _descriptor.Descriptor( + name='SSDRandomCropPadFixedAspectRatioOperation', + full_name='object_detection.protos.SSDRandomCropPadFixedAspectRatioOperation', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='min_object_covered', full_name='object_detection.protos.SSDRandomCropPadFixedAspectRatioOperation.min_object_covered', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='min_aspect_ratio', full_name='object_detection.protos.SSDRandomCropPadFixedAspectRatioOperation.min_aspect_ratio', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_aspect_ratio', full_name='object_detection.protos.SSDRandomCropPadFixedAspectRatioOperation.max_aspect_ratio', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='min_area', full_name='object_detection.protos.SSDRandomCropPadFixedAspectRatioOperation.min_area', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_area', full_name='object_detection.protos.SSDRandomCropPadFixedAspectRatioOperation.max_area', index=4, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='overlap_thresh', full_name='object_detection.protos.SSDRandomCropPadFixedAspectRatioOperation.overlap_thresh', index=5, + number=6, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='random_coef', full_name='object_detection.protos.SSDRandomCropPadFixedAspectRatioOperation.random_coef', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=5170, + serialized_end=5374, +) + + +_SSDRANDOMCROPPADFIXEDASPECTRATIO = _descriptor.Descriptor( + name='SSDRandomCropPadFixedAspectRatio', + full_name='object_detection.protos.SSDRandomCropPadFixedAspectRatio', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='operations', full_name='object_detection.protos.SSDRandomCropPadFixedAspectRatio.operations', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='aspect_ratio', full_name='object_detection.protos.SSDRandomCropPadFixedAspectRatio.aspect_ratio', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='min_padded_size_ratio', full_name='object_detection.protos.SSDRandomCropPadFixedAspectRatio.min_padded_size_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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_padded_size_ratio', full_name='object_detection.protos.SSDRandomCropPadFixedAspectRatio.max_padded_size_ratio', index=3, + number=4, 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=5377, + serialized_end=5586, +) + +_PREPROCESSINGSTEP.fields_by_name['normalize_image'].message_type = _NORMALIZEIMAGE +_PREPROCESSINGSTEP.fields_by_name['random_horizontal_flip'].message_type = _RANDOMHORIZONTALFLIP +_PREPROCESSINGSTEP.fields_by_name['random_pixel_value_scale'].message_type = _RANDOMPIXELVALUESCALE +_PREPROCESSINGSTEP.fields_by_name['random_image_scale'].message_type = _RANDOMIMAGESCALE +_PREPROCESSINGSTEP.fields_by_name['random_rgb_to_gray'].message_type = _RANDOMRGBTOGRAY +_PREPROCESSINGSTEP.fields_by_name['random_adjust_brightness'].message_type = _RANDOMADJUSTBRIGHTNESS +_PREPROCESSINGSTEP.fields_by_name['random_adjust_contrast'].message_type = _RANDOMADJUSTCONTRAST +_PREPROCESSINGSTEP.fields_by_name['random_adjust_hue'].message_type = _RANDOMADJUSTHUE +_PREPROCESSINGSTEP.fields_by_name['random_adjust_saturation'].message_type = _RANDOMADJUSTSATURATION +_PREPROCESSINGSTEP.fields_by_name['random_distort_color'].message_type = _RANDOMDISTORTCOLOR +_PREPROCESSINGSTEP.fields_by_name['random_jitter_boxes'].message_type = _RANDOMJITTERBOXES +_PREPROCESSINGSTEP.fields_by_name['random_crop_image'].message_type = _RANDOMCROPIMAGE +_PREPROCESSINGSTEP.fields_by_name['random_pad_image'].message_type = _RANDOMPADIMAGE +_PREPROCESSINGSTEP.fields_by_name['random_crop_pad_image'].message_type = _RANDOMCROPPADIMAGE +_PREPROCESSINGSTEP.fields_by_name['random_crop_to_aspect_ratio'].message_type = _RANDOMCROPTOASPECTRATIO +_PREPROCESSINGSTEP.fields_by_name['random_black_patches'].message_type = _RANDOMBLACKPATCHES +_PREPROCESSINGSTEP.fields_by_name['random_resize_method'].message_type = _RANDOMRESIZEMETHOD +_PREPROCESSINGSTEP.fields_by_name['scale_boxes_to_pixel_coordinates'].message_type = _SCALEBOXESTOPIXELCOORDINATES +_PREPROCESSINGSTEP.fields_by_name['resize_image'].message_type = _RESIZEIMAGE +_PREPROCESSINGSTEP.fields_by_name['subtract_channel_mean'].message_type = _SUBTRACTCHANNELMEAN +_PREPROCESSINGSTEP.fields_by_name['ssd_random_crop'].message_type = _SSDRANDOMCROP +_PREPROCESSINGSTEP.fields_by_name['ssd_random_crop_pad'].message_type = _SSDRANDOMCROPPAD +_PREPROCESSINGSTEP.fields_by_name['ssd_random_crop_fixed_aspect_ratio'].message_type = _SSDRANDOMCROPFIXEDASPECTRATIO +_PREPROCESSINGSTEP.fields_by_name['ssd_random_crop_pad_fixed_aspect_ratio'].message_type = _SSDRANDOMCROPPADFIXEDASPECTRATIO +_PREPROCESSINGSTEP.fields_by_name['random_vertical_flip'].message_type = _RANDOMVERTICALFLIP +_PREPROCESSINGSTEP.fields_by_name['random_rotation90'].message_type = _RANDOMROTATION90 +_PREPROCESSINGSTEP.fields_by_name['rgb_to_gray'].message_type = _RGBTOGRAY +_PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'].fields.append( + _PREPROCESSINGSTEP.fields_by_name['normalize_image']) +_PREPROCESSINGSTEP.fields_by_name['normalize_image'].containing_oneof = _PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'] +_PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'].fields.append( + _PREPROCESSINGSTEP.fields_by_name['random_horizontal_flip']) +_PREPROCESSINGSTEP.fields_by_name['random_horizontal_flip'].containing_oneof = _PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'] +_PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'].fields.append( + _PREPROCESSINGSTEP.fields_by_name['random_pixel_value_scale']) +_PREPROCESSINGSTEP.fields_by_name['random_pixel_value_scale'].containing_oneof = _PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'] +_PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'].fields.append( + _PREPROCESSINGSTEP.fields_by_name['random_image_scale']) +_PREPROCESSINGSTEP.fields_by_name['random_image_scale'].containing_oneof = _PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'] +_PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'].fields.append( + _PREPROCESSINGSTEP.fields_by_name['random_rgb_to_gray']) +_PREPROCESSINGSTEP.fields_by_name['random_rgb_to_gray'].containing_oneof = _PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'] +_PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'].fields.append( + _PREPROCESSINGSTEP.fields_by_name['random_adjust_brightness']) +_PREPROCESSINGSTEP.fields_by_name['random_adjust_brightness'].containing_oneof = _PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'] +_PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'].fields.append( + _PREPROCESSINGSTEP.fields_by_name['random_adjust_contrast']) +_PREPROCESSINGSTEP.fields_by_name['random_adjust_contrast'].containing_oneof = _PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'] +_PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'].fields.append( + _PREPROCESSINGSTEP.fields_by_name['random_adjust_hue']) +_PREPROCESSINGSTEP.fields_by_name['random_adjust_hue'].containing_oneof = _PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'] +_PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'].fields.append( + _PREPROCESSINGSTEP.fields_by_name['random_adjust_saturation']) +_PREPROCESSINGSTEP.fields_by_name['random_adjust_saturation'].containing_oneof = _PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'] +_PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'].fields.append( + _PREPROCESSINGSTEP.fields_by_name['random_distort_color']) +_PREPROCESSINGSTEP.fields_by_name['random_distort_color'].containing_oneof = _PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'] +_PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'].fields.append( + _PREPROCESSINGSTEP.fields_by_name['random_jitter_boxes']) +_PREPROCESSINGSTEP.fields_by_name['random_jitter_boxes'].containing_oneof = _PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'] +_PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'].fields.append( + _PREPROCESSINGSTEP.fields_by_name['random_crop_image']) +_PREPROCESSINGSTEP.fields_by_name['random_crop_image'].containing_oneof = _PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'] +_PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'].fields.append( + _PREPROCESSINGSTEP.fields_by_name['random_pad_image']) +_PREPROCESSINGSTEP.fields_by_name['random_pad_image'].containing_oneof = _PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'] +_PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'].fields.append( + _PREPROCESSINGSTEP.fields_by_name['random_crop_pad_image']) +_PREPROCESSINGSTEP.fields_by_name['random_crop_pad_image'].containing_oneof = _PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'] +_PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'].fields.append( + _PREPROCESSINGSTEP.fields_by_name['random_crop_to_aspect_ratio']) +_PREPROCESSINGSTEP.fields_by_name['random_crop_to_aspect_ratio'].containing_oneof = _PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'] +_PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'].fields.append( + _PREPROCESSINGSTEP.fields_by_name['random_black_patches']) +_PREPROCESSINGSTEP.fields_by_name['random_black_patches'].containing_oneof = _PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'] +_PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'].fields.append( + _PREPROCESSINGSTEP.fields_by_name['random_resize_method']) +_PREPROCESSINGSTEP.fields_by_name['random_resize_method'].containing_oneof = _PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'] +_PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'].fields.append( + _PREPROCESSINGSTEP.fields_by_name['scale_boxes_to_pixel_coordinates']) +_PREPROCESSINGSTEP.fields_by_name['scale_boxes_to_pixel_coordinates'].containing_oneof = _PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'] +_PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'].fields.append( + _PREPROCESSINGSTEP.fields_by_name['resize_image']) +_PREPROCESSINGSTEP.fields_by_name['resize_image'].containing_oneof = _PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'] +_PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'].fields.append( + _PREPROCESSINGSTEP.fields_by_name['subtract_channel_mean']) +_PREPROCESSINGSTEP.fields_by_name['subtract_channel_mean'].containing_oneof = _PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'] +_PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'].fields.append( + _PREPROCESSINGSTEP.fields_by_name['ssd_random_crop']) +_PREPROCESSINGSTEP.fields_by_name['ssd_random_crop'].containing_oneof = _PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'] +_PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'].fields.append( + _PREPROCESSINGSTEP.fields_by_name['ssd_random_crop_pad']) +_PREPROCESSINGSTEP.fields_by_name['ssd_random_crop_pad'].containing_oneof = _PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'] +_PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'].fields.append( + _PREPROCESSINGSTEP.fields_by_name['ssd_random_crop_fixed_aspect_ratio']) +_PREPROCESSINGSTEP.fields_by_name['ssd_random_crop_fixed_aspect_ratio'].containing_oneof = _PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'] +_PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'].fields.append( + _PREPROCESSINGSTEP.fields_by_name['ssd_random_crop_pad_fixed_aspect_ratio']) +_PREPROCESSINGSTEP.fields_by_name['ssd_random_crop_pad_fixed_aspect_ratio'].containing_oneof = _PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'] +_PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'].fields.append( + _PREPROCESSINGSTEP.fields_by_name['random_vertical_flip']) +_PREPROCESSINGSTEP.fields_by_name['random_vertical_flip'].containing_oneof = _PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'] +_PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'].fields.append( + _PREPROCESSINGSTEP.fields_by_name['random_rotation90']) +_PREPROCESSINGSTEP.fields_by_name['random_rotation90'].containing_oneof = _PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'] +_PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'].fields.append( + _PREPROCESSINGSTEP.fields_by_name['rgb_to_gray']) +_PREPROCESSINGSTEP.fields_by_name['rgb_to_gray'].containing_oneof = _PREPROCESSINGSTEP.oneofs_by_name['preprocessing_step'] +_RESIZEIMAGE.fields_by_name['method'].enum_type = _RESIZEIMAGE_METHOD +_RESIZEIMAGE_METHOD.containing_type = _RESIZEIMAGE +_SSDRANDOMCROP.fields_by_name['operations'].message_type = _SSDRANDOMCROPOPERATION +_SSDRANDOMCROPPAD.fields_by_name['operations'].message_type = _SSDRANDOMCROPPADOPERATION +_SSDRANDOMCROPFIXEDASPECTRATIO.fields_by_name['operations'].message_type = _SSDRANDOMCROPFIXEDASPECTRATIOOPERATION +_SSDRANDOMCROPPADFIXEDASPECTRATIO.fields_by_name['operations'].message_type = _SSDRANDOMCROPPADFIXEDASPECTRATIOOPERATION +DESCRIPTOR.message_types_by_name['PreprocessingStep'] = _PREPROCESSINGSTEP +DESCRIPTOR.message_types_by_name['NormalizeImage'] = _NORMALIZEIMAGE +DESCRIPTOR.message_types_by_name['RandomHorizontalFlip'] = _RANDOMHORIZONTALFLIP +DESCRIPTOR.message_types_by_name['RandomVerticalFlip'] = _RANDOMVERTICALFLIP +DESCRIPTOR.message_types_by_name['RandomRotation90'] = _RANDOMROTATION90 +DESCRIPTOR.message_types_by_name['RandomPixelValueScale'] = _RANDOMPIXELVALUESCALE +DESCRIPTOR.message_types_by_name['RandomImageScale'] = _RANDOMIMAGESCALE +DESCRIPTOR.message_types_by_name['RandomRGBtoGray'] = _RANDOMRGBTOGRAY +DESCRIPTOR.message_types_by_name['RandomAdjustBrightness'] = _RANDOMADJUSTBRIGHTNESS +DESCRIPTOR.message_types_by_name['RandomAdjustContrast'] = _RANDOMADJUSTCONTRAST +DESCRIPTOR.message_types_by_name['RandomAdjustHue'] = _RANDOMADJUSTHUE +DESCRIPTOR.message_types_by_name['RandomAdjustSaturation'] = _RANDOMADJUSTSATURATION +DESCRIPTOR.message_types_by_name['RandomDistortColor'] = _RANDOMDISTORTCOLOR +DESCRIPTOR.message_types_by_name['RandomJitterBoxes'] = _RANDOMJITTERBOXES +DESCRIPTOR.message_types_by_name['RandomCropImage'] = _RANDOMCROPIMAGE +DESCRIPTOR.message_types_by_name['RandomPadImage'] = _RANDOMPADIMAGE +DESCRIPTOR.message_types_by_name['RandomCropPadImage'] = _RANDOMCROPPADIMAGE +DESCRIPTOR.message_types_by_name['RandomCropToAspectRatio'] = _RANDOMCROPTOASPECTRATIO +DESCRIPTOR.message_types_by_name['RandomBlackPatches'] = _RANDOMBLACKPATCHES +DESCRIPTOR.message_types_by_name['RandomResizeMethod'] = _RANDOMRESIZEMETHOD +DESCRIPTOR.message_types_by_name['RGBtoGray'] = _RGBTOGRAY +DESCRIPTOR.message_types_by_name['ScaleBoxesToPixelCoordinates'] = _SCALEBOXESTOPIXELCOORDINATES +DESCRIPTOR.message_types_by_name['ResizeImage'] = _RESIZEIMAGE +DESCRIPTOR.message_types_by_name['SubtractChannelMean'] = _SUBTRACTCHANNELMEAN +DESCRIPTOR.message_types_by_name['SSDRandomCropOperation'] = _SSDRANDOMCROPOPERATION +DESCRIPTOR.message_types_by_name['SSDRandomCrop'] = _SSDRANDOMCROP +DESCRIPTOR.message_types_by_name['SSDRandomCropPadOperation'] = _SSDRANDOMCROPPADOPERATION +DESCRIPTOR.message_types_by_name['SSDRandomCropPad'] = _SSDRANDOMCROPPAD +DESCRIPTOR.message_types_by_name['SSDRandomCropFixedAspectRatioOperation'] = _SSDRANDOMCROPFIXEDASPECTRATIOOPERATION +DESCRIPTOR.message_types_by_name['SSDRandomCropFixedAspectRatio'] = _SSDRANDOMCROPFIXEDASPECTRATIO +DESCRIPTOR.message_types_by_name['SSDRandomCropPadFixedAspectRatioOperation'] = _SSDRANDOMCROPPADFIXEDASPECTRATIOOPERATION +DESCRIPTOR.message_types_by_name['SSDRandomCropPadFixedAspectRatio'] = _SSDRANDOMCROPPADFIXEDASPECTRATIO +_sym_db.RegisterFileDescriptor(DESCRIPTOR) + +PreprocessingStep = _reflection.GeneratedProtocolMessageType('PreprocessingStep', (_message.Message,), dict( + DESCRIPTOR = _PREPROCESSINGSTEP, + __module__ = 'object_detection.protos.preprocessor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.PreprocessingStep) + )) +_sym_db.RegisterMessage(PreprocessingStep) + +NormalizeImage = _reflection.GeneratedProtocolMessageType('NormalizeImage', (_message.Message,), dict( + DESCRIPTOR = _NORMALIZEIMAGE, + __module__ = 'object_detection.protos.preprocessor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.NormalizeImage) + )) +_sym_db.RegisterMessage(NormalizeImage) + +RandomHorizontalFlip = _reflection.GeneratedProtocolMessageType('RandomHorizontalFlip', (_message.Message,), dict( + DESCRIPTOR = _RANDOMHORIZONTALFLIP, + __module__ = 'object_detection.protos.preprocessor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.RandomHorizontalFlip) + )) +_sym_db.RegisterMessage(RandomHorizontalFlip) + +RandomVerticalFlip = _reflection.GeneratedProtocolMessageType('RandomVerticalFlip', (_message.Message,), dict( + DESCRIPTOR = _RANDOMVERTICALFLIP, + __module__ = 'object_detection.protos.preprocessor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.RandomVerticalFlip) + )) +_sym_db.RegisterMessage(RandomVerticalFlip) + +RandomRotation90 = _reflection.GeneratedProtocolMessageType('RandomRotation90', (_message.Message,), dict( + DESCRIPTOR = _RANDOMROTATION90, + __module__ = 'object_detection.protos.preprocessor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.RandomRotation90) + )) +_sym_db.RegisterMessage(RandomRotation90) + +RandomPixelValueScale = _reflection.GeneratedProtocolMessageType('RandomPixelValueScale', (_message.Message,), dict( + DESCRIPTOR = _RANDOMPIXELVALUESCALE, + __module__ = 'object_detection.protos.preprocessor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.RandomPixelValueScale) + )) +_sym_db.RegisterMessage(RandomPixelValueScale) + +RandomImageScale = _reflection.GeneratedProtocolMessageType('RandomImageScale', (_message.Message,), dict( + DESCRIPTOR = _RANDOMIMAGESCALE, + __module__ = 'object_detection.protos.preprocessor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.RandomImageScale) + )) +_sym_db.RegisterMessage(RandomImageScale) + +RandomRGBtoGray = _reflection.GeneratedProtocolMessageType('RandomRGBtoGray', (_message.Message,), dict( + DESCRIPTOR = _RANDOMRGBTOGRAY, + __module__ = 'object_detection.protos.preprocessor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.RandomRGBtoGray) + )) +_sym_db.RegisterMessage(RandomRGBtoGray) + +RandomAdjustBrightness = _reflection.GeneratedProtocolMessageType('RandomAdjustBrightness', (_message.Message,), dict( + DESCRIPTOR = _RANDOMADJUSTBRIGHTNESS, + __module__ = 'object_detection.protos.preprocessor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.RandomAdjustBrightness) + )) +_sym_db.RegisterMessage(RandomAdjustBrightness) + +RandomAdjustContrast = _reflection.GeneratedProtocolMessageType('RandomAdjustContrast', (_message.Message,), dict( + DESCRIPTOR = _RANDOMADJUSTCONTRAST, + __module__ = 'object_detection.protos.preprocessor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.RandomAdjustContrast) + )) +_sym_db.RegisterMessage(RandomAdjustContrast) + +RandomAdjustHue = _reflection.GeneratedProtocolMessageType('RandomAdjustHue', (_message.Message,), dict( + DESCRIPTOR = _RANDOMADJUSTHUE, + __module__ = 'object_detection.protos.preprocessor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.RandomAdjustHue) + )) +_sym_db.RegisterMessage(RandomAdjustHue) + +RandomAdjustSaturation = _reflection.GeneratedProtocolMessageType('RandomAdjustSaturation', (_message.Message,), dict( + DESCRIPTOR = _RANDOMADJUSTSATURATION, + __module__ = 'object_detection.protos.preprocessor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.RandomAdjustSaturation) + )) +_sym_db.RegisterMessage(RandomAdjustSaturation) + +RandomDistortColor = _reflection.GeneratedProtocolMessageType('RandomDistortColor', (_message.Message,), dict( + DESCRIPTOR = _RANDOMDISTORTCOLOR, + __module__ = 'object_detection.protos.preprocessor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.RandomDistortColor) + )) +_sym_db.RegisterMessage(RandomDistortColor) + +RandomJitterBoxes = _reflection.GeneratedProtocolMessageType('RandomJitterBoxes', (_message.Message,), dict( + DESCRIPTOR = _RANDOMJITTERBOXES, + __module__ = 'object_detection.protos.preprocessor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.RandomJitterBoxes) + )) +_sym_db.RegisterMessage(RandomJitterBoxes) + +RandomCropImage = _reflection.GeneratedProtocolMessageType('RandomCropImage', (_message.Message,), dict( + DESCRIPTOR = _RANDOMCROPIMAGE, + __module__ = 'object_detection.protos.preprocessor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.RandomCropImage) + )) +_sym_db.RegisterMessage(RandomCropImage) + +RandomPadImage = _reflection.GeneratedProtocolMessageType('RandomPadImage', (_message.Message,), dict( + DESCRIPTOR = _RANDOMPADIMAGE, + __module__ = 'object_detection.protos.preprocessor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.RandomPadImage) + )) +_sym_db.RegisterMessage(RandomPadImage) + +RandomCropPadImage = _reflection.GeneratedProtocolMessageType('RandomCropPadImage', (_message.Message,), dict( + DESCRIPTOR = _RANDOMCROPPADIMAGE, + __module__ = 'object_detection.protos.preprocessor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.RandomCropPadImage) + )) +_sym_db.RegisterMessage(RandomCropPadImage) + +RandomCropToAspectRatio = _reflection.GeneratedProtocolMessageType('RandomCropToAspectRatio', (_message.Message,), dict( + DESCRIPTOR = _RANDOMCROPTOASPECTRATIO, + __module__ = 'object_detection.protos.preprocessor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.RandomCropToAspectRatio) + )) +_sym_db.RegisterMessage(RandomCropToAspectRatio) + +RandomBlackPatches = _reflection.GeneratedProtocolMessageType('RandomBlackPatches', (_message.Message,), dict( + DESCRIPTOR = _RANDOMBLACKPATCHES, + __module__ = 'object_detection.protos.preprocessor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.RandomBlackPatches) + )) +_sym_db.RegisterMessage(RandomBlackPatches) + +RandomResizeMethod = _reflection.GeneratedProtocolMessageType('RandomResizeMethod', (_message.Message,), dict( + DESCRIPTOR = _RANDOMRESIZEMETHOD, + __module__ = 'object_detection.protos.preprocessor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.RandomResizeMethod) + )) +_sym_db.RegisterMessage(RandomResizeMethod) + +RGBtoGray = _reflection.GeneratedProtocolMessageType('RGBtoGray', (_message.Message,), dict( + DESCRIPTOR = _RGBTOGRAY, + __module__ = 'object_detection.protos.preprocessor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.RGBtoGray) + )) +_sym_db.RegisterMessage(RGBtoGray) + +ScaleBoxesToPixelCoordinates = _reflection.GeneratedProtocolMessageType('ScaleBoxesToPixelCoordinates', (_message.Message,), dict( + DESCRIPTOR = _SCALEBOXESTOPIXELCOORDINATES, + __module__ = 'object_detection.protos.preprocessor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.ScaleBoxesToPixelCoordinates) + )) +_sym_db.RegisterMessage(ScaleBoxesToPixelCoordinates) + +ResizeImage = _reflection.GeneratedProtocolMessageType('ResizeImage', (_message.Message,), dict( + DESCRIPTOR = _RESIZEIMAGE, + __module__ = 'object_detection.protos.preprocessor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.ResizeImage) + )) +_sym_db.RegisterMessage(ResizeImage) + +SubtractChannelMean = _reflection.GeneratedProtocolMessageType('SubtractChannelMean', (_message.Message,), dict( + DESCRIPTOR = _SUBTRACTCHANNELMEAN, + __module__ = 'object_detection.protos.preprocessor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.SubtractChannelMean) + )) +_sym_db.RegisterMessage(SubtractChannelMean) + +SSDRandomCropOperation = _reflection.GeneratedProtocolMessageType('SSDRandomCropOperation', (_message.Message,), dict( + DESCRIPTOR = _SSDRANDOMCROPOPERATION, + __module__ = 'object_detection.protos.preprocessor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.SSDRandomCropOperation) + )) +_sym_db.RegisterMessage(SSDRandomCropOperation) + +SSDRandomCrop = _reflection.GeneratedProtocolMessageType('SSDRandomCrop', (_message.Message,), dict( + DESCRIPTOR = _SSDRANDOMCROP, + __module__ = 'object_detection.protos.preprocessor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.SSDRandomCrop) + )) +_sym_db.RegisterMessage(SSDRandomCrop) + +SSDRandomCropPadOperation = _reflection.GeneratedProtocolMessageType('SSDRandomCropPadOperation', (_message.Message,), dict( + DESCRIPTOR = _SSDRANDOMCROPPADOPERATION, + __module__ = 'object_detection.protos.preprocessor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.SSDRandomCropPadOperation) + )) +_sym_db.RegisterMessage(SSDRandomCropPadOperation) + +SSDRandomCropPad = _reflection.GeneratedProtocolMessageType('SSDRandomCropPad', (_message.Message,), dict( + DESCRIPTOR = _SSDRANDOMCROPPAD, + __module__ = 'object_detection.protos.preprocessor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.SSDRandomCropPad) + )) +_sym_db.RegisterMessage(SSDRandomCropPad) + +SSDRandomCropFixedAspectRatioOperation = _reflection.GeneratedProtocolMessageType('SSDRandomCropFixedAspectRatioOperation', (_message.Message,), dict( + DESCRIPTOR = _SSDRANDOMCROPFIXEDASPECTRATIOOPERATION, + __module__ = 'object_detection.protos.preprocessor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.SSDRandomCropFixedAspectRatioOperation) + )) +_sym_db.RegisterMessage(SSDRandomCropFixedAspectRatioOperation) + +SSDRandomCropFixedAspectRatio = _reflection.GeneratedProtocolMessageType('SSDRandomCropFixedAspectRatio', (_message.Message,), dict( + DESCRIPTOR = _SSDRANDOMCROPFIXEDASPECTRATIO, + __module__ = 'object_detection.protos.preprocessor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.SSDRandomCropFixedAspectRatio) + )) +_sym_db.RegisterMessage(SSDRandomCropFixedAspectRatio) + +SSDRandomCropPadFixedAspectRatioOperation = _reflection.GeneratedProtocolMessageType('SSDRandomCropPadFixedAspectRatioOperation', (_message.Message,), dict( + DESCRIPTOR = _SSDRANDOMCROPPADFIXEDASPECTRATIOOPERATION, + __module__ = 'object_detection.protos.preprocessor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.SSDRandomCropPadFixedAspectRatioOperation) + )) +_sym_db.RegisterMessage(SSDRandomCropPadFixedAspectRatioOperation) + +SSDRandomCropPadFixedAspectRatio = _reflection.GeneratedProtocolMessageType('SSDRandomCropPadFixedAspectRatio', (_message.Message,), dict( + DESCRIPTOR = _SSDRANDOMCROPPADFIXEDASPECTRATIO, + __module__ = 'object_detection.protos.preprocessor_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.SSDRandomCropPadFixedAspectRatio) + )) +_sym_db.RegisterMessage(SSDRandomCropPadFixedAspectRatio) + + +# @@protoc_insertion_point(module_scope) diff --git a/research/object_detection/protos/region_similarity_calculator.proto b/research/object_detection/object_detection/protos/region_similarity_calculator.proto similarity index 100% rename from research/object_detection/protos/region_similarity_calculator.proto rename to research/object_detection/object_detection/protos/region_similarity_calculator.proto diff --git a/research/object_detection/object_detection/protos/region_similarity_calculator_pb2.py b/research/object_detection/object_detection/protos/region_similarity_calculator_pb2.py new file mode 100644 index 00000000000..9e260d84600 --- /dev/null +++ b/research/object_detection/object_detection/protos/region_similarity_calculator_pb2.py @@ -0,0 +1,194 @@ +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: object_detection/protos/region_similarity_calculator.proto + +import sys +_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) +from google.protobuf import descriptor as _descriptor +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +from google.protobuf import descriptor_pb2 +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor.FileDescriptor( + name='object_detection/protos/region_similarity_calculator.proto', + package='object_detection.protos', + syntax='proto2', + serialized_pb=_b('\n:object_detection/protos/region_similarity_calculator.proto\x12\x17object_detection.protos\"\x85\x02\n\x1aRegionSimilarityCalculator\x12N\n\x16neg_sq_dist_similarity\x18\x01 \x01(\x0b\x32,.object_detection.protos.NegSqDistSimilarityH\x00\x12@\n\x0eiou_similarity\x18\x02 \x01(\x0b\x32&.object_detection.protos.IouSimilarityH\x00\x12@\n\x0eioa_similarity\x18\x03 \x01(\x0b\x32&.object_detection.protos.IoaSimilarityH\x00\x42\x13\n\x11region_similarity\"\x15\n\x13NegSqDistSimilarity\"\x0f\n\rIouSimilarity\"\x0f\n\rIoaSimilarity') +) + + + + +_REGIONSIMILARITYCALCULATOR = _descriptor.Descriptor( + name='RegionSimilarityCalculator', + full_name='object_detection.protos.RegionSimilarityCalculator', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='neg_sq_dist_similarity', full_name='object_detection.protos.RegionSimilarityCalculator.neg_sq_dist_similarity', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='iou_similarity', full_name='object_detection.protos.RegionSimilarityCalculator.iou_similarity', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='ioa_similarity', full_name='object_detection.protos.RegionSimilarityCalculator.ioa_similarity', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + _descriptor.OneofDescriptor( + name='region_similarity', full_name='object_detection.protos.RegionSimilarityCalculator.region_similarity', + index=0, containing_type=None, fields=[]), + ], + serialized_start=88, + serialized_end=349, +) + + +_NEGSQDISTSIMILARITY = _descriptor.Descriptor( + name='NegSqDistSimilarity', + full_name='object_detection.protos.NegSqDistSimilarity', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=351, + serialized_end=372, +) + + +_IOUSIMILARITY = _descriptor.Descriptor( + name='IouSimilarity', + full_name='object_detection.protos.IouSimilarity', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=374, + serialized_end=389, +) + + +_IOASIMILARITY = _descriptor.Descriptor( + name='IoaSimilarity', + full_name='object_detection.protos.IoaSimilarity', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=391, + serialized_end=406, +) + +_REGIONSIMILARITYCALCULATOR.fields_by_name['neg_sq_dist_similarity'].message_type = _NEGSQDISTSIMILARITY +_REGIONSIMILARITYCALCULATOR.fields_by_name['iou_similarity'].message_type = _IOUSIMILARITY +_REGIONSIMILARITYCALCULATOR.fields_by_name['ioa_similarity'].message_type = _IOASIMILARITY +_REGIONSIMILARITYCALCULATOR.oneofs_by_name['region_similarity'].fields.append( + _REGIONSIMILARITYCALCULATOR.fields_by_name['neg_sq_dist_similarity']) +_REGIONSIMILARITYCALCULATOR.fields_by_name['neg_sq_dist_similarity'].containing_oneof = _REGIONSIMILARITYCALCULATOR.oneofs_by_name['region_similarity'] +_REGIONSIMILARITYCALCULATOR.oneofs_by_name['region_similarity'].fields.append( + _REGIONSIMILARITYCALCULATOR.fields_by_name['iou_similarity']) +_REGIONSIMILARITYCALCULATOR.fields_by_name['iou_similarity'].containing_oneof = _REGIONSIMILARITYCALCULATOR.oneofs_by_name['region_similarity'] +_REGIONSIMILARITYCALCULATOR.oneofs_by_name['region_similarity'].fields.append( + _REGIONSIMILARITYCALCULATOR.fields_by_name['ioa_similarity']) +_REGIONSIMILARITYCALCULATOR.fields_by_name['ioa_similarity'].containing_oneof = _REGIONSIMILARITYCALCULATOR.oneofs_by_name['region_similarity'] +DESCRIPTOR.message_types_by_name['RegionSimilarityCalculator'] = _REGIONSIMILARITYCALCULATOR +DESCRIPTOR.message_types_by_name['NegSqDistSimilarity'] = _NEGSQDISTSIMILARITY +DESCRIPTOR.message_types_by_name['IouSimilarity'] = _IOUSIMILARITY +DESCRIPTOR.message_types_by_name['IoaSimilarity'] = _IOASIMILARITY +_sym_db.RegisterFileDescriptor(DESCRIPTOR) + +RegionSimilarityCalculator = _reflection.GeneratedProtocolMessageType('RegionSimilarityCalculator', (_message.Message,), dict( + DESCRIPTOR = _REGIONSIMILARITYCALCULATOR, + __module__ = 'object_detection.protos.region_similarity_calculator_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.RegionSimilarityCalculator) + )) +_sym_db.RegisterMessage(RegionSimilarityCalculator) + +NegSqDistSimilarity = _reflection.GeneratedProtocolMessageType('NegSqDistSimilarity', (_message.Message,), dict( + DESCRIPTOR = _NEGSQDISTSIMILARITY, + __module__ = 'object_detection.protos.region_similarity_calculator_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.NegSqDistSimilarity) + )) +_sym_db.RegisterMessage(NegSqDistSimilarity) + +IouSimilarity = _reflection.GeneratedProtocolMessageType('IouSimilarity', (_message.Message,), dict( + DESCRIPTOR = _IOUSIMILARITY, + __module__ = 'object_detection.protos.region_similarity_calculator_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.IouSimilarity) + )) +_sym_db.RegisterMessage(IouSimilarity) + +IoaSimilarity = _reflection.GeneratedProtocolMessageType('IoaSimilarity', (_message.Message,), dict( + DESCRIPTOR = _IOASIMILARITY, + __module__ = 'object_detection.protos.region_similarity_calculator_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.IoaSimilarity) + )) +_sym_db.RegisterMessage(IoaSimilarity) + + +# @@protoc_insertion_point(module_scope) diff --git a/research/object_detection/protos/square_box_coder.proto b/research/object_detection/object_detection/protos/square_box_coder.proto similarity index 100% rename from research/object_detection/protos/square_box_coder.proto rename to research/object_detection/object_detection/protos/square_box_coder.proto diff --git a/research/object_detection/object_detection/protos/square_box_coder_pb2.py b/research/object_detection/object_detection/protos/square_box_coder_pb2.py new file mode 100644 index 00000000000..a75e3d06688 --- /dev/null +++ b/research/object_detection/object_detection/protos/square_box_coder_pb2.py @@ -0,0 +1,83 @@ +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: object_detection/protos/square_box_coder.proto + +import sys +_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) +from google.protobuf import descriptor as _descriptor +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +from google.protobuf import descriptor_pb2 +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor.FileDescriptor( + name='object_detection/protos/square_box_coder.proto', + package='object_detection.protos', + syntax='proto2', + serialized_pb=_b('\n.object_detection/protos/square_box_coder.proto\x12\x17object_detection.protos\"S\n\x0eSquareBoxCoder\x12\x13\n\x07y_scale\x18\x01 \x01(\x02:\x02\x31\x30\x12\x13\n\x07x_scale\x18\x02 \x01(\x02:\x02\x31\x30\x12\x17\n\x0clength_scale\x18\x03 \x01(\x02:\x01\x35') +) + + + + +_SQUAREBOXCODER = _descriptor.Descriptor( + name='SquareBoxCoder', + full_name='object_detection.protos.SquareBoxCoder', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='y_scale', full_name='object_detection.protos.SquareBoxCoder.y_scale', index=0, + number=1, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(10), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='x_scale', full_name='object_detection.protos.SquareBoxCoder.x_scale', index=1, + number=2, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(10), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='length_scale', full_name='object_detection.protos.SquareBoxCoder.length_scale', index=2, + number=3, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(5), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=75, + serialized_end=158, +) + +DESCRIPTOR.message_types_by_name['SquareBoxCoder'] = _SQUAREBOXCODER +_sym_db.RegisterFileDescriptor(DESCRIPTOR) + +SquareBoxCoder = _reflection.GeneratedProtocolMessageType('SquareBoxCoder', (_message.Message,), dict( + DESCRIPTOR = _SQUAREBOXCODER, + __module__ = 'object_detection.protos.square_box_coder_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.SquareBoxCoder) + )) +_sym_db.RegisterMessage(SquareBoxCoder) + + +# @@protoc_insertion_point(module_scope) diff --git a/research/object_detection/protos/ssd.proto b/research/object_detection/object_detection/protos/ssd.proto similarity index 100% rename from research/object_detection/protos/ssd.proto rename to research/object_detection/object_detection/protos/ssd.proto diff --git a/research/object_detection/protos/ssd_anchor_generator.proto b/research/object_detection/object_detection/protos/ssd_anchor_generator.proto similarity index 100% rename from research/object_detection/protos/ssd_anchor_generator.proto rename to research/object_detection/object_detection/protos/ssd_anchor_generator.proto diff --git a/research/object_detection/object_detection/protos/ssd_anchor_generator_pb2.py b/research/object_detection/object_detection/protos/ssd_anchor_generator_pb2.py new file mode 100644 index 00000000000..25fe45a0261 --- /dev/null +++ b/research/object_detection/object_detection/protos/ssd_anchor_generator_pb2.py @@ -0,0 +1,153 @@ +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: object_detection/protos/ssd_anchor_generator.proto + +import sys +_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) +from google.protobuf import descriptor as _descriptor +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +from google.protobuf import descriptor_pb2 +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor.FileDescriptor( + name='object_detection/protos/ssd_anchor_generator.proto', + package='object_detection.protos', + syntax='proto2', + serialized_pb=_b('\n2object_detection/protos/ssd_anchor_generator.proto\x12\x17object_detection.protos\"\xf2\x02\n\x12SsdAnchorGenerator\x12\x15\n\nnum_layers\x18\x01 \x01(\x05:\x01\x36\x12\x16\n\tmin_scale\x18\x02 \x01(\x02:\x03\x30.2\x12\x17\n\tmax_scale\x18\x03 \x01(\x02:\x04\x30.95\x12\x0e\n\x06scales\x18\x0c \x03(\x02\x12\x15\n\raspect_ratios\x18\x04 \x03(\x02\x12*\n\x1finterpolated_scale_aspect_ratio\x18\r \x01(\x02:\x01\x31\x12*\n\x1creduce_boxes_in_lowest_layer\x18\x05 \x01(\x08:\x04true\x12\x1d\n\x12\x62\x61se_anchor_height\x18\x06 \x01(\x02:\x01\x31\x12\x1c\n\x11\x62\x61se_anchor_width\x18\x07 \x01(\x02:\x01\x31\x12\x15\n\rheight_stride\x18\x08 \x03(\x05\x12\x14\n\x0cwidth_stride\x18\t \x03(\x05\x12\x15\n\rheight_offset\x18\n \x03(\x05\x12\x14\n\x0cwidth_offset\x18\x0b \x03(\x05') +) + + + + +_SSDANCHORGENERATOR = _descriptor.Descriptor( + name='SsdAnchorGenerator', + full_name='object_detection.protos.SsdAnchorGenerator', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='num_layers', full_name='object_detection.protos.SsdAnchorGenerator.num_layers', index=0, + number=1, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=6, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='min_scale', full_name='object_detection.protos.SsdAnchorGenerator.min_scale', index=1, + number=2, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(0.2), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_scale', full_name='object_detection.protos.SsdAnchorGenerator.max_scale', index=2, + number=3, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(0.95), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='scales', full_name='object_detection.protos.SsdAnchorGenerator.scales', index=3, + number=12, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='aspect_ratios', full_name='object_detection.protos.SsdAnchorGenerator.aspect_ratios', index=4, + number=4, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='interpolated_scale_aspect_ratio', full_name='object_detection.protos.SsdAnchorGenerator.interpolated_scale_aspect_ratio', index=5, + number=13, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='reduce_boxes_in_lowest_layer', full_name='object_detection.protos.SsdAnchorGenerator.reduce_boxes_in_lowest_layer', index=6, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='base_anchor_height', full_name='object_detection.protos.SsdAnchorGenerator.base_anchor_height', index=7, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='base_anchor_width', full_name='object_detection.protos.SsdAnchorGenerator.base_anchor_width', index=8, + number=7, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='height_stride', full_name='object_detection.protos.SsdAnchorGenerator.height_stride', index=9, + number=8, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='width_stride', full_name='object_detection.protos.SsdAnchorGenerator.width_stride', index=10, + number=9, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='height_offset', full_name='object_detection.protos.SsdAnchorGenerator.height_offset', index=11, + number=10, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='width_offset', full_name='object_detection.protos.SsdAnchorGenerator.width_offset', index=12, + number=11, 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=80, + serialized_end=450, +) + +DESCRIPTOR.message_types_by_name['SsdAnchorGenerator'] = _SSDANCHORGENERATOR +_sym_db.RegisterFileDescriptor(DESCRIPTOR) + +SsdAnchorGenerator = _reflection.GeneratedProtocolMessageType('SsdAnchorGenerator', (_message.Message,), dict( + DESCRIPTOR = _SSDANCHORGENERATOR, + __module__ = 'object_detection.protos.ssd_anchor_generator_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.SsdAnchorGenerator) + )) +_sym_db.RegisterMessage(SsdAnchorGenerator) + + +# @@protoc_insertion_point(module_scope) diff --git a/research/object_detection/object_detection/protos/ssd_pb2.py b/research/object_detection/object_detection/protos/ssd_pb2.py new file mode 100644 index 00000000000..f2613781062 --- /dev/null +++ b/research/object_detection/object_detection/protos/ssd_pb2.py @@ -0,0 +1,282 @@ +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: object_detection/protos/ssd.proto + +import sys +_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) +from google.protobuf import descriptor as _descriptor +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +from google.protobuf import descriptor_pb2 +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + +from object_detection.protos import anchor_generator_pb2 as object__detection_dot_protos_dot_anchor__generator__pb2 +from object_detection.protos import box_coder_pb2 as object__detection_dot_protos_dot_box__coder__pb2 +from object_detection.protos import box_predictor_pb2 as object__detection_dot_protos_dot_box__predictor__pb2 +from object_detection.protos import hyperparams_pb2 as object__detection_dot_protos_dot_hyperparams__pb2 +from object_detection.protos import image_resizer_pb2 as object__detection_dot_protos_dot_image__resizer__pb2 +from object_detection.protos import matcher_pb2 as object__detection_dot_protos_dot_matcher__pb2 +from object_detection.protos import losses_pb2 as object__detection_dot_protos_dot_losses__pb2 +from object_detection.protos import post_processing_pb2 as object__detection_dot_protos_dot_post__processing__pb2 +from object_detection.protos import region_similarity_calculator_pb2 as object__detection_dot_protos_dot_region__similarity__calculator__pb2 + + +DESCRIPTOR = _descriptor.FileDescriptor( + name='object_detection/protos/ssd.proto', + package='object_detection.protos', + syntax='proto2', + serialized_pb=_b('\n!object_detection/protos/ssd.proto\x12\x17object_detection.protos\x1a.object_detection/protos/anchor_generator.proto\x1a\'object_detection/protos/box_coder.proto\x1a+object_detection/protos/box_predictor.proto\x1a)object_detection/protos/hyperparams.proto\x1a+object_detection/protos/image_resizer.proto\x1a%object_detection/protos/matcher.proto\x1a$object_detection/protos/losses.proto\x1a-object_detection/protos/post_processing.proto\x1a:object_detection/protos/region_similarity_calculator.proto\"\xc2\x06\n\x03Ssd\x12\x13\n\x0bnum_classes\x18\x01 \x01(\x05\x12<\n\rimage_resizer\x18\x02 \x01(\x0b\x32%.object_detection.protos.ImageResizer\x12G\n\x11\x66\x65\x61ture_extractor\x18\x03 \x01(\x0b\x32,.object_detection.protos.SsdFeatureExtractor\x12\x34\n\tbox_coder\x18\x04 \x01(\x0b\x32!.object_detection.protos.BoxCoder\x12\x31\n\x07matcher\x18\x05 \x01(\x0b\x32 .object_detection.protos.Matcher\x12R\n\x15similarity_calculator\x18\x06 \x01(\x0b\x32\x33.object_detection.protos.RegionSimilarityCalculator\x12)\n\x1a\x65ncode_background_as_zeros\x18\x0c \x01(\x08:\x05\x66\x61lse\x12 \n\x15negative_class_weight\x18\r \x01(\x02:\x01\x31\x12<\n\rbox_predictor\x18\x07 \x01(\x0b\x32%.object_detection.protos.BoxPredictor\x12\x42\n\x10\x61nchor_generator\x18\x08 \x01(\x0b\x32(.object_detection.protos.AnchorGenerator\x12@\n\x0fpost_processing\x18\t \x01(\x0b\x32\'.object_detection.protos.PostProcessing\x12+\n\x1dnormalize_loss_by_num_matches\x18\n \x01(\x08:\x04true\x12-\n\x1enormalize_loc_loss_by_codesize\x18\x0e \x01(\x08:\x05\x66\x61lse\x12+\n\x04loss\x18\x0b \x01(\x0b\x32\x1d.object_detection.protos.Loss\x12\x1f\n\x10\x66reeze_batchnorm\x18\x10 \x01(\x08:\x05\x66\x61lse\x12\'\n\x18inplace_batchnorm_update\x18\x0f \x01(\x08:\x05\x66\x61lse\"\xb8\x02\n\x13SsdFeatureExtractor\x12\x0c\n\x04type\x18\x01 \x01(\t\x12\x1b\n\x10\x64\x65pth_multiplier\x18\x02 \x01(\x02:\x01\x31\x12\x15\n\tmin_depth\x18\x03 \x01(\x05:\x02\x31\x36\x12>\n\x10\x63onv_hyperparams\x18\x04 \x01(\x0b\x32$.object_detection.protos.Hyperparams\x12:\n+override_base_feature_extractor_hyperparams\x18\t \x01(\x08:\x05\x66\x61lse\x12\x1a\n\x0fpad_to_multiple\x18\x05 \x01(\x05:\x01\x31\x12#\n\x14use_explicit_padding\x18\x07 \x01(\x08:\x05\x66\x61lse\x12\x1c\n\ruse_depthwise\x18\x08 \x01(\x08:\x05\x66\x61lseJ\x04\x08\x06\x10\x07') + , + dependencies=[object__detection_dot_protos_dot_anchor__generator__pb2.DESCRIPTOR,object__detection_dot_protos_dot_box__coder__pb2.DESCRIPTOR,object__detection_dot_protos_dot_box__predictor__pb2.DESCRIPTOR,object__detection_dot_protos_dot_hyperparams__pb2.DESCRIPTOR,object__detection_dot_protos_dot_image__resizer__pb2.DESCRIPTOR,object__detection_dot_protos_dot_matcher__pb2.DESCRIPTOR,object__detection_dot_protos_dot_losses__pb2.DESCRIPTOR,object__detection_dot_protos_dot_post__processing__pb2.DESCRIPTOR,object__detection_dot_protos_dot_region__similarity__calculator__pb2.DESCRIPTOR,]) + + + + +_SSD = _descriptor.Descriptor( + name='Ssd', + full_name='object_detection.protos.Ssd', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='num_classes', full_name='object_detection.protos.Ssd.num_classes', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='image_resizer', full_name='object_detection.protos.Ssd.image_resizer', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='feature_extractor', full_name='object_detection.protos.Ssd.feature_extractor', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='box_coder', full_name='object_detection.protos.Ssd.box_coder', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='matcher', full_name='object_detection.protos.Ssd.matcher', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='similarity_calculator', full_name='object_detection.protos.Ssd.similarity_calculator', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='encode_background_as_zeros', full_name='object_detection.protos.Ssd.encode_background_as_zeros', index=6, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='negative_class_weight', full_name='object_detection.protos.Ssd.negative_class_weight', index=7, + number=13, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='box_predictor', full_name='object_detection.protos.Ssd.box_predictor', index=8, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='anchor_generator', full_name='object_detection.protos.Ssd.anchor_generator', index=9, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='post_processing', full_name='object_detection.protos.Ssd.post_processing', index=10, + number=9, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='normalize_loss_by_num_matches', full_name='object_detection.protos.Ssd.normalize_loss_by_num_matches', index=11, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='normalize_loc_loss_by_codesize', full_name='object_detection.protos.Ssd.normalize_loc_loss_by_codesize', index=12, + number=14, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='loss', full_name='object_detection.protos.Ssd.loss', index=13, + number=11, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='freeze_batchnorm', full_name='object_detection.protos.Ssd.freeze_batchnorm', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='inplace_batchnorm_update', full_name='object_detection.protos.Ssd.inplace_batchnorm_update', index=15, + number=15, 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=469, + serialized_end=1303, +) + + +_SSDFEATUREEXTRACTOR = _descriptor.Descriptor( + name='SsdFeatureExtractor', + full_name='object_detection.protos.SsdFeatureExtractor', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='type', full_name='object_detection.protos.SsdFeatureExtractor.type', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='depth_multiplier', full_name='object_detection.protos.SsdFeatureExtractor.depth_multiplier', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='min_depth', full_name='object_detection.protos.SsdFeatureExtractor.min_depth', index=2, + number=3, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=16, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='conv_hyperparams', full_name='object_detection.protos.SsdFeatureExtractor.conv_hyperparams', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='override_base_feature_extractor_hyperparams', full_name='object_detection.protos.SsdFeatureExtractor.override_base_feature_extractor_hyperparams', index=4, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='pad_to_multiple', full_name='object_detection.protos.SsdFeatureExtractor.pad_to_multiple', index=5, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='use_explicit_padding', full_name='object_detection.protos.SsdFeatureExtractor.use_explicit_padding', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='use_depthwise', full_name='object_detection.protos.SsdFeatureExtractor.use_depthwise', 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=1306, + serialized_end=1618, +) + +_SSD.fields_by_name['image_resizer'].message_type = object__detection_dot_protos_dot_image__resizer__pb2._IMAGERESIZER +_SSD.fields_by_name['feature_extractor'].message_type = _SSDFEATUREEXTRACTOR +_SSD.fields_by_name['box_coder'].message_type = object__detection_dot_protos_dot_box__coder__pb2._BOXCODER +_SSD.fields_by_name['matcher'].message_type = object__detection_dot_protos_dot_matcher__pb2._MATCHER +_SSD.fields_by_name['similarity_calculator'].message_type = object__detection_dot_protos_dot_region__similarity__calculator__pb2._REGIONSIMILARITYCALCULATOR +_SSD.fields_by_name['box_predictor'].message_type = object__detection_dot_protos_dot_box__predictor__pb2._BOXPREDICTOR +_SSD.fields_by_name['anchor_generator'].message_type = object__detection_dot_protos_dot_anchor__generator__pb2._ANCHORGENERATOR +_SSD.fields_by_name['post_processing'].message_type = object__detection_dot_protos_dot_post__processing__pb2._POSTPROCESSING +_SSD.fields_by_name['loss'].message_type = object__detection_dot_protos_dot_losses__pb2._LOSS +_SSDFEATUREEXTRACTOR.fields_by_name['conv_hyperparams'].message_type = object__detection_dot_protos_dot_hyperparams__pb2._HYPERPARAMS +DESCRIPTOR.message_types_by_name['Ssd'] = _SSD +DESCRIPTOR.message_types_by_name['SsdFeatureExtractor'] = _SSDFEATUREEXTRACTOR +_sym_db.RegisterFileDescriptor(DESCRIPTOR) + +Ssd = _reflection.GeneratedProtocolMessageType('Ssd', (_message.Message,), dict( + DESCRIPTOR = _SSD, + __module__ = 'object_detection.protos.ssd_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.Ssd) + )) +_sym_db.RegisterMessage(Ssd) + +SsdFeatureExtractor = _reflection.GeneratedProtocolMessageType('SsdFeatureExtractor', (_message.Message,), dict( + DESCRIPTOR = _SSDFEATUREEXTRACTOR, + __module__ = 'object_detection.protos.ssd_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.SsdFeatureExtractor) + )) +_sym_db.RegisterMessage(SsdFeatureExtractor) + + +# @@protoc_insertion_point(module_scope) diff --git a/research/object_detection/protos/string_int_label_map.proto b/research/object_detection/object_detection/protos/string_int_label_map.proto similarity index 100% rename from research/object_detection/protos/string_int_label_map.proto rename to research/object_detection/object_detection/protos/string_int_label_map.proto diff --git a/research/object_detection/object_detection/protos/string_int_label_map_pb2.py b/research/object_detection/object_detection/protos/string_int_label_map_pb2.py new file mode 100644 index 00000000000..381d5526656 --- /dev/null +++ b/research/object_detection/object_detection/protos/string_int_label_map_pb2.py @@ -0,0 +1,123 @@ +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: object_detection/protos/string_int_label_map.proto + +import sys +_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) +from google.protobuf import descriptor as _descriptor +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +from google.protobuf import descriptor_pb2 +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor.FileDescriptor( + name='object_detection/protos/string_int_label_map.proto', + package='object_detection.protos', + syntax='proto2', + serialized_pb=_b('\n2object_detection/protos/string_int_label_map.proto\x12\x17object_detection.protos\"G\n\x15StringIntLabelMapItem\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\n\n\x02id\x18\x02 \x01(\x05\x12\x14\n\x0c\x64isplay_name\x18\x03 \x01(\t\"Q\n\x11StringIntLabelMap\x12<\n\x04item\x18\x01 \x03(\x0b\x32..object_detection.protos.StringIntLabelMapItem') +) + + + + +_STRINGINTLABELMAPITEM = _descriptor.Descriptor( + name='StringIntLabelMapItem', + full_name='object_detection.protos.StringIntLabelMapItem', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='name', full_name='object_detection.protos.StringIntLabelMapItem.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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='id', full_name='object_detection.protos.StringIntLabelMapItem.id', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='display_name', full_name='object_detection.protos.StringIntLabelMapItem.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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=79, + serialized_end=150, +) + + +_STRINGINTLABELMAP = _descriptor.Descriptor( + name='StringIntLabelMap', + full_name='object_detection.protos.StringIntLabelMap', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='item', full_name='object_detection.protos.StringIntLabelMap.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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=152, + serialized_end=233, +) + +_STRINGINTLABELMAP.fields_by_name['item'].message_type = _STRINGINTLABELMAPITEM +DESCRIPTOR.message_types_by_name['StringIntLabelMapItem'] = _STRINGINTLABELMAPITEM +DESCRIPTOR.message_types_by_name['StringIntLabelMap'] = _STRINGINTLABELMAP +_sym_db.RegisterFileDescriptor(DESCRIPTOR) + +StringIntLabelMapItem = _reflection.GeneratedProtocolMessageType('StringIntLabelMapItem', (_message.Message,), dict( + DESCRIPTOR = _STRINGINTLABELMAPITEM, + __module__ = 'object_detection.protos.string_int_label_map_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.StringIntLabelMapItem) + )) +_sym_db.RegisterMessage(StringIntLabelMapItem) + +StringIntLabelMap = _reflection.GeneratedProtocolMessageType('StringIntLabelMap', (_message.Message,), dict( + DESCRIPTOR = _STRINGINTLABELMAP, + __module__ = 'object_detection.protos.string_int_label_map_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.StringIntLabelMap) + )) +_sym_db.RegisterMessage(StringIntLabelMap) + + +# @@protoc_insertion_point(module_scope) diff --git a/research/object_detection/protos/train.proto b/research/object_detection/object_detection/protos/train.proto similarity index 100% rename from research/object_detection/protos/train.proto rename to research/object_detection/object_detection/protos/train.proto diff --git a/research/object_detection/object_detection/protos/train_pb2.py b/research/object_detection/object_detection/protos/train_pb2.py new file mode 100644 index 00000000000..48797420018 --- /dev/null +++ b/research/object_detection/object_detection/protos/train_pb2.py @@ -0,0 +1,239 @@ +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: object_detection/protos/train.proto + +import sys +_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) +from google.protobuf import descriptor as _descriptor +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +from google.protobuf import descriptor_pb2 +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + +from object_detection.protos import optimizer_pb2 as object__detection_dot_protos_dot_optimizer__pb2 +from object_detection.protos import preprocessor_pb2 as object__detection_dot_protos_dot_preprocessor__pb2 + + +DESCRIPTOR = _descriptor.FileDescriptor( + name='object_detection/protos/train.proto', + package='object_detection.protos', + syntax='proto2', + serialized_pb=_b('\n#object_detection/protos/train.proto\x12\x17object_detection.protos\x1a\'object_detection/protos/optimizer.proto\x1a*object_detection/protos/preprocessor.proto\"\xaf\x07\n\x0bTrainConfig\x12\x16\n\nbatch_size\x18\x01 \x01(\r:\x02\x33\x32\x12M\n\x19\x64\x61ta_augmentation_options\x18\x02 \x03(\x0b\x32*.object_detection.protos.PreprocessingStep\x12\x1c\n\rsync_replicas\x18\x03 \x01(\x08:\x05\x66\x61lse\x12+\n\x1dkeep_checkpoint_every_n_hours\x18\x04 \x01(\r:\x04\x31\x30\x30\x30\x12\x35\n\toptimizer\x18\x05 \x01(\x0b\x32\".object_detection.protos.Optimizer\x12$\n\x19gradient_clipping_by_norm\x18\x06 \x01(\x02:\x01\x30\x12\x1e\n\x14\x66ine_tune_checkpoint\x18\x07 \x01(\t:\x00\x12#\n\x19\x66ine_tune_checkpoint_type\x18\x16 \x01(\t:\x00\x12,\n\x19\x66rom_detection_checkpoint\x18\x08 \x01(\x08:\x05\x66\x61lseB\x02\x18\x01\x12\x31\n\"load_all_detection_checkpoint_vars\x18\x13 \x01(\x08:\x05\x66\x61lse\x12\x14\n\tnum_steps\x18\t \x01(\r:\x01\x30\x12\x1f\n\x13startup_delay_steps\x18\n \x01(\x02:\x02\x31\x35\x12\x1f\n\x14\x62ias_grad_multiplier\x18\x0b \x01(\x02:\x01\x30\x12\x18\n\x10\x66reeze_variables\x18\x0c \x03(\t\x12 \n\x15replicas_to_aggregate\x18\r \x01(\x05:\x01\x31\x12!\n\x14\x62\x61tch_queue_capacity\x18\x0e \x01(\x05:\x03\x31\x35\x30\x12\"\n\x17num_batch_queue_threads\x18\x0f \x01(\x05:\x01\x38\x12\"\n\x17prefetch_queue_capacity\x18\x10 \x01(\x05:\x01\x35\x12)\n\x1amerge_multiple_label_boxes\x18\x11 \x01(\x08:\x05\x66\x61lse\x12$\n\x15use_multiclass_scores\x18\x18 \x01(\x08:\x05\x66\x61lse\x12%\n\x17\x61\x64\x64_regularization_loss\x18\x12 \x01(\x08:\x04true\x12$\n\x13max_number_of_boxes\x18\x14 \x01(\x05:\x03\x31\x30\x30\x42\x02\x18\x01\x12\'\n\x19unpad_groundtruth_tensors\x18\x15 \x01(\x08:\x04true\x12%\n\x16retain_original_images\x18\x17 \x01(\x08:\x05\x66\x61lse') + , + dependencies=[object__detection_dot_protos_dot_optimizer__pb2.DESCRIPTOR,object__detection_dot_protos_dot_preprocessor__pb2.DESCRIPTOR,]) + + + + +_TRAINCONFIG = _descriptor.Descriptor( + name='TrainConfig', + full_name='object_detection.protos.TrainConfig', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='batch_size', full_name='object_detection.protos.TrainConfig.batch_size', index=0, + number=1, type=13, cpp_type=3, label=1, + has_default_value=True, default_value=32, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='data_augmentation_options', full_name='object_detection.protos.TrainConfig.data_augmentation_options', index=1, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='sync_replicas', full_name='object_detection.protos.TrainConfig.sync_replicas', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='keep_checkpoint_every_n_hours', full_name='object_detection.protos.TrainConfig.keep_checkpoint_every_n_hours', index=3, + number=4, type=13, cpp_type=3, label=1, + has_default_value=True, default_value=1000, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='optimizer', full_name='object_detection.protos.TrainConfig.optimizer', 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='gradient_clipping_by_norm', full_name='object_detection.protos.TrainConfig.gradient_clipping_by_norm', index=5, + number=6, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='fine_tune_checkpoint', full_name='object_detection.protos.TrainConfig.fine_tune_checkpoint', index=6, + number=7, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='fine_tune_checkpoint_type', full_name='object_detection.protos.TrainConfig.fine_tune_checkpoint_type', index=7, + number=22, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='from_detection_checkpoint', full_name='object_detection.protos.TrainConfig.from_detection_checkpoint', index=8, + 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, + options=_descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\030\001')), file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='load_all_detection_checkpoint_vars', full_name='object_detection.protos.TrainConfig.load_all_detection_checkpoint_vars', index=9, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='num_steps', full_name='object_detection.protos.TrainConfig.num_steps', index=10, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='startup_delay_steps', full_name='object_detection.protos.TrainConfig.startup_delay_steps', index=11, + number=10, type=2, cpp_type=6, label=1, + has_default_value=True, default_value=float(15), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='bias_grad_multiplier', full_name='object_detection.protos.TrainConfig.bias_grad_multiplier', index=12, + 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, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='freeze_variables', full_name='object_detection.protos.TrainConfig.freeze_variables', index=13, + number=12, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='replicas_to_aggregate', full_name='object_detection.protos.TrainConfig.replicas_to_aggregate', index=14, + number=13, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='batch_queue_capacity', full_name='object_detection.protos.TrainConfig.batch_queue_capacity', index=15, + number=14, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=150, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='num_batch_queue_threads', full_name='object_detection.protos.TrainConfig.num_batch_queue_threads', index=16, + number=15, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=8, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='prefetch_queue_capacity', full_name='object_detection.protos.TrainConfig.prefetch_queue_capacity', index=17, + number=16, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=5, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='merge_multiple_label_boxes', full_name='object_detection.protos.TrainConfig.merge_multiple_label_boxes', index=18, + 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='use_multiclass_scores', full_name='object_detection.protos.TrainConfig.use_multiclass_scores', index=19, + number=24, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='add_regularization_loss', full_name='object_detection.protos.TrainConfig.add_regularization_loss', index=20, + number=18, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_number_of_boxes', full_name='object_detection.protos.TrainConfig.max_number_of_boxes', index=21, + number=20, type=5, cpp_type=1, label=1, + has_default_value=True, default_value=100, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + options=_descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\030\001')), file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='unpad_groundtruth_tensors', full_name='object_detection.protos.TrainConfig.unpad_groundtruth_tensors', index=22, + number=21, 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, + options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='retain_original_images', full_name='object_detection.protos.TrainConfig.retain_original_images', index=23, + 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, + options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=150, + serialized_end=1093, +) + +_TRAINCONFIG.fields_by_name['data_augmentation_options'].message_type = object__detection_dot_protos_dot_preprocessor__pb2._PREPROCESSINGSTEP +_TRAINCONFIG.fields_by_name['optimizer'].message_type = object__detection_dot_protos_dot_optimizer__pb2._OPTIMIZER +DESCRIPTOR.message_types_by_name['TrainConfig'] = _TRAINCONFIG +_sym_db.RegisterFileDescriptor(DESCRIPTOR) + +TrainConfig = _reflection.GeneratedProtocolMessageType('TrainConfig', (_message.Message,), dict( + DESCRIPTOR = _TRAINCONFIG, + __module__ = 'object_detection.protos.train_pb2' + # @@protoc_insertion_point(class_scope:object_detection.protos.TrainConfig) + )) +_sym_db.RegisterMessage(TrainConfig) + + +_TRAINCONFIG.fields_by_name['from_detection_checkpoint'].has_options = True +_TRAINCONFIG.fields_by_name['from_detection_checkpoint']._options = _descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\030\001')) +_TRAINCONFIG.fields_by_name['max_number_of_boxes'].has_options = True +_TRAINCONFIG.fields_by_name['max_number_of_boxes']._options = _descriptor._ParseOptions(descriptor_pb2.FieldOptions(), _b('\030\001')) +# @@protoc_insertion_point(module_scope) diff --git a/research/object_detection/samples/cloud/cloud.yml b/research/object_detection/object_detection/samples/cloud/cloud.yml similarity index 100% rename from research/object_detection/samples/cloud/cloud.yml rename to research/object_detection/object_detection/samples/cloud/cloud.yml diff --git a/research/object_detection/samples/configs/embedded_ssd_mobilenet_v1_coco.config b/research/object_detection/object_detection/samples/configs/embedded_ssd_mobilenet_v1_coco.config similarity index 100% rename from research/object_detection/samples/configs/embedded_ssd_mobilenet_v1_coco.config rename to research/object_detection/object_detection/samples/configs/embedded_ssd_mobilenet_v1_coco.config diff --git a/research/object_detection/samples/configs/faster_rcnn_inception_resnet_v2_atrous_coco.config b/research/object_detection/object_detection/samples/configs/faster_rcnn_inception_resnet_v2_atrous_coco.config similarity index 100% rename from research/object_detection/samples/configs/faster_rcnn_inception_resnet_v2_atrous_coco.config rename to research/object_detection/object_detection/samples/configs/faster_rcnn_inception_resnet_v2_atrous_coco.config diff --git a/research/object_detection/samples/configs/faster_rcnn_inception_resnet_v2_atrous_cosine_lr_coco.config b/research/object_detection/object_detection/samples/configs/faster_rcnn_inception_resnet_v2_atrous_cosine_lr_coco.config similarity index 100% rename from research/object_detection/samples/configs/faster_rcnn_inception_resnet_v2_atrous_cosine_lr_coco.config rename to research/object_detection/object_detection/samples/configs/faster_rcnn_inception_resnet_v2_atrous_cosine_lr_coco.config diff --git a/research/object_detection/samples/configs/faster_rcnn_inception_resnet_v2_atrous_oid.config b/research/object_detection/object_detection/samples/configs/faster_rcnn_inception_resnet_v2_atrous_oid.config similarity index 100% rename from research/object_detection/samples/configs/faster_rcnn_inception_resnet_v2_atrous_oid.config rename to research/object_detection/object_detection/samples/configs/faster_rcnn_inception_resnet_v2_atrous_oid.config diff --git a/research/object_detection/samples/configs/faster_rcnn_inception_resnet_v2_atrous_pets.config b/research/object_detection/object_detection/samples/configs/faster_rcnn_inception_resnet_v2_atrous_pets.config similarity index 100% rename from research/object_detection/samples/configs/faster_rcnn_inception_resnet_v2_atrous_pets.config rename to research/object_detection/object_detection/samples/configs/faster_rcnn_inception_resnet_v2_atrous_pets.config diff --git a/research/object_detection/samples/configs/faster_rcnn_inception_v2_coco.config b/research/object_detection/object_detection/samples/configs/faster_rcnn_inception_v2_coco.config similarity index 100% rename from research/object_detection/samples/configs/faster_rcnn_inception_v2_coco.config rename to research/object_detection/object_detection/samples/configs/faster_rcnn_inception_v2_coco.config diff --git a/research/object_detection/samples/configs/faster_rcnn_inception_v2_pets.config b/research/object_detection/object_detection/samples/configs/faster_rcnn_inception_v2_pets.config similarity index 100% rename from research/object_detection/samples/configs/faster_rcnn_inception_v2_pets.config rename to research/object_detection/object_detection/samples/configs/faster_rcnn_inception_v2_pets.config diff --git a/research/object_detection/samples/configs/faster_rcnn_nas_coco.config b/research/object_detection/object_detection/samples/configs/faster_rcnn_nas_coco.config similarity index 100% rename from research/object_detection/samples/configs/faster_rcnn_nas_coco.config rename to research/object_detection/object_detection/samples/configs/faster_rcnn_nas_coco.config diff --git a/research/object_detection/samples/configs/faster_rcnn_resnet101_atrous_coco.config b/research/object_detection/object_detection/samples/configs/faster_rcnn_resnet101_atrous_coco.config similarity index 100% rename from research/object_detection/samples/configs/faster_rcnn_resnet101_atrous_coco.config rename to research/object_detection/object_detection/samples/configs/faster_rcnn_resnet101_atrous_coco.config diff --git a/research/object_detection/samples/configs/faster_rcnn_resnet101_ava_v2.1.config b/research/object_detection/object_detection/samples/configs/faster_rcnn_resnet101_ava_v2.1.config similarity index 100% rename from research/object_detection/samples/configs/faster_rcnn_resnet101_ava_v2.1.config rename to research/object_detection/object_detection/samples/configs/faster_rcnn_resnet101_ava_v2.1.config diff --git a/research/object_detection/samples/configs/faster_rcnn_resnet101_coco.config b/research/object_detection/object_detection/samples/configs/faster_rcnn_resnet101_coco.config similarity index 100% rename from research/object_detection/samples/configs/faster_rcnn_resnet101_coco.config rename to research/object_detection/object_detection/samples/configs/faster_rcnn_resnet101_coco.config diff --git a/research/object_detection/samples/configs/faster_rcnn_resnet101_kitti.config b/research/object_detection/object_detection/samples/configs/faster_rcnn_resnet101_kitti.config similarity index 100% rename from research/object_detection/samples/configs/faster_rcnn_resnet101_kitti.config rename to research/object_detection/object_detection/samples/configs/faster_rcnn_resnet101_kitti.config diff --git a/research/object_detection/samples/configs/faster_rcnn_resnet101_pets.config b/research/object_detection/object_detection/samples/configs/faster_rcnn_resnet101_pets.config similarity index 100% rename from research/object_detection/samples/configs/faster_rcnn_resnet101_pets.config rename to research/object_detection/object_detection/samples/configs/faster_rcnn_resnet101_pets.config diff --git a/research/object_detection/samples/configs/faster_rcnn_resnet101_voc07.config b/research/object_detection/object_detection/samples/configs/faster_rcnn_resnet101_voc07.config similarity index 100% rename from research/object_detection/samples/configs/faster_rcnn_resnet101_voc07.config rename to research/object_detection/object_detection/samples/configs/faster_rcnn_resnet101_voc07.config diff --git a/research/object_detection/samples/configs/faster_rcnn_resnet152_coco.config b/research/object_detection/object_detection/samples/configs/faster_rcnn_resnet152_coco.config similarity index 100% rename from research/object_detection/samples/configs/faster_rcnn_resnet152_coco.config rename to research/object_detection/object_detection/samples/configs/faster_rcnn_resnet152_coco.config diff --git a/research/object_detection/samples/configs/faster_rcnn_resnet152_pets.config b/research/object_detection/object_detection/samples/configs/faster_rcnn_resnet152_pets.config similarity index 100% rename from research/object_detection/samples/configs/faster_rcnn_resnet152_pets.config rename to research/object_detection/object_detection/samples/configs/faster_rcnn_resnet152_pets.config diff --git a/research/object_detection/samples/configs/faster_rcnn_resnet50_coco.config b/research/object_detection/object_detection/samples/configs/faster_rcnn_resnet50_coco.config similarity index 100% rename from research/object_detection/samples/configs/faster_rcnn_resnet50_coco.config rename to research/object_detection/object_detection/samples/configs/faster_rcnn_resnet50_coco.config diff --git a/research/object_detection/samples/configs/faster_rcnn_resnet50_pets.config b/research/object_detection/object_detection/samples/configs/faster_rcnn_resnet50_pets.config similarity index 100% rename from research/object_detection/samples/configs/faster_rcnn_resnet50_pets.config rename to research/object_detection/object_detection/samples/configs/faster_rcnn_resnet50_pets.config diff --git a/research/object_detection/samples/configs/mask_rcnn_inception_resnet_v2_atrous_coco.config b/research/object_detection/object_detection/samples/configs/mask_rcnn_inception_resnet_v2_atrous_coco.config similarity index 100% rename from research/object_detection/samples/configs/mask_rcnn_inception_resnet_v2_atrous_coco.config rename to research/object_detection/object_detection/samples/configs/mask_rcnn_inception_resnet_v2_atrous_coco.config diff --git a/research/object_detection/samples/configs/mask_rcnn_inception_v2_coco.config b/research/object_detection/object_detection/samples/configs/mask_rcnn_inception_v2_coco.config similarity index 100% rename from research/object_detection/samples/configs/mask_rcnn_inception_v2_coco.config rename to research/object_detection/object_detection/samples/configs/mask_rcnn_inception_v2_coco.config diff --git a/research/object_detection/samples/configs/mask_rcnn_resnet101_atrous_coco.config b/research/object_detection/object_detection/samples/configs/mask_rcnn_resnet101_atrous_coco.config similarity index 100% rename from research/object_detection/samples/configs/mask_rcnn_resnet101_atrous_coco.config rename to research/object_detection/object_detection/samples/configs/mask_rcnn_resnet101_atrous_coco.config diff --git a/research/object_detection/samples/configs/mask_rcnn_resnet101_pets.config b/research/object_detection/object_detection/samples/configs/mask_rcnn_resnet101_pets.config similarity index 100% rename from research/object_detection/samples/configs/mask_rcnn_resnet101_pets.config rename to research/object_detection/object_detection/samples/configs/mask_rcnn_resnet101_pets.config diff --git a/research/object_detection/samples/configs/mask_rcnn_resnet50_atrous_coco.config b/research/object_detection/object_detection/samples/configs/mask_rcnn_resnet50_atrous_coco.config similarity index 100% rename from research/object_detection/samples/configs/mask_rcnn_resnet50_atrous_coco.config rename to research/object_detection/object_detection/samples/configs/mask_rcnn_resnet50_atrous_coco.config diff --git a/research/object_detection/samples/configs/rfcn_resnet101_coco.config b/research/object_detection/object_detection/samples/configs/rfcn_resnet101_coco.config similarity index 100% rename from research/object_detection/samples/configs/rfcn_resnet101_coco.config rename to research/object_detection/object_detection/samples/configs/rfcn_resnet101_coco.config diff --git a/research/object_detection/samples/configs/rfcn_resnet101_pets.config b/research/object_detection/object_detection/samples/configs/rfcn_resnet101_pets.config similarity index 100% rename from research/object_detection/samples/configs/rfcn_resnet101_pets.config rename to research/object_detection/object_detection/samples/configs/rfcn_resnet101_pets.config diff --git a/research/object_detection/samples/configs/ssd_inception_v2_coco.config b/research/object_detection/object_detection/samples/configs/ssd_inception_v2_coco.config similarity index 100% rename from research/object_detection/samples/configs/ssd_inception_v2_coco.config rename to research/object_detection/object_detection/samples/configs/ssd_inception_v2_coco.config diff --git a/research/object_detection/samples/configs/ssd_inception_v2_pets.config b/research/object_detection/object_detection/samples/configs/ssd_inception_v2_pets.config similarity index 100% rename from research/object_detection/samples/configs/ssd_inception_v2_pets.config rename to research/object_detection/object_detection/samples/configs/ssd_inception_v2_pets.config diff --git a/research/object_detection/samples/configs/ssd_inception_v3_pets.config b/research/object_detection/object_detection/samples/configs/ssd_inception_v3_pets.config similarity index 100% rename from research/object_detection/samples/configs/ssd_inception_v3_pets.config rename to research/object_detection/object_detection/samples/configs/ssd_inception_v3_pets.config diff --git a/research/object_detection/samples/configs/ssd_mobilenet_v1_0.75_depth_300x300_coco14_sync.config b/research/object_detection/object_detection/samples/configs/ssd_mobilenet_v1_0.75_depth_300x300_coco14_sync.config similarity index 100% rename from research/object_detection/samples/configs/ssd_mobilenet_v1_0.75_depth_300x300_coco14_sync.config rename to research/object_detection/object_detection/samples/configs/ssd_mobilenet_v1_0.75_depth_300x300_coco14_sync.config diff --git a/research/object_detection/samples/configs/ssd_mobilenet_v1_0.75_depth_quantized_300x300_coco14_sync.config b/research/object_detection/object_detection/samples/configs/ssd_mobilenet_v1_0.75_depth_quantized_300x300_coco14_sync.config similarity index 100% rename from research/object_detection/samples/configs/ssd_mobilenet_v1_0.75_depth_quantized_300x300_coco14_sync.config rename to research/object_detection/object_detection/samples/configs/ssd_mobilenet_v1_0.75_depth_quantized_300x300_coco14_sync.config diff --git a/research/object_detection/samples/configs/ssd_mobilenet_v1_0.75_depth_quantized_300x300_pets_sync.config b/research/object_detection/object_detection/samples/configs/ssd_mobilenet_v1_0.75_depth_quantized_300x300_pets_sync.config similarity index 100% rename from research/object_detection/samples/configs/ssd_mobilenet_v1_0.75_depth_quantized_300x300_pets_sync.config rename to research/object_detection/object_detection/samples/configs/ssd_mobilenet_v1_0.75_depth_quantized_300x300_pets_sync.config diff --git a/research/object_detection/samples/configs/ssd_mobilenet_v1_300x300_coco14_sync.config b/research/object_detection/object_detection/samples/configs/ssd_mobilenet_v1_300x300_coco14_sync.config similarity index 100% rename from research/object_detection/samples/configs/ssd_mobilenet_v1_300x300_coco14_sync.config rename to research/object_detection/object_detection/samples/configs/ssd_mobilenet_v1_300x300_coco14_sync.config diff --git a/research/object_detection/samples/configs/ssd_mobilenet_v1_coco.config b/research/object_detection/object_detection/samples/configs/ssd_mobilenet_v1_coco.config similarity index 100% rename from research/object_detection/samples/configs/ssd_mobilenet_v1_coco.config rename to research/object_detection/object_detection/samples/configs/ssd_mobilenet_v1_coco.config diff --git a/research/object_detection/samples/configs/ssd_mobilenet_v1_focal_loss_pets.config b/research/object_detection/object_detection/samples/configs/ssd_mobilenet_v1_focal_loss_pets.config similarity index 100% rename from research/object_detection/samples/configs/ssd_mobilenet_v1_focal_loss_pets.config rename to research/object_detection/object_detection/samples/configs/ssd_mobilenet_v1_focal_loss_pets.config diff --git a/research/object_detection/samples/configs/ssd_mobilenet_v1_fpn_shared_box_predictor_640x640_coco14_sync.config b/research/object_detection/object_detection/samples/configs/ssd_mobilenet_v1_fpn_shared_box_predictor_640x640_coco14_sync.config similarity index 100% rename from research/object_detection/samples/configs/ssd_mobilenet_v1_fpn_shared_box_predictor_640x640_coco14_sync.config rename to research/object_detection/object_detection/samples/configs/ssd_mobilenet_v1_fpn_shared_box_predictor_640x640_coco14_sync.config diff --git a/research/object_detection/samples/configs/ssd_mobilenet_v1_pets.config b/research/object_detection/object_detection/samples/configs/ssd_mobilenet_v1_pets.config similarity index 100% rename from research/object_detection/samples/configs/ssd_mobilenet_v1_pets.config rename to research/object_detection/object_detection/samples/configs/ssd_mobilenet_v1_pets.config diff --git a/research/object_detection/samples/configs/ssd_mobilenet_v1_ppn_shared_box_predictor_300x300_coco14_sync.config b/research/object_detection/object_detection/samples/configs/ssd_mobilenet_v1_ppn_shared_box_predictor_300x300_coco14_sync.config similarity index 100% rename from research/object_detection/samples/configs/ssd_mobilenet_v1_ppn_shared_box_predictor_300x300_coco14_sync.config rename to research/object_detection/object_detection/samples/configs/ssd_mobilenet_v1_ppn_shared_box_predictor_300x300_coco14_sync.config diff --git a/research/object_detection/samples/configs/ssd_mobilenet_v1_quantized_300x300_coco14_sync.config b/research/object_detection/object_detection/samples/configs/ssd_mobilenet_v1_quantized_300x300_coco14_sync.config similarity index 100% rename from research/object_detection/samples/configs/ssd_mobilenet_v1_quantized_300x300_coco14_sync.config rename to research/object_detection/object_detection/samples/configs/ssd_mobilenet_v1_quantized_300x300_coco14_sync.config diff --git a/research/object_detection/samples/configs/ssd_mobilenet_v2_coco.config b/research/object_detection/object_detection/samples/configs/ssd_mobilenet_v2_coco.config similarity index 100% rename from research/object_detection/samples/configs/ssd_mobilenet_v2_coco.config rename to research/object_detection/object_detection/samples/configs/ssd_mobilenet_v2_coco.config diff --git a/research/object_detection/samples/configs/ssd_resnet50_v1_fpn_shared_box_predictor_640x640_coco14_sync.config b/research/object_detection/object_detection/samples/configs/ssd_resnet50_v1_fpn_shared_box_predictor_640x640_coco14_sync.config similarity index 100% rename from research/object_detection/samples/configs/ssd_resnet50_v1_fpn_shared_box_predictor_640x640_coco14_sync.config rename to research/object_detection/object_detection/samples/configs/ssd_resnet50_v1_fpn_shared_box_predictor_640x640_coco14_sync.config diff --git a/research/object_detection/samples/configs/ssdlite_mobilenet_v1_coco.config b/research/object_detection/object_detection/samples/configs/ssdlite_mobilenet_v1_coco.config similarity index 100% rename from research/object_detection/samples/configs/ssdlite_mobilenet_v1_coco.config rename to research/object_detection/object_detection/samples/configs/ssdlite_mobilenet_v1_coco.config diff --git a/research/object_detection/samples/configs/ssdlite_mobilenet_v2_coco.config b/research/object_detection/object_detection/samples/configs/ssdlite_mobilenet_v2_coco.config similarity index 100% rename from research/object_detection/samples/configs/ssdlite_mobilenet_v2_coco.config rename to research/object_detection/object_detection/samples/configs/ssdlite_mobilenet_v2_coco.config diff --git a/research/object_detection/test_ckpt/ssd_inception_v2.pb b/research/object_detection/object_detection/test_ckpt/ssd_inception_v2.pb similarity index 100% rename from research/object_detection/test_ckpt/ssd_inception_v2.pb rename to research/object_detection/object_detection/test_ckpt/ssd_inception_v2.pb diff --git a/research/object_detection/test_data/pets_examples.record b/research/object_detection/object_detection/test_data/pets_examples.record similarity index 100% rename from research/object_detection/test_data/pets_examples.record rename to research/object_detection/object_detection/test_data/pets_examples.record diff --git a/research/object_detection/test_images/image1.jpg b/research/object_detection/object_detection/test_images/image1.jpg similarity index 100% rename from research/object_detection/test_images/image1.jpg rename to research/object_detection/object_detection/test_images/image1.jpg diff --git a/research/object_detection/test_images/image2.jpg b/research/object_detection/object_detection/test_images/image2.jpg similarity index 100% rename from research/object_detection/test_images/image2.jpg rename to research/object_detection/object_detection/test_images/image2.jpg diff --git a/research/object_detection/test_images/image_info.txt b/research/object_detection/object_detection/test_images/image_info.txt similarity index 100% rename from research/object_detection/test_images/image_info.txt rename to research/object_detection/object_detection/test_images/image_info.txt diff --git a/research/object_detection/utils/__init__.py b/research/object_detection/object_detection/utils/__init__.py similarity index 100% rename from research/object_detection/utils/__init__.py rename to research/object_detection/object_detection/utils/__init__.py diff --git a/research/object_detection/utils/category_util.py b/research/object_detection/object_detection/utils/category_util.py similarity index 100% rename from research/object_detection/utils/category_util.py rename to research/object_detection/object_detection/utils/category_util.py diff --git a/research/object_detection/utils/category_util_test.py b/research/object_detection/object_detection/utils/category_util_test.py similarity index 100% rename from research/object_detection/utils/category_util_test.py rename to research/object_detection/object_detection/utils/category_util_test.py diff --git a/research/object_detection/utils/config_util.py b/research/object_detection/object_detection/utils/config_util.py similarity index 100% rename from research/object_detection/utils/config_util.py rename to research/object_detection/object_detection/utils/config_util.py diff --git a/research/object_detection/utils/config_util_test.py b/research/object_detection/object_detection/utils/config_util_test.py similarity index 100% rename from research/object_detection/utils/config_util_test.py rename to research/object_detection/object_detection/utils/config_util_test.py diff --git a/research/object_detection/utils/context_manager.py b/research/object_detection/object_detection/utils/context_manager.py similarity index 100% rename from research/object_detection/utils/context_manager.py rename to research/object_detection/object_detection/utils/context_manager.py diff --git a/research/object_detection/utils/context_manager_test.py b/research/object_detection/object_detection/utils/context_manager_test.py similarity index 100% rename from research/object_detection/utils/context_manager_test.py rename to research/object_detection/object_detection/utils/context_manager_test.py diff --git a/research/object_detection/utils/dataset_util.py b/research/object_detection/object_detection/utils/dataset_util.py similarity index 100% rename from research/object_detection/utils/dataset_util.py rename to research/object_detection/object_detection/utils/dataset_util.py diff --git a/research/object_detection/utils/dataset_util_test.py b/research/object_detection/object_detection/utils/dataset_util_test.py similarity index 100% rename from research/object_detection/utils/dataset_util_test.py rename to research/object_detection/object_detection/utils/dataset_util_test.py diff --git a/research/object_detection/utils/json_utils.py b/research/object_detection/object_detection/utils/json_utils.py similarity index 100% rename from research/object_detection/utils/json_utils.py rename to research/object_detection/object_detection/utils/json_utils.py diff --git a/research/object_detection/utils/json_utils_test.py b/research/object_detection/object_detection/utils/json_utils_test.py similarity index 100% rename from research/object_detection/utils/json_utils_test.py rename to research/object_detection/object_detection/utils/json_utils_test.py diff --git a/research/object_detection/utils/label_map_util.py b/research/object_detection/object_detection/utils/label_map_util.py similarity index 100% rename from research/object_detection/utils/label_map_util.py rename to research/object_detection/object_detection/utils/label_map_util.py diff --git a/research/object_detection/utils/label_map_util_test.py b/research/object_detection/object_detection/utils/label_map_util_test.py similarity index 100% rename from research/object_detection/utils/label_map_util_test.py rename to research/object_detection/object_detection/utils/label_map_util_test.py diff --git a/research/object_detection/utils/learning_schedules.py b/research/object_detection/object_detection/utils/learning_schedules.py similarity index 100% rename from research/object_detection/utils/learning_schedules.py rename to research/object_detection/object_detection/utils/learning_schedules.py diff --git a/research/object_detection/utils/learning_schedules_test.py b/research/object_detection/object_detection/utils/learning_schedules_test.py similarity index 100% rename from research/object_detection/utils/learning_schedules_test.py rename to research/object_detection/object_detection/utils/learning_schedules_test.py diff --git a/research/object_detection/utils/metrics.py b/research/object_detection/object_detection/utils/metrics.py similarity index 100% rename from research/object_detection/utils/metrics.py rename to research/object_detection/object_detection/utils/metrics.py diff --git a/research/object_detection/utils/metrics_test.py b/research/object_detection/object_detection/utils/metrics_test.py similarity index 100% rename from research/object_detection/utils/metrics_test.py rename to research/object_detection/object_detection/utils/metrics_test.py diff --git a/research/object_detection/utils/np_box_list.py b/research/object_detection/object_detection/utils/np_box_list.py similarity index 100% rename from research/object_detection/utils/np_box_list.py rename to research/object_detection/object_detection/utils/np_box_list.py diff --git a/research/object_detection/utils/np_box_list_ops.py b/research/object_detection/object_detection/utils/np_box_list_ops.py similarity index 100% rename from research/object_detection/utils/np_box_list_ops.py rename to research/object_detection/object_detection/utils/np_box_list_ops.py diff --git a/research/object_detection/utils/np_box_list_ops_test.py b/research/object_detection/object_detection/utils/np_box_list_ops_test.py similarity index 100% rename from research/object_detection/utils/np_box_list_ops_test.py rename to research/object_detection/object_detection/utils/np_box_list_ops_test.py diff --git a/research/object_detection/utils/np_box_list_test.py b/research/object_detection/object_detection/utils/np_box_list_test.py similarity index 100% rename from research/object_detection/utils/np_box_list_test.py rename to research/object_detection/object_detection/utils/np_box_list_test.py diff --git a/research/object_detection/utils/np_box_mask_list.py b/research/object_detection/object_detection/utils/np_box_mask_list.py similarity index 100% rename from research/object_detection/utils/np_box_mask_list.py rename to research/object_detection/object_detection/utils/np_box_mask_list.py diff --git a/research/object_detection/utils/np_box_mask_list_ops.py b/research/object_detection/object_detection/utils/np_box_mask_list_ops.py similarity index 100% rename from research/object_detection/utils/np_box_mask_list_ops.py rename to research/object_detection/object_detection/utils/np_box_mask_list_ops.py diff --git a/research/object_detection/utils/np_box_mask_list_ops_test.py b/research/object_detection/object_detection/utils/np_box_mask_list_ops_test.py similarity index 100% rename from research/object_detection/utils/np_box_mask_list_ops_test.py rename to research/object_detection/object_detection/utils/np_box_mask_list_ops_test.py diff --git a/research/object_detection/utils/np_box_mask_list_test.py b/research/object_detection/object_detection/utils/np_box_mask_list_test.py similarity index 100% rename from research/object_detection/utils/np_box_mask_list_test.py rename to research/object_detection/object_detection/utils/np_box_mask_list_test.py diff --git a/research/object_detection/utils/np_box_ops.py b/research/object_detection/object_detection/utils/np_box_ops.py similarity index 100% rename from research/object_detection/utils/np_box_ops.py rename to research/object_detection/object_detection/utils/np_box_ops.py diff --git a/research/object_detection/utils/np_box_ops_test.py b/research/object_detection/object_detection/utils/np_box_ops_test.py similarity index 100% rename from research/object_detection/utils/np_box_ops_test.py rename to research/object_detection/object_detection/utils/np_box_ops_test.py diff --git a/research/object_detection/utils/np_mask_ops.py b/research/object_detection/object_detection/utils/np_mask_ops.py similarity index 100% rename from research/object_detection/utils/np_mask_ops.py rename to research/object_detection/object_detection/utils/np_mask_ops.py diff --git a/research/object_detection/utils/np_mask_ops_test.py b/research/object_detection/object_detection/utils/np_mask_ops_test.py similarity index 100% rename from research/object_detection/utils/np_mask_ops_test.py rename to research/object_detection/object_detection/utils/np_mask_ops_test.py diff --git a/research/object_detection/utils/object_detection_evaluation.py b/research/object_detection/object_detection/utils/object_detection_evaluation.py similarity index 100% rename from research/object_detection/utils/object_detection_evaluation.py rename to research/object_detection/object_detection/utils/object_detection_evaluation.py diff --git a/research/object_detection/utils/object_detection_evaluation_test.py b/research/object_detection/object_detection/utils/object_detection_evaluation_test.py similarity index 100% rename from research/object_detection/utils/object_detection_evaluation_test.py rename to research/object_detection/object_detection/utils/object_detection_evaluation_test.py diff --git a/research/object_detection/utils/ops.py b/research/object_detection/object_detection/utils/ops.py similarity index 100% rename from research/object_detection/utils/ops.py rename to research/object_detection/object_detection/utils/ops.py diff --git a/research/object_detection/utils/ops_test.py b/research/object_detection/object_detection/utils/ops_test.py similarity index 100% rename from research/object_detection/utils/ops_test.py rename to research/object_detection/object_detection/utils/ops_test.py diff --git a/research/object_detection/utils/per_image_evaluation.py b/research/object_detection/object_detection/utils/per_image_evaluation.py similarity index 100% rename from research/object_detection/utils/per_image_evaluation.py rename to research/object_detection/object_detection/utils/per_image_evaluation.py diff --git a/research/object_detection/utils/per_image_evaluation_test.py b/research/object_detection/object_detection/utils/per_image_evaluation_test.py similarity index 100% rename from research/object_detection/utils/per_image_evaluation_test.py rename to research/object_detection/object_detection/utils/per_image_evaluation_test.py diff --git a/research/object_detection/utils/per_image_vrd_evaluation.py b/research/object_detection/object_detection/utils/per_image_vrd_evaluation.py similarity index 100% rename from research/object_detection/utils/per_image_vrd_evaluation.py rename to research/object_detection/object_detection/utils/per_image_vrd_evaluation.py diff --git a/research/object_detection/utils/per_image_vrd_evaluation_test.py b/research/object_detection/object_detection/utils/per_image_vrd_evaluation_test.py similarity index 100% rename from research/object_detection/utils/per_image_vrd_evaluation_test.py rename to research/object_detection/object_detection/utils/per_image_vrd_evaluation_test.py diff --git a/research/object_detection/utils/shape_utils.py b/research/object_detection/object_detection/utils/shape_utils.py similarity index 100% rename from research/object_detection/utils/shape_utils.py rename to research/object_detection/object_detection/utils/shape_utils.py diff --git a/research/object_detection/utils/shape_utils_test.py b/research/object_detection/object_detection/utils/shape_utils_test.py similarity index 100% rename from research/object_detection/utils/shape_utils_test.py rename to research/object_detection/object_detection/utils/shape_utils_test.py diff --git a/research/object_detection/utils/static_shape.py b/research/object_detection/object_detection/utils/static_shape.py similarity index 100% rename from research/object_detection/utils/static_shape.py rename to research/object_detection/object_detection/utils/static_shape.py diff --git a/research/object_detection/utils/static_shape_test.py b/research/object_detection/object_detection/utils/static_shape_test.py similarity index 100% rename from research/object_detection/utils/static_shape_test.py rename to research/object_detection/object_detection/utils/static_shape_test.py diff --git a/research/object_detection/utils/test_case.py b/research/object_detection/object_detection/utils/test_case.py similarity index 100% rename from research/object_detection/utils/test_case.py rename to research/object_detection/object_detection/utils/test_case.py diff --git a/research/object_detection/utils/test_utils.py b/research/object_detection/object_detection/utils/test_utils.py similarity index 100% rename from research/object_detection/utils/test_utils.py rename to research/object_detection/object_detection/utils/test_utils.py diff --git a/research/object_detection/utils/test_utils_test.py b/research/object_detection/object_detection/utils/test_utils_test.py similarity index 100% rename from research/object_detection/utils/test_utils_test.py rename to research/object_detection/object_detection/utils/test_utils_test.py diff --git a/research/object_detection/utils/variables_helper.py b/research/object_detection/object_detection/utils/variables_helper.py similarity index 100% rename from research/object_detection/utils/variables_helper.py rename to research/object_detection/object_detection/utils/variables_helper.py diff --git a/research/object_detection/utils/variables_helper_test.py b/research/object_detection/object_detection/utils/variables_helper_test.py similarity index 100% rename from research/object_detection/utils/variables_helper_test.py rename to research/object_detection/object_detection/utils/variables_helper_test.py diff --git a/research/object_detection/utils/visualization_utils.py b/research/object_detection/object_detection/utils/visualization_utils.py similarity index 100% rename from research/object_detection/utils/visualization_utils.py rename to research/object_detection/object_detection/utils/visualization_utils.py diff --git a/research/object_detection/utils/visualization_utils_test.py b/research/object_detection/object_detection/utils/visualization_utils_test.py similarity index 100% rename from research/object_detection/utils/visualization_utils_test.py rename to research/object_detection/object_detection/utils/visualization_utils_test.py diff --git a/research/object_detection/utils/vrd_evaluation.py b/research/object_detection/object_detection/utils/vrd_evaluation.py similarity index 100% rename from research/object_detection/utils/vrd_evaluation.py rename to research/object_detection/object_detection/utils/vrd_evaluation.py diff --git a/research/object_detection/utils/vrd_evaluation_test.py b/research/object_detection/object_detection/utils/vrd_evaluation_test.py similarity index 100% rename from research/object_detection/utils/vrd_evaluation_test.py rename to research/object_detection/object_detection/utils/vrd_evaluation_test.py diff --git a/research/object_detection/setup.py b/research/object_detection/setup.py new file mode 100644 index 00000000000..b473b37dcf9 --- /dev/null +++ b/research/object_detection/setup.py @@ -0,0 +1,67 @@ +"""Setup script for object_detection.""" + +import sys +import os +import subprocess +import shutil + +from setuptools import find_packages +from setuptools import setup +from setuptools.command.install import install +from setuptools.command.develop import develop +from setuptools.command.egg_info import egg_info + +REQUIRED_PACKAGES = ['Pillow>=1.0', 'Matplotlib>=2.1', 'Cython>=0.28.1'] +packages = [p for p in find_packages() if p.startswith('object_detection')] + + +def protobuf_compiler(): + print("Compiling Protobuf files...") + + protoc_binary = shutil.which("protoc") + pb_dir_path = os.path.join("object_detection", "protos") + + for pb_filename in os.listdir(pb_dir_path): + if pb_filename.endswith(".proto"): + + protoc_command = [protoc_binary, os.path.join(pb_dir_path, pb_filename), "--python_out=."] + if subprocess.call(protoc_command) != 0: + print("**ERROR during Protobuf files compilation: {}**".format(pb_filename)) + sys.exit(1) + + print("Protobuf files Compilation Done.") + + +class CustomInstallCommand(install): + def run(self): + protobuf_compiler() + install.run(self) + + +class CustomDevelopCommand(develop): + def run(self): + protobuf_compiler() + develop.run(self) + + +class CustomEggInfoCommand(egg_info): + def run(self): + protobuf_compiler() + egg_info.run(self) + + +setup(name='object_detection', + version='0.1', + install_requires=REQUIRED_PACKAGES, + include_package_data=True, + packages=packages, + cmdclass={"install": CustomInstallCommand, + "develop": CustomDevelopCommand, + "egg_info": CustomEggInfoCommand}, + description='Tensorflow Object Detection Library', + entry_points={ + 'console_scripts': [ + 'model_builder_test = object_detection.builders.model_builder_test:tf.test.main', + ], + } + ) diff --git a/research/slim/__init__.py b/research/slim/slim/__init__.py similarity index 100% rename from research/slim/__init__.py rename to research/slim/slim/__init__.py diff --git a/research/slim/datasets/__init__.py b/research/slim/slim/datasets/__init__.py similarity index 100% rename from research/slim/datasets/__init__.py rename to research/slim/slim/datasets/__init__.py diff --git a/research/slim/datasets/build_imagenet_data.py b/research/slim/slim/datasets/build_imagenet_data.py similarity index 100% rename from research/slim/datasets/build_imagenet_data.py rename to research/slim/slim/datasets/build_imagenet_data.py diff --git a/research/slim/datasets/cifar10.py b/research/slim/slim/datasets/cifar10.py similarity index 100% rename from research/slim/datasets/cifar10.py rename to research/slim/slim/datasets/cifar10.py diff --git a/research/slim/datasets/dataset_factory.py b/research/slim/slim/datasets/dataset_factory.py similarity index 100% rename from research/slim/datasets/dataset_factory.py rename to research/slim/slim/datasets/dataset_factory.py diff --git a/research/slim/datasets/dataset_utils.py b/research/slim/slim/datasets/dataset_utils.py similarity index 100% rename from research/slim/datasets/dataset_utils.py rename to research/slim/slim/datasets/dataset_utils.py diff --git a/research/slim/datasets/download_and_convert_cifar10.py b/research/slim/slim/datasets/download_and_convert_cifar10.py similarity index 100% rename from research/slim/datasets/download_and_convert_cifar10.py rename to research/slim/slim/datasets/download_and_convert_cifar10.py diff --git a/research/slim/datasets/download_and_convert_flowers.py b/research/slim/slim/datasets/download_and_convert_flowers.py similarity index 100% rename from research/slim/datasets/download_and_convert_flowers.py rename to research/slim/slim/datasets/download_and_convert_flowers.py diff --git a/research/slim/datasets/download_and_convert_imagenet.sh b/research/slim/slim/datasets/download_and_convert_imagenet.sh similarity index 100% rename from research/slim/datasets/download_and_convert_imagenet.sh rename to research/slim/slim/datasets/download_and_convert_imagenet.sh diff --git a/research/slim/datasets/download_and_convert_mnist.py b/research/slim/slim/datasets/download_and_convert_mnist.py similarity index 100% rename from research/slim/datasets/download_and_convert_mnist.py rename to research/slim/slim/datasets/download_and_convert_mnist.py diff --git a/research/slim/datasets/download_imagenet.sh b/research/slim/slim/datasets/download_imagenet.sh similarity index 100% rename from research/slim/datasets/download_imagenet.sh rename to research/slim/slim/datasets/download_imagenet.sh diff --git a/research/slim/datasets/flowers.py b/research/slim/slim/datasets/flowers.py similarity index 100% rename from research/slim/datasets/flowers.py rename to research/slim/slim/datasets/flowers.py diff --git a/research/slim/datasets/imagenet.py b/research/slim/slim/datasets/imagenet.py similarity index 100% rename from research/slim/datasets/imagenet.py rename to research/slim/slim/datasets/imagenet.py diff --git a/research/slim/datasets/imagenet_2012_validation_synset_labels.txt b/research/slim/slim/datasets/imagenet_2012_validation_synset_labels.txt similarity index 100% rename from research/slim/datasets/imagenet_2012_validation_synset_labels.txt rename to research/slim/slim/datasets/imagenet_2012_validation_synset_labels.txt diff --git a/research/slim/datasets/imagenet_lsvrc_2015_synsets.txt b/research/slim/slim/datasets/imagenet_lsvrc_2015_synsets.txt similarity index 100% rename from research/slim/datasets/imagenet_lsvrc_2015_synsets.txt rename to research/slim/slim/datasets/imagenet_lsvrc_2015_synsets.txt diff --git a/research/slim/datasets/imagenet_metadata.txt b/research/slim/slim/datasets/imagenet_metadata.txt similarity index 100% rename from research/slim/datasets/imagenet_metadata.txt rename to research/slim/slim/datasets/imagenet_metadata.txt diff --git a/research/slim/datasets/mnist.py b/research/slim/slim/datasets/mnist.py similarity index 100% rename from research/slim/datasets/mnist.py rename to research/slim/slim/datasets/mnist.py diff --git a/research/slim/datasets/preprocess_imagenet_validation_data.py b/research/slim/slim/datasets/preprocess_imagenet_validation_data.py similarity index 100% rename from research/slim/datasets/preprocess_imagenet_validation_data.py rename to research/slim/slim/datasets/preprocess_imagenet_validation_data.py diff --git a/research/slim/datasets/process_bounding_boxes.py b/research/slim/slim/datasets/process_bounding_boxes.py similarity index 100% rename from research/slim/datasets/process_bounding_boxes.py rename to research/slim/slim/datasets/process_bounding_boxes.py diff --git a/research/slim/deployment/__init__.py b/research/slim/slim/deployment/__init__.py similarity index 100% rename from research/slim/deployment/__init__.py rename to research/slim/slim/deployment/__init__.py diff --git a/research/slim/deployment/model_deploy.py b/research/slim/slim/deployment/model_deploy.py similarity index 100% rename from research/slim/deployment/model_deploy.py rename to research/slim/slim/deployment/model_deploy.py diff --git a/research/slim/deployment/model_deploy_test.py b/research/slim/slim/deployment/model_deploy_test.py similarity index 100% rename from research/slim/deployment/model_deploy_test.py rename to research/slim/slim/deployment/model_deploy_test.py diff --git a/research/slim/nets/__init__.py b/research/slim/slim/nets/__init__.py similarity index 100% rename from research/slim/nets/__init__.py rename to research/slim/slim/nets/__init__.py diff --git a/research/slim/nets/alexnet.py b/research/slim/slim/nets/alexnet.py similarity index 100% rename from research/slim/nets/alexnet.py rename to research/slim/slim/nets/alexnet.py diff --git a/research/slim/nets/alexnet_test.py b/research/slim/slim/nets/alexnet_test.py similarity index 99% rename from research/slim/nets/alexnet_test.py rename to research/slim/slim/nets/alexnet_test.py index 6f85d33f980..4b7374f3210 100644 --- a/research/slim/nets/alexnet_test.py +++ b/research/slim/slim/nets/alexnet_test.py @@ -19,7 +19,7 @@ import tensorflow as tf -from nets import alexnet +from slim.nets import alexnet slim = tf.contrib.slim diff --git a/research/slim/nets/cifarnet.py b/research/slim/slim/nets/cifarnet.py similarity index 100% rename from research/slim/nets/cifarnet.py rename to research/slim/slim/nets/cifarnet.py diff --git a/research/slim/nets/cyclegan.py b/research/slim/slim/nets/cyclegan.py similarity index 100% rename from research/slim/nets/cyclegan.py rename to research/slim/slim/nets/cyclegan.py diff --git a/research/slim/nets/cyclegan_test.py b/research/slim/slim/nets/cyclegan_test.py similarity index 99% rename from research/slim/nets/cyclegan_test.py rename to research/slim/slim/nets/cyclegan_test.py index 395773eaec9..696098e8131 100644 --- a/research/slim/nets/cyclegan_test.py +++ b/research/slim/slim/nets/cyclegan_test.py @@ -20,7 +20,7 @@ import tensorflow as tf -from nets import cyclegan +from slim.nets import cyclegan # TODO(joelshor): Add a test to check generator endpoints. diff --git a/research/slim/nets/dcgan.py b/research/slim/slim/nets/dcgan.py similarity index 100% rename from research/slim/nets/dcgan.py rename to research/slim/slim/nets/dcgan.py diff --git a/research/slim/nets/dcgan_test.py b/research/slim/slim/nets/dcgan_test.py similarity index 99% rename from research/slim/nets/dcgan_test.py rename to research/slim/slim/nets/dcgan_test.py index 343de628f04..547d66cd0dd 100644 --- a/research/slim/nets/dcgan_test.py +++ b/research/slim/slim/nets/dcgan_test.py @@ -21,7 +21,7 @@ from six.moves import xrange # pylint: disable=redefined-builtin import tensorflow as tf -from nets import dcgan +from slim.nets import dcgan class DCGANTest(tf.test.TestCase): diff --git a/research/slim/nets/inception.py b/research/slim/slim/nets/inception.py similarity index 52% rename from research/slim/nets/inception.py rename to research/slim/slim/nets/inception.py index b69cd2aacbe..e2f12e2b3a4 100644 --- a/research/slim/nets/inception.py +++ b/research/slim/slim/nets/inception.py @@ -19,19 +19,19 @@ from __future__ import print_function # pylint: disable=unused-import -from nets.inception_resnet_v2 import inception_resnet_v2 -from nets.inception_resnet_v2 import inception_resnet_v2_arg_scope -from nets.inception_resnet_v2 import inception_resnet_v2_base -from nets.inception_v1 import inception_v1 -from nets.inception_v1 import inception_v1_arg_scope -from nets.inception_v1 import inception_v1_base -from nets.inception_v2 import inception_v2 -from nets.inception_v2 import inception_v2_arg_scope -from nets.inception_v2 import inception_v2_base -from nets.inception_v3 import inception_v3 -from nets.inception_v3 import inception_v3_arg_scope -from nets.inception_v3 import inception_v3_base -from nets.inception_v4 import inception_v4 -from nets.inception_v4 import inception_v4_arg_scope -from nets.inception_v4 import inception_v4_base +from slim.nets.inception_resnet_v2 import inception_resnet_v2 +from slim.nets.inception_resnet_v2 import inception_resnet_v2_arg_scope +from slim.nets.inception_resnet_v2 import inception_resnet_v2_base +from slim.nets.inception_v1 import inception_v1 +from slim.nets.inception_v1 import inception_v1_arg_scope +from slim.nets.inception_v1 import inception_v1_base +from slim.nets.inception_v2 import inception_v2 +from slim.nets.inception_v2 import inception_v2_arg_scope +from slim.nets.inception_v2 import inception_v2_base +from slim.nets.inception_v3 import inception_v3 +from slim.nets.inception_v3 import inception_v3_arg_scope +from slim.nets.inception_v3 import inception_v3_base +from slim.nets.inception_v4 import inception_v4 +from slim.nets.inception_v4 import inception_v4_arg_scope +from slim.nets.inception_v4 import inception_v4_base # pylint: enable=unused-import diff --git a/research/slim/nets/inception_resnet_v2.py b/research/slim/slim/nets/inception_resnet_v2.py similarity index 100% rename from research/slim/nets/inception_resnet_v2.py rename to research/slim/slim/nets/inception_resnet_v2.py diff --git a/research/slim/nets/inception_resnet_v2_test.py b/research/slim/slim/nets/inception_resnet_v2_test.py similarity index 99% rename from research/slim/nets/inception_resnet_v2_test.py rename to research/slim/slim/nets/inception_resnet_v2_test.py index 9914de9ba87..65849f8ea11 100644 --- a/research/slim/nets/inception_resnet_v2_test.py +++ b/research/slim/slim/nets/inception_resnet_v2_test.py @@ -19,7 +19,7 @@ import tensorflow as tf -from nets import inception +from slim.nets import inception class InceptionTest(tf.test.TestCase): diff --git a/research/slim/nets/inception_utils.py b/research/slim/slim/nets/inception_utils.py similarity index 100% rename from research/slim/nets/inception_utils.py rename to research/slim/slim/nets/inception_utils.py diff --git a/research/slim/nets/inception_v1.py b/research/slim/slim/nets/inception_v1.py similarity index 99% rename from research/slim/nets/inception_v1.py rename to research/slim/slim/nets/inception_v1.py index d98716597cf..97e906f79ad 100644 --- a/research/slim/nets/inception_v1.py +++ b/research/slim/slim/nets/inception_v1.py @@ -20,7 +20,7 @@ import tensorflow as tf -from nets import inception_utils +from slim.nets import inception_utils slim = tf.contrib.slim trunc_normal = lambda stddev: tf.truncated_normal_initializer(0.0, stddev) diff --git a/research/slim/nets/inception_v1_test.py b/research/slim/slim/nets/inception_v1_test.py similarity index 99% rename from research/slim/nets/inception_v1_test.py rename to research/slim/slim/nets/inception_v1_test.py index 302eafa90e8..17578cb6ab0 100644 --- a/research/slim/nets/inception_v1_test.py +++ b/research/slim/slim/nets/inception_v1_test.py @@ -21,7 +21,7 @@ import numpy as np import tensorflow as tf -from nets import inception +from slim.nets import inception slim = tf.contrib.slim diff --git a/research/slim/nets/inception_v2.py b/research/slim/slim/nets/inception_v2.py similarity index 99% rename from research/slim/nets/inception_v2.py rename to research/slim/slim/nets/inception_v2.py index 66290b4db95..13a6b402274 100644 --- a/research/slim/nets/inception_v2.py +++ b/research/slim/slim/nets/inception_v2.py @@ -20,7 +20,7 @@ import tensorflow as tf -from nets import inception_utils +from slim.nets import inception_utils slim = tf.contrib.slim trunc_normal = lambda stddev: tf.truncated_normal_initializer(0.0, stddev) diff --git a/research/slim/nets/inception_v2_test.py b/research/slim/slim/nets/inception_v2_test.py similarity index 99% rename from research/slim/nets/inception_v2_test.py rename to research/slim/slim/nets/inception_v2_test.py index e6b58a118a6..38c33d1f487 100644 --- a/research/slim/nets/inception_v2_test.py +++ b/research/slim/slim/nets/inception_v2_test.py @@ -21,7 +21,7 @@ import numpy as np import tensorflow as tf -from nets import inception +from slim.nets import inception slim = tf.contrib.slim diff --git a/research/slim/nets/inception_v3.py b/research/slim/slim/nets/inception_v3.py similarity index 99% rename from research/slim/nets/inception_v3.py rename to research/slim/slim/nets/inception_v3.py index 1221779168b..785d57c965e 100644 --- a/research/slim/nets/inception_v3.py +++ b/research/slim/slim/nets/inception_v3.py @@ -20,7 +20,7 @@ import tensorflow as tf -from nets import inception_utils +from slim.nets import inception_utils slim = tf.contrib.slim trunc_normal = lambda stddev: tf.truncated_normal_initializer(0.0, stddev) diff --git a/research/slim/nets/inception_v3_test.py b/research/slim/slim/nets/inception_v3_test.py similarity index 99% rename from research/slim/nets/inception_v3_test.py rename to research/slim/slim/nets/inception_v3_test.py index 32c379f168b..f831e326bc8 100644 --- a/research/slim/nets/inception_v3_test.py +++ b/research/slim/slim/nets/inception_v3_test.py @@ -21,7 +21,7 @@ import numpy as np import tensorflow as tf -from nets import inception +from slim.nets import inception slim = tf.contrib.slim diff --git a/research/slim/nets/inception_v4.py b/research/slim/slim/nets/inception_v4.py similarity index 99% rename from research/slim/nets/inception_v4.py rename to research/slim/slim/nets/inception_v4.py index bab406a8cf2..664fb600f81 100644 --- a/research/slim/nets/inception_v4.py +++ b/research/slim/slim/nets/inception_v4.py @@ -26,7 +26,7 @@ import tensorflow as tf -from nets import inception_utils +from slim.nets import inception_utils slim = tf.contrib.slim diff --git a/research/slim/nets/inception_v4_test.py b/research/slim/slim/nets/inception_v4_test.py similarity index 99% rename from research/slim/nets/inception_v4_test.py rename to research/slim/slim/nets/inception_v4_test.py index 0c3abf0e593..e378021cabb 100644 --- a/research/slim/nets/inception_v4_test.py +++ b/research/slim/slim/nets/inception_v4_test.py @@ -19,7 +19,7 @@ import tensorflow as tf -from nets import inception +from slim.nets import inception class InceptionTest(tf.test.TestCase): diff --git a/research/slim/nets/lenet.py b/research/slim/slim/nets/lenet.py similarity index 100% rename from research/slim/nets/lenet.py rename to research/slim/slim/nets/lenet.py diff --git a/research/slim/nets/mobilenet/README.md b/research/slim/slim/nets/mobilenet/README.md similarity index 100% rename from research/slim/nets/mobilenet/README.md rename to research/slim/slim/nets/mobilenet/README.md diff --git a/research/slim/nets/mobilenet/__init__.py b/research/slim/slim/nets/mobilenet/__init__.py similarity index 100% rename from research/slim/nets/mobilenet/__init__.py rename to research/slim/slim/nets/mobilenet/__init__.py diff --git a/research/slim/nets/mobilenet/conv_blocks.py b/research/slim/slim/nets/mobilenet/conv_blocks.py similarity index 100% rename from research/slim/nets/mobilenet/conv_blocks.py rename to research/slim/slim/nets/mobilenet/conv_blocks.py diff --git a/research/slim/nets/mobilenet/madds_top1_accuracy.png b/research/slim/slim/nets/mobilenet/madds_top1_accuracy.png similarity index 100% rename from research/slim/nets/mobilenet/madds_top1_accuracy.png rename to research/slim/slim/nets/mobilenet/madds_top1_accuracy.png diff --git a/research/slim/nets/mobilenet/mnet_v1_vs_v2_pixel1_latency.png b/research/slim/slim/nets/mobilenet/mnet_v1_vs_v2_pixel1_latency.png similarity index 100% rename from research/slim/nets/mobilenet/mnet_v1_vs_v2_pixel1_latency.png rename to research/slim/slim/nets/mobilenet/mnet_v1_vs_v2_pixel1_latency.png diff --git a/research/slim/nets/mobilenet/mobilenet.py b/research/slim/slim/nets/mobilenet/mobilenet.py similarity index 100% rename from research/slim/nets/mobilenet/mobilenet.py rename to research/slim/slim/nets/mobilenet/mobilenet.py diff --git a/research/slim/nets/mobilenet/mobilenet_example.ipynb b/research/slim/slim/nets/mobilenet/mobilenet_example.ipynb similarity index 100% rename from research/slim/nets/mobilenet/mobilenet_example.ipynb rename to research/slim/slim/nets/mobilenet/mobilenet_example.ipynb diff --git a/research/slim/nets/mobilenet/mobilenet_v2.py b/research/slim/slim/nets/mobilenet/mobilenet_v2.py similarity index 98% rename from research/slim/nets/mobilenet/mobilenet_v2.py rename to research/slim/slim/nets/mobilenet/mobilenet_v2.py index 0dd286b7934..f08c59da808 100644 --- a/research/slim/nets/mobilenet/mobilenet_v2.py +++ b/research/slim/slim/nets/mobilenet/mobilenet_v2.py @@ -28,8 +28,8 @@ import tensorflow as tf -from nets.mobilenet import conv_blocks as ops -from nets.mobilenet import mobilenet as lib +from slim.nets.mobilenet import conv_blocks as ops +from slim.nets.mobilenet import mobilenet as lib slim = tf.contrib.slim op = lib.op diff --git a/research/slim/nets/mobilenet/mobilenet_v2_test.py b/research/slim/slim/nets/mobilenet/mobilenet_v2_test.py similarity index 98% rename from research/slim/nets/mobilenet/mobilenet_v2_test.py rename to research/slim/slim/nets/mobilenet/mobilenet_v2_test.py index 7ce1993cbdc..e719ae969ea 100644 --- a/research/slim/nets/mobilenet/mobilenet_v2_test.py +++ b/research/slim/slim/nets/mobilenet/mobilenet_v2_test.py @@ -19,9 +19,9 @@ from __future__ import print_function import copy import tensorflow as tf -from nets.mobilenet import conv_blocks as ops -from nets.mobilenet import mobilenet -from nets.mobilenet import mobilenet_v2 +from slim.nets.mobilenet import conv_blocks as ops +from slim.nets.mobilenet import mobilenet +from slim.nets.mobilenet import mobilenet_v2 slim = tf.contrib.slim diff --git a/research/slim/nets/mobilenet_v1.md b/research/slim/slim/nets/mobilenet_v1.md similarity index 100% rename from research/slim/nets/mobilenet_v1.md rename to research/slim/slim/nets/mobilenet_v1.md diff --git a/research/slim/nets/mobilenet_v1.png b/research/slim/slim/nets/mobilenet_v1.png similarity index 100% rename from research/slim/nets/mobilenet_v1.png rename to research/slim/slim/nets/mobilenet_v1.png diff --git a/research/slim/nets/mobilenet_v1.py b/research/slim/slim/nets/mobilenet_v1.py similarity index 100% rename from research/slim/nets/mobilenet_v1.py rename to research/slim/slim/nets/mobilenet_v1.py diff --git a/research/slim/nets/mobilenet_v1_eval.py b/research/slim/slim/nets/mobilenet_v1_eval.py similarity index 99% rename from research/slim/nets/mobilenet_v1_eval.py rename to research/slim/slim/nets/mobilenet_v1_eval.py index 5b42a382042..e65d8e65f16 100644 --- a/research/slim/nets/mobilenet_v1_eval.py +++ b/research/slim/slim/nets/mobilenet_v1_eval.py @@ -22,7 +22,7 @@ import tensorflow as tf from datasets import dataset_factory -from nets import mobilenet_v1 +from slim.nets import mobilenet_v1 from preprocessing import preprocessing_factory slim = tf.contrib.slim diff --git a/research/slim/nets/mobilenet_v1_test.py b/research/slim/slim/nets/mobilenet_v1_test.py similarity index 99% rename from research/slim/nets/mobilenet_v1_test.py rename to research/slim/slim/nets/mobilenet_v1_test.py index 669a3ae3f74..e1ec9edb62a 100644 --- a/research/slim/nets/mobilenet_v1_test.py +++ b/research/slim/slim/nets/mobilenet_v1_test.py @@ -21,7 +21,7 @@ import numpy as np import tensorflow as tf -from nets import mobilenet_v1 +from slim.nets import mobilenet_v1 slim = tf.contrib.slim diff --git a/research/slim/nets/mobilenet_v1_train.py b/research/slim/slim/nets/mobilenet_v1_train.py similarity index 99% rename from research/slim/nets/mobilenet_v1_train.py rename to research/slim/slim/nets/mobilenet_v1_train.py index f8328aa36db..77d84a8cdf2 100644 --- a/research/slim/nets/mobilenet_v1_train.py +++ b/research/slim/slim/nets/mobilenet_v1_train.py @@ -21,7 +21,7 @@ import tensorflow as tf from datasets import dataset_factory -from nets import mobilenet_v1 +from slim.nets import mobilenet_v1 from preprocessing import preprocessing_factory slim = tf.contrib.slim diff --git a/research/slim/nets/nasnet/README.md b/research/slim/slim/nets/nasnet/README.md similarity index 100% rename from research/slim/nets/nasnet/README.md rename to research/slim/slim/nets/nasnet/README.md diff --git a/research/slim/nets/nasnet/__init__.py b/research/slim/slim/nets/nasnet/__init__.py similarity index 100% rename from research/slim/nets/nasnet/__init__.py rename to research/slim/slim/nets/nasnet/__init__.py diff --git a/research/slim/nets/nasnet/nasnet.py b/research/slim/slim/nets/nasnet/nasnet.py similarity index 99% rename from research/slim/nets/nasnet/nasnet.py rename to research/slim/slim/nets/nasnet/nasnet.py index c188c4aa458..4293eff4786 100644 --- a/research/slim/nets/nasnet/nasnet.py +++ b/research/slim/slim/nets/nasnet/nasnet.py @@ -23,7 +23,7 @@ import copy import tensorflow as tf -from nets.nasnet import nasnet_utils +from slim.nets.nasnet import nasnet_utils arg_scope = tf.contrib.framework.arg_scope slim = tf.contrib.slim diff --git a/research/slim/nets/nasnet/nasnet_test.py b/research/slim/slim/nets/nasnet/nasnet_test.py similarity index 100% rename from research/slim/nets/nasnet/nasnet_test.py rename to research/slim/slim/nets/nasnet/nasnet_test.py diff --git a/research/slim/nets/nasnet/nasnet_utils.py b/research/slim/slim/nets/nasnet/nasnet_utils.py similarity index 100% rename from research/slim/nets/nasnet/nasnet_utils.py rename to research/slim/slim/nets/nasnet/nasnet_utils.py diff --git a/research/slim/nets/nasnet/nasnet_utils_test.py b/research/slim/slim/nets/nasnet/nasnet_utils_test.py similarity index 100% rename from research/slim/nets/nasnet/nasnet_utils_test.py rename to research/slim/slim/nets/nasnet/nasnet_utils_test.py diff --git a/research/slim/nets/nasnet/pnasnet.py b/research/slim/slim/nets/nasnet/pnasnet.py similarity index 98% rename from research/slim/nets/nasnet/pnasnet.py rename to research/slim/slim/nets/nasnet/pnasnet.py index ab71d8660ec..0d74ec9ccda 100644 --- a/research/slim/nets/nasnet/pnasnet.py +++ b/research/slim/slim/nets/nasnet/pnasnet.py @@ -24,8 +24,8 @@ import copy import tensorflow as tf -from nets.nasnet import nasnet -from nets.nasnet import nasnet_utils +from slim.nets.nasnet import nasnet +from slim.nets.nasnet import nasnet_utils arg_scope = tf.contrib.framework.arg_scope slim = tf.contrib.slim diff --git a/research/slim/nets/nasnet/pnasnet_test.py b/research/slim/slim/nets/nasnet/pnasnet_test.py similarity index 100% rename from research/slim/nets/nasnet/pnasnet_test.py rename to research/slim/slim/nets/nasnet/pnasnet_test.py diff --git a/research/slim/nets/nets_factory.py b/research/slim/slim/nets/nets_factory.py similarity index 94% rename from research/slim/nets/nets_factory.py rename to research/slim/slim/nets/nets_factory.py index 99f3d3462e2..6fc267ef56d 100644 --- a/research/slim/nets/nets_factory.py +++ b/research/slim/slim/nets/nets_factory.py @@ -21,18 +21,18 @@ import tensorflow as tf -from nets import alexnet -from nets import cifarnet -from nets import inception -from nets import lenet -from nets import mobilenet_v1 -from nets import overfeat -from nets import resnet_v1 -from nets import resnet_v2 -from nets import vgg -from nets.mobilenet import mobilenet_v2 -from nets.nasnet import nasnet -from nets.nasnet import pnasnet +from slim.nets import alexnet +from slim.nets import cifarnet +from slim.nets import inception +from slim.nets import lenet +from slim.nets import mobilenet_v1 +from slim.nets import overfeat +from slim.nets import resnet_v1 +from slim.nets import resnet_v2 +from slim.nets import vgg +from slim.nets.mobilenet import mobilenet_v2 +from slim.nets.nasnet import nasnet +from slim.nets.nasnet import pnasnet slim = tf.contrib.slim diff --git a/research/slim/nets/nets_factory_test.py b/research/slim/slim/nets/nets_factory_test.py similarity index 98% rename from research/slim/nets/nets_factory_test.py rename to research/slim/slim/nets/nets_factory_test.py index 2bf25efadf3..689777670cc 100644 --- a/research/slim/nets/nets_factory_test.py +++ b/research/slim/slim/nets/nets_factory_test.py @@ -22,7 +22,7 @@ import tensorflow as tf -from nets import nets_factory +from slim.nets import nets_factory class NetworksTest(tf.test.TestCase): diff --git a/research/slim/nets/overfeat.py b/research/slim/slim/nets/overfeat.py similarity index 100% rename from research/slim/nets/overfeat.py rename to research/slim/slim/nets/overfeat.py diff --git a/research/slim/nets/overfeat_test.py b/research/slim/slim/nets/overfeat_test.py similarity index 99% rename from research/slim/nets/overfeat_test.py rename to research/slim/slim/nets/overfeat_test.py index dab0039eec6..ff7e8f4b72d 100644 --- a/research/slim/nets/overfeat_test.py +++ b/research/slim/slim/nets/overfeat_test.py @@ -19,7 +19,7 @@ import tensorflow as tf -from nets import overfeat +from slim.nets import overfeat slim = tf.contrib.slim diff --git a/research/slim/nets/pix2pix.py b/research/slim/slim/nets/pix2pix.py similarity index 100% rename from research/slim/nets/pix2pix.py rename to research/slim/slim/nets/pix2pix.py diff --git a/research/slim/nets/pix2pix_test.py b/research/slim/slim/nets/pix2pix_test.py similarity index 99% rename from research/slim/nets/pix2pix_test.py rename to research/slim/slim/nets/pix2pix_test.py index ab5acb5c197..caada745742 100644 --- a/research/slim/nets/pix2pix_test.py +++ b/research/slim/slim/nets/pix2pix_test.py @@ -19,7 +19,7 @@ from __future__ import print_function import tensorflow as tf -from nets import pix2pix +from slim.nets import pix2pix class GeneratorTest(tf.test.TestCase): diff --git a/research/slim/nets/resnet_utils.py b/research/slim/slim/nets/resnet_utils.py similarity index 100% rename from research/slim/nets/resnet_utils.py rename to research/slim/slim/nets/resnet_utils.py diff --git a/research/slim/nets/resnet_v1.py b/research/slim/slim/nets/resnet_v1.py similarity index 99% rename from research/slim/nets/resnet_v1.py rename to research/slim/slim/nets/resnet_v1.py index 95e1a11c298..93ad5966057 100644 --- a/research/slim/nets/resnet_v1.py +++ b/research/slim/slim/nets/resnet_v1.py @@ -58,7 +58,7 @@ import tensorflow as tf -from nets import resnet_utils +from slim.nets import resnet_utils resnet_arg_scope = resnet_utils.resnet_arg_scope diff --git a/research/slim/nets/resnet_v1_test.py b/research/slim/slim/nets/resnet_v1_test.py similarity index 99% rename from research/slim/nets/resnet_v1_test.py rename to research/slim/slim/nets/resnet_v1_test.py index c40e7f881aa..626c03fc088 100644 --- a/research/slim/nets/resnet_v1_test.py +++ b/research/slim/slim/nets/resnet_v1_test.py @@ -21,8 +21,8 @@ import numpy as np import tensorflow as tf -from nets import resnet_utils -from nets import resnet_v1 +from slim.nets import resnet_utils +from slim.nets import resnet_v1 slim = tf.contrib.slim diff --git a/research/slim/nets/resnet_v2.py b/research/slim/slim/nets/resnet_v2.py similarity index 99% rename from research/slim/nets/resnet_v2.py rename to research/slim/slim/nets/resnet_v2.py index c719c1bd9cd..1f40b04b31f 100644 --- a/research/slim/nets/resnet_v2.py +++ b/research/slim/slim/nets/resnet_v2.py @@ -52,7 +52,7 @@ import tensorflow as tf -from nets import resnet_utils +from slim.nets import resnet_utils slim = tf.contrib.slim resnet_arg_scope = resnet_utils.resnet_arg_scope diff --git a/research/slim/nets/resnet_v2_test.py b/research/slim/slim/nets/resnet_v2_test.py similarity index 99% rename from research/slim/nets/resnet_v2_test.py rename to research/slim/slim/nets/resnet_v2_test.py index 891816d1320..f68bc9e9fdd 100644 --- a/research/slim/nets/resnet_v2_test.py +++ b/research/slim/slim/nets/resnet_v2_test.py @@ -21,8 +21,8 @@ import numpy as np import tensorflow as tf -from nets import resnet_utils -from nets import resnet_v2 +from slim.nets import resnet_utils +from slim.nets import resnet_v2 slim = tf.contrib.slim diff --git a/research/slim/nets/vgg.py b/research/slim/slim/nets/vgg.py similarity index 100% rename from research/slim/nets/vgg.py rename to research/slim/slim/nets/vgg.py diff --git a/research/slim/nets/vgg_test.py b/research/slim/slim/nets/vgg_test.py similarity index 99% rename from research/slim/nets/vgg_test.py rename to research/slim/slim/nets/vgg_test.py index 6760c36857f..1cd949fb19b 100644 --- a/research/slim/nets/vgg_test.py +++ b/research/slim/slim/nets/vgg_test.py @@ -19,7 +19,7 @@ import tensorflow as tf -from nets import vgg +from slim.nets import vgg slim = tf.contrib.slim diff --git a/research/slim/preprocessing/__init__.py b/research/slim/slim/preprocessing/__init__.py similarity index 100% rename from research/slim/preprocessing/__init__.py rename to research/slim/slim/preprocessing/__init__.py diff --git a/research/slim/preprocessing/cifarnet_preprocessing.py b/research/slim/slim/preprocessing/cifarnet_preprocessing.py similarity index 100% rename from research/slim/preprocessing/cifarnet_preprocessing.py rename to research/slim/slim/preprocessing/cifarnet_preprocessing.py diff --git a/research/slim/preprocessing/inception_preprocessing.py b/research/slim/slim/preprocessing/inception_preprocessing.py similarity index 100% rename from research/slim/preprocessing/inception_preprocessing.py rename to research/slim/slim/preprocessing/inception_preprocessing.py diff --git a/research/slim/preprocessing/lenet_preprocessing.py b/research/slim/slim/preprocessing/lenet_preprocessing.py similarity index 100% rename from research/slim/preprocessing/lenet_preprocessing.py rename to research/slim/slim/preprocessing/lenet_preprocessing.py diff --git a/research/slim/preprocessing/preprocessing_factory.py b/research/slim/slim/preprocessing/preprocessing_factory.py similarity index 100% rename from research/slim/preprocessing/preprocessing_factory.py rename to research/slim/slim/preprocessing/preprocessing_factory.py diff --git a/research/slim/preprocessing/vgg_preprocessing.py b/research/slim/slim/preprocessing/vgg_preprocessing.py similarity index 100% rename from research/slim/preprocessing/vgg_preprocessing.py rename to research/slim/slim/preprocessing/vgg_preprocessing.py