You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
26 lines
461 B
26 lines
461 B
#ifndef __OPENCV_DNN_LAYERS_SLICE_LAYER_HPP__ |
|
#define __OPENCV_DNN_LAYERS_SLICE_LAYER_HPP__ |
|
#include "../precomp.hpp" |
|
|
|
namespace cv |
|
{ |
|
namespace dnn |
|
{ |
|
|
|
class SliceLayer : public Layer |
|
{ |
|
public: |
|
SliceLayer(LayerParams ¶ms); |
|
|
|
void allocate(const std::vector<Blob*> &inputs, std::vector<Blob> &outputs); |
|
|
|
void forward(std::vector<Blob*> &inputs, std::vector<Blob> &outputs); |
|
|
|
private: |
|
int inAxis; |
|
std::vector<int> slicePoints; |
|
}; |
|
|
|
} |
|
} |
|
#endif |