Caffe2 - C++ API
A deep learning, cross platform ML framework
heatmap_max_keypoint_op.h
1 // Copyright 2004-present Facebook. All Rights Reserved.
2 
3 #ifndef HEATMAP_MAX_KEYPOINT_OP_H_
4 #define HEATMAP_MAX_KEYPOINT_OP_H_
5 
6 #include "caffe2/core/context.h"
7 #include "caffe2/core/logging.h"
8 #include "caffe2/core/operator.h"
9 #include "caffe2/utils/math.h"
10 
11 namespace caffe2 {
12 
13 template <typename T, class Context>
14 class HeatmapMaxKeypointOp final : public Operator<Context> {
15  public:
16  template <class... Args>
17  explicit HeatmapMaxKeypointOp(Args&&... args)
18  : Operator<Context>(std::forward<Args>(args)...),
19  should_output_softmax_(this->template GetSingleArgument<bool>(
20  "should_output_softmax",
21  false)) {}
22  USE_OPERATOR_CONTEXT_FUNCTIONS;
23 
24  bool RunOnDevice() override;
25 
26  protected:
27  bool should_output_softmax_ = false;
28 };
29 
30 } // namespace caffe2
31 
32 #endif // HEATMAP_MAX_KEYPOINT_OP_H_
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...
Definition: blob.h:13