Skip to content

GitLab

"magic_pdf/git@developer.sourcefind.cn:wangsen/mineru.git" did not exist on "c4010ae02f02ab13f31c336161659cc6eae39104"
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.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