Created
January 31, 2019 09:32
-
-
Save joyhuang9473/139cd9ed31020a9ca113a0002cadf0bc to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
- [export_inference_graph.py](https://github.com/tensorflow/models/blob/master/research/object_detection/export_inference_graph.py#L143) | |
- exporter.export_inference_graph | |
- [exporter.py](https://github.com/tensorflow/models/blob/master/research/object_detection/exporter.py#L414) | |
- export_inference_graph() | |
- detection_model = model_builder.build [link](https://github.com/tensorflow/models/blob/master/research/object_detection/exporter.py#L438) | |
- [builders/model_builder.py](https://github.com/tensorflow/models/blob/master/research/object_detection/builders/model_builder.py#L101) | |
- build(model_config, is_training, add_summaries=True) | |
- _build_ssd_model(model_config.ssd, is_training, add_summaries) [link](https://github.com/tensorflow/models/blob/master/research/object_detection/builders/model_builder.py#L119) | |
- feature_extractor = _build_ssd_feature_extractor( [link](https://github.com/tensorflow/models/blob/master/research/object_detection/builders/model_builder.py#L233) | |
- conv_hyperparams = hyperparams_builder.build( [link](https://github.com/tensorflow/models/blob/master/research/object_detection/builders/model_builder.py#L159) | |
- feature_extractor_class = SSD_FEATURE_EXTRACTOR_CLASS_MAP[feature_type] [link](https://github.com/tensorflow/models/blob/master/research/object_detection/builders/model_builder.py#L172) | |
- 'ssd_mobilenet_v2': SSDMobileNetV2FeatureExtractor, [link](https://github.com/tensorflow/models/blob/master/research/object_detection/builders/model_builder.py#L64) | |
- [models/ssd_mobilenet_v2_feature_extractor.py](https://github.com/tensorflow/models/blob/master/research/object_detection/models/ssd_mobilenet_v2_feature_extractor.py#L31) | |
- kwargs.update({ [link](https://github.com/tensorflow/models/blob/master/research/object_detection/builders/model_builder.py#L197) | |
- return feature_extractor_class(**kwargs) [link](https://github.com/tensorflow/models/blob/master/research/object_detection/builders/model_builder.py#L212) | |
- image_resizer_fn = image_resizer_builder.build(ssd_config.image_resizer) [link](https://github.com/tensorflow/models/blob/master/research/object_detection/builders/model_builder.py#L261) | |
- [builders/image_resizer_builder.py](https://github.com/tensorflow/models/blob/master/research/object_detection/builders/image_resizer_builder.py#L52) | |
- build(image_resizer_config) | |
- keep_aspect_ratio_config = image_resizer_config.keep_aspect_ratio_resizer [link](https://github.com/tensorflow/models/blob/master/research/object_detection/builders/image_resizer_builder.py#L77) | |
- method = _tf_resize_method(keep_aspect_ratio_config.resize_method) [link](https://github.com/tensorflow/models/blob/master/research/object_detection/builders/image_resizer_builder.py#L81) | |
- | |
``` | |
image_resizer_fn = functools.partial( | |
preprocessor.resize_to_range, | |
min_dimension=keep_aspect_ratio_config.min_dimension, | |
max_dimension=keep_aspect_ratio_config.max_dimension, | |
method=method, | |
pad_to_max_dimension=keep_aspect_ratio_config.pad_to_max_dimension, | |
per_channel_pad_value=per_channel_pad_value) | |
``` | |
[link] https://github.com/tensorflow/models/blob/master/research/object_detection/builders/image_resizer_builder.py#L86 | |
- [core/preprocessor.py](https://github.com/tensorflow/models/blob/master/research/object_detection/core/preprocessor.py#L2178) | |
- resize_to_range() | |
- new_size = _compute_new_dynamic_size(image, min_dimension, max_dimension) [link](https://github.com/tensorflow/models/blob/master/research/object_detection/core/preprocessor.py#L2235) | |
- _compute_new_dynamic_size(image, min_dimension, max_dimension) [link](https://github.com/tensorflow/models/blob/master/research/object_detection/core/preprocessor.py#L2140) | |
- | |
``` | |
new_size = tf.cond( | |
tf.to_float(tf.reduce_max(large_size)) > max_dimension, | |
lambda: small_size, lambda: large_size) | |
``` | |
[link](https://github.com/tensorflow/models/blob/master/research/object_detection/core/preprocessor.py#L2170) | |
- new_image = tf.image.resize_images [link](https://github.com/tensorflow/models/blob/master/research/object_detection/core/preprocessor.py#L2236) | |
- result = [new_image] [link](https://github.com/tensorflow/models/blob/master/research/object_detection/core/preprocessor.py#L2255) | |
- result.append(new_size) [link](https://github.com/tensorflow/models/blob/master/research/object_detection/core/preprocessor.py#L2269) | |
- return result |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment