1 #ifndef OPENPOSE_NET_RESIZE_AND_MERGE_CAFFE_HPP
2 #define OPENPOSE_NET_RESIZE_AND_MERGE_CAFFE_HPP
22 const T netFactor,
const T scaleFactor,
const bool mergeFirstDimension =
true,
25 virtual inline const char*
type()
const {
return "ResizeAndMerge"; }
42 std::vector<T*> mTempGPUData;
43 std::vector<T> mScaleRatios;
44 std::vector<std::array<int, 4>> mBottomSizes;
45 std::array<int, 4> mTopSize;
52 #endif // OPENPOSE_NET_RESIZE_AND_MERGE_CAFFE_HPP
void setScaleRatios(const std::vector< T > &scaleRatios)
virtual void Backward_gpu(const std::vector< caffe::Blob< T > * > &top, const std::vector< bool > &propagate_down, const std::vector< caffe::Blob< T > * > &bottom)
virtual ~ResizeAndMergeCaffe()
virtual void Forward_cpu(const std::vector< caffe::Blob< T > * > &bottom, const std::vector< caffe::Blob< T > * > &top)
Definition: resizeAndMergeCaffe.hpp:12
virtual void Reshape(const std::vector< caffe::Blob< T > * > &bottom, const std::vector< caffe::Blob< T > * > &top, const T netFactor, const T scaleFactor, const bool mergeFirstDimension=true, const int gpuID=0)
virtual const char * type() const
Definition: resizeAndMergeCaffe.hpp:25
virtual void Forward_ocl(const std::vector< caffe::Blob< T > * > &bottom, const std::vector< caffe::Blob< T > * > &top)
Definition: macros.hpp:80
virtual void Backward_cpu(const std::vector< caffe::Blob< T > * > &top, const std::vector< bool > &propagate_down, const std::vector< caffe::Blob< T > * > &bottom)
virtual void LayerSetUp(const std::vector< caffe::Blob< T > * > &bottom, const std::vector< caffe::Blob< T > * > &top)
std::vector< T, Alloc > vector
Definition: cl2.hpp:567
virtual void Forward_gpu(const std::vector< caffe::Blob< T > * > &bottom, const std::vector< caffe::Blob< T > * > &top)