3 #ifndef BBOX_TRANSFORM_OP_H_ 4 #define BBOX_TRANSFORM_OP_H_ 6 #include "caffe2/core/context.h" 7 #include "caffe2/core/logging.h" 8 #include "caffe2/core/operator.h" 9 #include "caffe2/utils/math.h" 13 template <
typename T,
class Context>
18 weights_(OperatorBase::GetRepeatedArgument<T>(
20 vector<T>{1.0f, 1.0f, 1.0f, 1.0f})),
22 OperatorBase::GetSingleArgument<bool>(
"apply_scale",
true)),
23 correct_transform_coords_(OperatorBase::GetSingleArgument<bool>(
24 "correct_transform_coords",
29 "weights size " + caffe2::to_string(weights_.size()) +
"must be 4.");
31 USE_OPERATOR_CONTEXT_FUNCTIONS;
33 bool RunOnDevice()
override;
42 bool apply_scale_{
true};
46 bool correct_transform_coords_{
false};
51 #endif // BBOX_TRANSFORM_OP_H_ 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 ...