2 #ifndef CAFFE2_OPERATORS_RMAC_REGIONS_OP_H 3 #define CAFFE2_OPERATORS_RMAC_REGIONS_OP_H 5 #include "caffe2/core/operator.h" 9 template <
class Context>
14 scales_(OperatorBase::GetSingleArgument<int>(
"scales", 3)),
15 overlap_(OperatorBase::GetSingleArgument<float>(
"overlap", 0.4f)) {}
17 USE_OPERATOR_CONTEXT_FUNCTIONS;
19 bool RunOnDevice()
override;
29 #endif // CAFFE2_OPERATORS_RMAC_REGIONS_OP_H Tensor is the basic class in Caffe2 that stores a contiguous memory with its shape information...
Workspace is a class that holds all the related objects created during runtime: (1) all blobs...
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...