Skip to content

GitLab

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

nms ops used by detection generator

parent fb2b278d
Show 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.keras.backend as K
from official.vision.beta.projects.yolo.ops import math_ops
from typing import Tuple, Union
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"""
import tensorflow as tf
import tensorflow.keras.backend as K
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