mirror of
https://github.com/opencv/opencv.git
synced 2026-07-30 15:53:03 +04:00
Add ShuffleChannel layer
This commit is contained in:
@@ -361,6 +361,23 @@ CV__DNN_EXPERIMENTAL_NS_BEGIN
|
||||
static Ptr<PermuteLayer> create(const LayerParams& params);
|
||||
};
|
||||
|
||||
/**
|
||||
* Permute channels of 4-dimensional input blob.
|
||||
* @param group Number of groups to split input channels and pick in turns
|
||||
* into output blob.
|
||||
*
|
||||
* \f[ groupSize = \frac{number\ of\ channels}{group} \f]
|
||||
* \f[ output(n, c, h, w) = input(n, groupSize \times (c \% group) + \lfloor \frac{c}{group} \rfloor, h, w) \f]
|
||||
* Read more at https://arxiv.org/pdf/1707.01083.pdf
|
||||
*/
|
||||
class CV_EXPORTS ShuffleChannelLayer : public Layer
|
||||
{
|
||||
public:
|
||||
static Ptr<Layer> create(const LayerParams& params);
|
||||
|
||||
int group;
|
||||
};
|
||||
|
||||
/**
|
||||
* @brief Adds extra values for specific axes.
|
||||
* @param paddings Vector of paddings in format
|
||||
|
||||
Reference in New Issue
Block a user