Skip to content

GitLab

"docs/git@developer.sourcefind.cn:hehl2/torchaudio.git" did not exist on "1f136671b84071a2fe1d5b762df64f3a76310c31"
Commit 5cee7220 authored by Zhenyu Tan's avatar Zhenyu Tan Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 332377554
parent e342145e
Show whitespace changes
Inline Side-by-side
Showing with 6 additions and 10 deletions
official/vision/keras_cv/ops/box_matcher.py
View file @ 5cee7220
...@@ -146,14 +146,10 @@ class BoxMatcher: ...@@ -146,14 +146,10 @@ class BoxMatcher:
else: else:
return matches return matches
if similarity_matrix.shape.is_fully_defined(): num_gt_boxes = similarity_matrix.shape.as_list()[0] or tf.shape(
if similarity_matrix.shape.dims[0].value == 0: similarity_matrix)[0]
return _match_when_rows_are_empty()
else:
return _match_when_rows_are_non_empty()
else:
return tf.cond( return tf.cond(
pred=tf.greater(tf.shape(similarity_matrix)[0], 0), pred=tf.greater(num_gt_boxes, 0),
true_fn=_match_when_rows_are_non_empty, true_fn=_match_when_rows_are_non_empty,
false_fn=_match_when_rows_are_empty) false_fn=_match_when_rows_are_empty)
......
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