Skip to content

GitLab

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

Internal change 

PiperOrigin-RevId: 406237483
parent fed7723c
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
official/vision/beta/modeling/retinanet_model.py
View file @ 067c893e
...@@ -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