Skip to content

GitLab

Commit b4aa41f5 authored by Kaushik Shivakumar's avatar Kaushik Shivakumar
Browse files

finalize context rcnn tf2

parent eef0a1e7
Show whitespace changes
Inline Side-by-side
Showing with 10 additions and 9 deletions
research/object_detection/meta_architectures/context_rcnn_lib_tf2.py
View file @ b4aa41f5
...@@ -59,6 +59,7 @@ class AttentionBlock(tf.keras.layers.Layer): ...@@ -59,6 +59,7 @@ class AttentionBlock(tf.keras.layers.Layer):
output feature. output feature.
is_training: A boolean Tensor (affecting batch normalization). is_training: A boolean Tensor (affecting batch normalization).
name: A string describing what to name the variables in this block. name: A string describing what to name the variables in this block.
**kwargs: Additional keyword arguments.
""" """
self._key_proj = ContextProjection(bottleneck_dimension) self._key_proj = ContextProjection(bottleneck_dimension)
......
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