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): ...@@ -18,7 +18,6 @@ class YoloLayer(tf.keras.Model):
nms_thresh=0.6, nms_thresh=0.6,
max_delta=10.0, max_delta=10.0,
loss_type='ciou', loss_type='ciou',
use_tie_breaker=True,
iou_normalizer=1.0, iou_normalizer=1.0,
cls_normalizer=1.0, cls_normalizer=1.0,
obj_normalizer=1.0, obj_normalizer=1.0,
...@@ -109,7 +108,6 @@ class YoloLayer(tf.keras.Model): ...@@ -109,7 +108,6 @@ class YoloLayer(tf.keras.Model):
self._max_delta = max_delta self._max_delta = max_delta
self._classes = classes self._classes = classes
self._loss_type = loss_type self._loss_type = loss_type
self._use_tie_breaker = use_tie_breaker
self._use_scaled_loss = use_scaled_loss self._use_scaled_loss = use_scaled_loss
self._darknet = darknet 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