resizeAndMergeBase.hpp

Go to the documentation of this file.

#ifndef OPENPOSE_CORE_RESIZE_AND_MERGE_BASE_HPP
#define OPENPOSE_CORE_RESIZE_AND_MERGE_BASE_HPP

#include <openpose/core/common.hpp>

namespace op
{
    template <typename T>
    OP_API void resizeAndMergeCpu(T* targetPtr, const std::vector<const T*>& sourcePtrs,
                                  const std::array<int, 4>& targetSize,
                                  const std::vector<std::array<int, 4>>& sourceSizes,
                                  const std::vector<T>& scaleInputToNetInputs = {1.f});

    template <typename T>
    OP_API void resizeAndMergeGpu(T* targetPtr, const std::vector<const T*>& sourcePtrs,
                                  const std::array<int, 4>& targetSize,
                                  const std::vector<std::array<int, 4>>& sourceSizes,
                                  const std::vector<T>& scaleInputToNetInputs = {1.f});

    template <typename T>
    OP_API void resizeAndMergeOcl(T* targetPtr, const std::vector<const T*>& sourcePtrs,
                                  const std::array<int, 4>& targetSize,
                                  const std::vector<std::array<int, 4>>& sourceSizes,
                                  const std::vector<T>& scaleInputToNetInputs = {1.f},
                                  const int gpuID = 0);
}

#endif // OPENPOSE_CORE_RESIZE_AND_MERGE_BASE_HPP