TransformEstimator.h

跳转至此文件的文档。

/*
 * 版权所有 2023-2024 NVIDIA Corporation。保留所有权利。
 *
 * 许可证持有者须知：
 *
 * 此源代码和/或文档（“许可交付物”）受美国和国际版权法项下的 NVIDIA 知识产权保护。
 * 受美国和
 * 国际版权法约束。
 *
 * 此处包含的这些许可交付物是 NVIDIA 的专有和机密信息，并根据 NVIDIA 与被许可方之间的 NVIDIA 软件许可协议（“许可协议”）条款和条件或被许可方以电子方式接受的条款和条件提供。
 * 机密信息，并根据 NVIDIA 软件许可协议的条款和
 * 条件提供，协议形式为 NVIDIA 与被许可方之间达成的协议（“许可协议”）或
 * 由被许可方以电子方式接受。尽管许可协议中有任何相反的条款或条件，
 * 由被许可方接受。尽管许可协议中有任何相反的条款或条件，
 * 许可协议中与此相反的任何条款或条件，未经 NVIDIA 明确书面同意，
 * 许可协议中与此相反的任何条款或条件，未经 NVIDIA 明确
 * 书面同意，禁止向任何第三方复制或披露许可交付物。
 *
 * 尽管许可协议中有任何相反的条款或条件，
 * 许可协议，NVIDIA 对这些许可交付物针对任何特定用途的适用性不做任何陈述。
 * 许可协议，NVIDIA 对这些许可交付物针对任何用途的适用性不做任何陈述。
 * 它们按“原样”提供，不附带任何明示或暗示的担保。
 * NVIDIA 声明不对这些许可交付物承担任何担保责任，包括所有关于适销性、
 * 交付物的所有担保，包括所有关于适销性、
 * 非侵权性以及针对特定用途的适用性的默示担保。
 * 尽管许可协议中有任何相反的条款或条件，
 * 许可协议，在任何情况下，NVIDIA 均不对任何
 * 特殊、间接、附带或后果性损害，或任何因使用、数据或利润损失而导致的任何损害承担责任，
 * 特殊、间接、附带或后果性损害，或任何
 * 无论是合同诉讼、过失或其他侵权行为，因使用或执行这些许可交付物而引起或与之相关的诉讼。
 * 行为，因使用或与使用或性能相关
 * 这些许可交付物。
 *
 * 美国政府最终用户。这些许可交付物是
 * “商业项目”，该术语在 48 C.F.R. 2.101 (OCT
 * 1995) 中定义，包括“商业计算机软件”和“商业
 * 计算机软件文档”，这些术语在 48
 * C.F.R. 12.212 (SEPT 1995) 中使用，并且仅作为商业最终项目提供给美国政府。
 * 仅作为商业最终项目。根据 48 C.F.R. 12.212 和
 * 48 C.F.R. 227.7202-1 至 227.7202-4 (JUNE 1995)，所有美国政府最终用户均获得许可交付物，
 * 48 C.F.R. 227.7202-1 至 227.7202-4 (JUNE 1995)，所有
 * 美国政府最终用户按照此处规定的权利获取许可交付物。
 * 仅根据此处规定的权利获得许可交付物。
 *
 * 在个人和商业软件中使用许可交付物时，必须在用户文档和代码的内部注释中包含上述免责声明和美国政府最终用户须知。
 * 软件必须包括，在用户文档和内部
 * 注释中，上述免责声明和美国政府最终
 */

#ifndef NV_VPI_ALGORITHMS_TRANSFORM_ESTIMATOR_H
#define NV_VPI_ALGORITHMS_TRANSFORM_ESTIMATOR_H

#include "../AlgoFlags.h"
#include "../Export.h"
#include "../Status.h"
#include "../Types.h"

#include <stdint.h>

#ifdef __cplusplus
extern "C" {
#endif

VPI_PUBLIC VPIStatus vpiCreateTransformEstimator(uint64_t backends, int32_t maxKeypoints, VPIPayload *payload);

typedef struct
{
 float minRotation, maxRotation;
 /* @} */

 float minXScale, maxXScale;
 float minYScale, maxYScale;
 float minXTranslation, maxXTranslation;
 float minYTranslation, maxYTranslation;
 float minShear, maxShear;
 int8_t isAffine;

 int8_t isIsotropicScale;
} VPIConstrainedHomography2DConfig;

typedef enum
{
 VPI_XFORM_CONSTRAINED_HOMOGRAPHY_2D
} VPITransformType;

typedef union
{
 VPIConstrainedHomography2DConfig constrainedHomography2D;
} VPITransformConfig;

typedef enum
{
 VPI_XFORM_ESTIM_METHOD_REGULAR,
 VPI_XFORM_ESTIM_METHOD_RANSAC,
} VPITransfomEstimationMethod;

typedef struct
{
 VPITransfomEstimationMethod method;

 int32_t solverMaxIterations;

 int32_t ransacMaxIterations;
 float ransacReprojErrorTolerance;
 float ransacConfidenceLevel;
 int32_t ransacSeed;
 int32_t maxRefinementIterations;

 VPITransformType xftype;
 VPITransformConfig xfcfg;
} VPITransformEstimatorParams;

VPI_PUBLIC VPIStatus vpiInitTransformEstimatorParams(VPITransformType type, VPITransformEstimatorParams *params);

VPI_PUBLIC VPIStatus vpiSubmitTransformEstimator(VPIStream stream, uint64_t backend, VPIPayload payload,
 VPIArray srcKeypoints, VPIArray tgtKeypoints, VPIArray matches,
 VPIArray outTransform, VPIArray outInliers,
 const VPITransformEstimatorParams *params);

#ifdef __cplusplus
}
#endif

#endif /* NV_VPI_ALGORITHMS_TRANSFORM_ESTIMATOR_H */