Skip to content

Instantly share code, notes, and snippets.

Show Gist options
  • Save joyhuang9473/139cd9ed31020a9ca113a0002cadf0bc to your computer and use it in GitHub Desktop.
Save joyhuang9473/139cd9ed31020a9ca113a0002cadf0bc to your computer and use it in GitHub Desktop.
- [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