Skip to content

GitLab

Commit 9a1b8990 authored by Vishnu Banna's avatar Vishnu Banna
Browse files

nms ops used by detection generator

parent fb2b278d
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 3 deletions
official/vision/beta/projects/yolo/ops/box_ops.py
View file @ 9a1b8990
import tensorflow as tf import tensorflow as tf
import tensorflow.keras.backend as K
from official.vision.beta.projects.yolo.ops import math_ops from official.vision.beta.projects.yolo.ops import math_ops
from typing import Tuple, Union
import math import math
......
official/vision/beta/projects/yolo/ops/math_ops.py
View file @ 9a1b8990
"""A set of private math operations used to safely implement the yolo loss""" """A set of private math operations used to safely implement the yolo loss"""
import tensorflow as tf import tensorflow as tf
import tensorflow.keras.backend as K
def rm_nan_inf(x, val=0.0): def rm_nan_inf(x, val=0.0):
......
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