Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
ResNet50_tensorflow
Commits
7f90664e
Commit
7f90664e
authored
Sep 13, 2021
by
Vishnu Banna
Browse files
tiny comments
parent
0126950d
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
3 additions
and
3 deletions
+3
-3
official/vision/beta/projects/yolo/modeling/layers/detection_generator.py
...beta/projects/yolo/modeling/layers/detection_generator.py
+1
-1
official/vision/beta/projects/yolo/ops/box_ops.py
official/vision/beta/projects/yolo/ops/box_ops.py
+1
-1
official/vision/beta/projects/yolo/ops/math_ops.py
official/vision/beta/projects/yolo/ops/math_ops.py
+1
-1
No files found.
official/vision/beta/projects/yolo/modeling/layers/detection_generator.py
View file @
7f90664e
...
...
@@ -272,7 +272,7 @@ class YoloLayer(tf.keras.Model):
@
property
def
losses
(
self
):
"""
Generates a dictionary of losses to apply to each path
"""Generates a dictionary of losses to apply to each path
Done in the detection generator because all parameters are the same
across both loss and detection generator
...
...
official/vision/beta/projects/yolo/ops/box_ops.py
View file @
7f90664e
...
...
@@ -324,4 +324,4 @@ def aggregated_comparitive_iou(boxes1, boxes2=None, iou_type=0, beta=0.6):
_
,
iou
=
compute_ciou
(
boxes1
,
boxes2
,
yxyx
=
True
)
else
:
iou
=
compute_iou
(
boxes1
,
boxes2
,
yxyx
=
True
)
return
iou
\ No newline at end of file
return
iou
official/vision/beta/projects/yolo/ops/math_ops.py
View file @
7f90664e
...
...
@@ -58,4 +58,4 @@ def divide_no_nan(a, b):
Returns:
a `Tensor` representing a divided by b, with all nan values removed.
"""
return
a
/
(
b
+
1e-9
)
\ No newline at end of file
return
a
/
(
b
+
1e-9
)
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment