Commit 26e24e21 authored by Kaushik Shivakumar's avatar Kaushik Shivakumar
Browse files

fix

parent 22ce50ad
......@@ -221,4 +221,5 @@ def compute_box_context_attention(box_features, context_features,
# Expands the dimension back to match with the original feature map.
output_features = output_features[:, :, tf.newaxis, tf.newaxis, :]
return output_features
\ No newline at end of file
return output_features
\ No newline at end of file
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