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.
21 lines
487 B
21 lines
487 B
#ifndef __OPENCV_DNN_LAYERS_SOFTMAX_LAYER_HPP__ |
|
#define __OPENCV_DNN_LAYERS_SOFTMAX_LAYER_HPP__ |
|
#include "../precomp.hpp" |
|
|
|
namespace cv |
|
{ |
|
namespace dnn |
|
{ |
|
class SoftMaxLayer : public Layer |
|
{ |
|
int axis_, axis; |
|
Blob maxAggregator; |
|
|
|
public: |
|
SoftMaxLayer(LayerParams ¶ms); |
|
void allocate(const std::vector<Blob*> &inputs, std::vector<Blob> &outputs); |
|
void forward(std::vector<Blob*> &inputs, std::vector<Blob> &outputs); |
|
}; |
|
} |
|
} |
|
#endif |