sources/libs/nvmultiobjecttracker/include/modules/ReID/ReIDFrameTransformTexture.hpp

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 2021-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 * SPDX-License-Identifier: Apache-2.0
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
 
#ifndef ReIDFrameTransform_HPP_
#define ReIDFrameTransform_HPP_
 
#if defined VIZ_DEBUG_REID || defined VIZ_DEBUG_SEG || defined VIZ_DEBUG_POSE
#include <opencv2/core.hpp>
#endif
 
#include "modules/FrameTransform/FrameTransform.hpp"
 
class ReIDFrameTransformTexture : public FrameTransformTexture
{
public:
    ReIDFrameTransformTexture();
 
    ~ReIDFrameTransformTexture();
 
    void initialize(
        const int debugVerboseLevel, const int maxStreamNum, const int maxBufSurfAddrSize, const int maxTargetNum,
        const NvMotDataTypes::Size srcImgSize, const uint srcImgPitch, const NvBufSurfaceColorFormat colorFormatType,
        const NvMotDataTypes::Size dstImgSize, const NvBufSurfaceColorFormat dstColorFormatType, const int dimOrder
    );
 
    bool frameTransformInBatch(
        const std::map<NvMOTStreamId, std::vector<std::pair<uint, NvMotDataTypes::Rect2f> > > &roisInBatch,
        const uint lastBboxInd, const float outputScaleFactor, const std::vector<float> outputOffsets
    );
 
    bool frameTransformInBatch_NV12(
        const std::map<NvMOTStreamId, std::vector<std::pair<uint, NvMotDataTypes::Rect2f> > > &roisInBatch,
        const uint lastBboxInd, const float outputScaleFactor, const std::vector<float> outputOffsets
    );
 
    float* getOutputImagePatchPtrs() { return m_odata_d; }
 
    #if defined VIZ_DEBUG_REID || defined VIZ_DEBUG_SEG || defined VIZ_DEBUG_POSE
 
        std::map<NvMOTStreamId, std::map<uint, cv::Mat > > getImagePatches2Host_OCV_batch(
            const float outputScaleFactor, const std::vector<float> outputOffsets
        );
    #endif
 
private:
    void initialize4Output(
        const NvMotDataTypes::Size dstSize, const NvBufSurfaceColorFormat dstColorFormat, const int dimOrder
    );
 
    void allocateOutputMemories();
 
    void deAllocateOutputMemories();
 
    bool m_isInit; 
 
    // [Output configuration specific to ReID]
    int m_width_out, m_height_out, m_channels_out; 
    int m_dimOrder; 
    NvBufSurfaceColorFormat m_dstColorFormatType; 
 
    // [Output memory specific to ReID]
    float *m_odata_d; 
    float *m_odata_host; 
    #if defined VIZ_DEBUG_REID || defined VIZ_DEBUG_SEG || defined VIZ_DEBUG_POSE
        float *m_odata_host_nchw; 
    #endif
};
 
#endif // ReIDFrameTransform_HPP_