addressing comments

c4a9fa69 · Vishnu Banna · 6de77679 · c4a9fa69
Commit c4a9fa69 authored Sep 07, 2021 by Vishnu Banna
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/vision/beta/projects/yolo/modeling/layers/detection_generator.py ...beta/projects/yolo/modeling/layers/detection_generator.py +1 -1

No files found.
--- a/official/vision/beta/projects/yolo/modeling/layers/detection_generator.py
+++ b/official/vision/beta/projects/yolo/modeling/layers/detection_generator.py
@@ -178,7 +178,7 @@ class YoloLayer(tf.keras.Model):

    # determine the number of classes
    classes = class_scores.get_shape().as_list()[
-        -1]  #tf.shape(class_scores)[-1]
+        -1]  

    # configurable to use the new coordinates in scaled Yolo v4 or not
    boxes = None