Skip to content

GitLab

Commit 73e9d6dd authored by Vishnu Banna's avatar Vishnu Banna
Browse files

addressing comments

parent e73c7f26
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
official/vision/beta/projects/yolo/modeling/layers/detection_generator.py
View file @ 73e9d6dd
......@@ -18,7 +18,6 @@ class YoloLayer(tf.keras.Model):
nms_thresh=0.6,
max_delta=10.0,
loss_type='ciou',
use_tie_breaker=True,
iou_normalizer=1.0,
cls_normalizer=1.0,
obj_normalizer=1.0,
......@@ -109,7 +108,6 @@ class YoloLayer(tf.keras.Model):
self._max_delta = max_delta
self._classes = classes
self._loss_type = loss_type
self._use_tie_breaker = use_tie_breaker
self._use_scaled_loss = use_scaled_loss
self._darknet = darknet
......
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