MaskRCNN

Copyright (c) organization

Author

btran

Date

2020-05-18

namespace Ort
class MaskRCNN : public Ort::ImageRecognitionOrtSessionHandlerBase

Public Functions

MaskRCNN(const uint16_t numClasses, const std::string &modelPath, const std::optional<size_t> &gpuIdx = std::nullopt, const std::optional<std::vector<std::vector<int64_t>>> &inputShapes = std::nullopt)
~MaskRCNN()
void preprocess(float *dst, const float *src, const int64_t targetImgWidth, const int64_t targetImgHeight, const int numChannels) const
void preprocess(float *dst, const cv::Mat &imgSrc, const int64_t targetImgWidth, const int64_t targetImgHeight, const int numChannels) const

Public Static Attributes

constexpr int64_t MIN_IMAGE_SIZE = 800
constexpr int64_t IMG_CHANNEL = 3