cn.smartjavaai.common.utils

类 LetterBoxUtils

java.lang.Object

cn.smartjavaai.common.utils.LetterBoxUtils

public class LetterBoxUtils
extends Object

按比例缩放，剩余空间用指定颜色填充

作者:

dwj

嵌套类概要

嵌套类

限定符和类型	类和说明
static class	LetterBoxUtils.PaddingPosition
static class	LetterBoxUtils.ResizeResult

构造器概要

构造器

构造器和说明
LetterBoxUtils()

方法概要

限定符和类型	方法和说明
static int[]	getResizeSize(int origW, int origH, int targetWidth, int targetHeight) 获取缩放后的图片大小
static LetterBoxUtils.ResizeResult	letterbox(ai.djl.ndarray.NDManager manager, ai.djl.ndarray.NDArray img, int targetW, int targetH, float padColor, LetterBoxUtils.PaddingPosition position) 按比例缩放 + padding
static LetterBoxUtils.ResizeResult	letterboxWithMeta(ai.djl.ndarray.NDArray paddingImg, float r, int left, int top)
static ai.djl.modality.cv.output.Landmark	restoreBox(ai.djl.modality.cv.output.Landmark landmark, float scale, int origImageWidth, int origImageHeight, int inputWidth, int inputHeight, boolean isNormalized) 恢复缩放后的 box(左上角坐标)
static ai.djl.ndarray.NDArray	restoreBox(ai.djl.ndarray.NDArray boxes, float scaleRatio, float left, float top, int keypointStart, int keypointDim) 恢复缩放后的 box
static ai.djl.modality.cv.output.Rectangle	restoreBox(ai.djl.modality.cv.output.Rectangle rectangle, float scale, int origImageWidth, int origImageHeight, int inputWidth, int inputHeight) 恢复缩放后的 box(左上角坐标)

从类继承的方法 java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

构造器详细资料

LetterBoxUtils

public LetterBoxUtils()

方法详细资料

letterboxWithMeta

public static LetterBoxUtils.ResizeResult letterboxWithMeta(ai.djl.ndarray.NDArray paddingImg,
                                                            float r,
                                                            int left,
                                                            int top)

letterbox

public static LetterBoxUtils.ResizeResult letterbox(ai.djl.ndarray.NDManager manager,
                                                    ai.djl.ndarray.NDArray img,
                                                    int targetW,
                                                    int targetH,
                                                    float padColor,
                                                    LetterBoxUtils.PaddingPosition position)

按比例缩放 + padding

参数:

img - 原图 NDArray HWC

targetW - 目标宽度

targetH - 目标高度

padColor - padding 填充颜色，RGB 归一化 0-1

position - padding 位置：CENTER / LEFT_TOP / RIGHT_BOTTOM

返回:

处理后的 NDArray

restoreBox

public static ai.djl.ndarray.NDArray restoreBox(ai.djl.ndarray.NDArray boxes,
                                                float scaleRatio,
                                                float left,
                                                float top,
                                                int keypointStart,
                                                int keypointDim)

恢复缩放后的 box

参数:

boxes -

scaleRatio -

left -

top -

keypointStart -

keypointDim -

返回:

restoreBox

public static ai.djl.modality.cv.output.Rectangle restoreBox(ai.djl.modality.cv.output.Rectangle rectangle,
                                                             float scale,
                                                             int origImageWidth,
                                                             int origImageHeight,
                                                             int inputWidth,
                                                             int inputHeight)

恢复缩放后的 box(左上角坐标)

参数:

rectangle -

scale -

origImageWidth -

origImageHeight -

restoreBox

public static ai.djl.modality.cv.output.Landmark restoreBox(ai.djl.modality.cv.output.Landmark landmark,
                                                            float scale,
                                                            int origImageWidth,
                                                            int origImageHeight,
                                                            int inputWidth,
                                                            int inputHeight,
                                                            boolean isNormalized)

恢复缩放后的 box(左上角坐标)

参数:

landmark -

scale -

origImageWidth -

origImageHeight -

getResizeSize

public static int[] getResizeSize(int origW,
                                  int origH,
                                  int targetWidth,
                                  int targetHeight)

获取缩放后的图片大小

参数:

origW - 原始图片宽度

origH - 原始图片高度

targetWidth - 目标图片宽度

targetHeight - 目标图片高度

返回: