finalize context rcnn tf2 support

8b6a4628 · Kaushik Shivakumar · 6d39f5e6 · 8b6a4628
Commit 8b6a4628 authored Jul 23, 2020 by Kaushik Shivakumar
Show whitespace changes
Inline Side-by-side

Showing with 9 additions and 10 deletions

research/object_detection/meta_architectures/context_rcnn_lib_tf2.py ...ject_detection/meta_architectures/context_rcnn_lib_tf2.py +9 -10

No files found.
--- a/research/object_detection/meta_architectures/context_rcnn_lib_tf2.py
+++ b/research/object_detection/meta_architectures/context_rcnn_lib_tf2.py
@@ -73,7 +73,6 @@ class AttentionBlock(tf.keras.layers.Layer):
    # box_features becomes [batch_size, max_num_proposals, channels].
    box_features = tf.reduce_mean(box_features, [2, 3])

-    with tf.name_scope("AttentionBlock"):
    queries = project_features(
        box_features, self._bottleneck_dimension, self._is_training,
        self._query_proj, normalize=True)