add side inputs

ff2bf0b0 · Kaushik Shivakumar · 35fda973 · ff2bf0b0
Commit ff2bf0b0 authored Jun 30, 2020 by Kaushik Shivakumar
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

research/object_detection/model_lib_v2.py research/object_detection/model_lib_v2.py +2 -1

No files found.
--- a/research/object_detection/model_lib_v2.py
+++ b/research/object_detection/model_lib_v2.py
@@ -119,7 +119,8 @@ def _compute_losses_and_predictions_dicts(

  prediction_dict = model.predict(
      preprocessed_images,
-      features[fields.InputDataFields.true_image_shape])
+      features[fields.InputDataFields.true_image_shape],
+      model.get_side_inputs(features))
  prediction_dict = ops.bfloat16_to_float32_nested(prediction_dict)

  losses_dict = model.loss(