Skip to content

GitLab

Commit db9ef288 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 406237483
parent ac7f9e7f
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
official/vision/beta/modeling/retinanet_model.py
View file @ db9ef288
...@@ -123,9 +123,9 @@ class RetinaNetModel(tf.keras.Model): ...@@ -123,9 +123,9 @@ class RetinaNetModel(tf.keras.Model):
{'backbone_{}'.format(k): v for k, v in features.items()}) {'backbone_{}'.format(k): v for k, v in features.items()})
if self.decoder: if self.decoder:
features = self.decoder(features) features = self.decoder(features)
if output_intermediate_features: if output_intermediate_features:
outputs.update( outputs.update(
{'decoder_{}'.format(k): v for k, v in features.items()}) {'decoder_{}'.format(k): v for k, v in features.items()})
# Dense prediction. `raw_attributes` can be empty. # Dense prediction. `raw_attributes` can be empty.
raw_scores, raw_boxes, raw_attributes = self.head(features) raw_scores, raw_boxes, raw_attributes = self.head(features)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment