#ifndef CAFFE2_OPERATORS_RMAC_REGIONS_OP_H #define CAFFE2_OPERATORS_RMAC_REGIONS_OP_H #include "caffe2/core/operator.h" namespace caffe2 { template class RMACRegionsOp final : public Operator { public: template explicit RMACRegionsOp(Args&&... args) : Operator(std::forward(args)...), scales_(this->template GetSingleArgument("scales", 3)), overlap_(this->template GetSingleArgument("overlap", 0.4f)) {} USE_OPERATOR_CONTEXT_FUNCTIONS; bool RunOnDevice() override; protected: int scales_; float overlap_; Tensor num_rois_; }; } // namespace caffe2 #endif // CAFFE2_OPERATORS_RMAC_REGIONS_OP_H