DCFTracker.h

前往此文件的文档。

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

#ifndef NV_VPI_ALGORITHMS_DCF_TRACKER_H
#define NV_VPI_ALGORITHMS_DCF_TRACKER_H

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

#include <stdint.h>

#ifdef __cplusplus
extern "C" {
#endif

typedef enum
{
 VPI_DCFTRACKER_USE_COLORNAMES = 1,
 VPI_DCFTRACKER_USE_HOG = 2,
 VPI_DCFTRACKER_USE_HIGH_PRECISION_FEATURES = 4,
 VPI_DCFTRACKER_DETERMINISTIC = 8
} VPIDCFTrackerCreationFlag;

typedef struct
{
 uint32_t flags;

 int32_t featurePatchSize;

 int32_t hogCellSize;

 float featureFocusVertOffsetFactor;

 float gaussianSigma;
} VPIDCFTrackerCreationParams;

VPI_PUBLIC VPIStatus vpiInitDCFTrackerCreationParams(VPIDCFTrackerCreationParams *params);

VPI_PUBLIC VPIStatus vpiCreateDCFTracker(uint64_t backends, int32_t maxNumSequences, int32_t maxNumObjects,
 const VPIDCFTrackerCreationParams *params, VPIPayload *payload);

typedef struct
{
 float dcfRegFactor;
} VPIDCFTrackerParams;

VPI_PUBLIC VPIStatus vpiInitDCFTrackerParams(VPIDCFTrackerParams *params);

VPI_PUBLIC VPIStatus vpiSubmitDCFTrackerLocalizeBatch(
 VPIStream stream, uint64_t backend, VPIPayload payload, const int32_t *enabledSequences, int32_t numSequences,
 VPIImage featureMaskingWindow, VPIImage inPatches, VPIArray inObjects, VPIArray outObjects,
 VPIImage outCorrelationResponses, VPIArray outMaxCorrelationResponses, const VPIDCFTrackerParams *params);

VPI_PUBLIC VPIStatus vpiSubmitDCFTrackerUpdateBatch(VPIStream stream, uint64_t backend, VPIPayload payload,
 const int32_t *enabledSequences, int32_t numSequences,
 VPIImage featureMaskingWindow, VPIImage modelMaskingWindow,
 VPIImage inPatches, VPIArray trackedObjects,
 const VPIDCFTrackerParams *params);

VPI_PUBLIC VPIStatus vpiDCFTrackerGetChannelWeights(VPIPayload payload, VPIArray *channelWeights,
 int32_t *numFeatureChannels);

#ifdef __cplusplus
}
#endif

#endif /* NV_VPI_ALGORITHMS_DCF_TRACKER_H */