mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
aba6ad5da7
* support ncnn mmdet fcos.py * support fsaf * fix_lint * fix yapf * fix clang-format * Delete output_ncnn.jpg * Delete output_pytorch.jpg * remove comments and fix typo * fix blank line * Fix typo * Remove unnessisary comments * Add comment * Add comments
26 lines
484 B
C++
26 lines
484 B
C++
#ifndef LAYER_TOPK_H
|
|
#define LAYER_TOPK_H
|
|
|
|
#include "layer.h"
|
|
|
|
namespace mmlab {
|
|
|
|
class TopK : public ncnn::Layer {
|
|
public:
|
|
TopK();
|
|
virtual int load_param(const ncnn::ParamDict& pd);
|
|
virtual int forward(const std::vector<ncnn::Mat>& bottom_blobs,
|
|
std::vector<ncnn::Mat>& top_blobs,
|
|
const ncnn::Option& opt) const;
|
|
|
|
public:
|
|
int axis;
|
|
int largest;
|
|
int sorted;
|
|
int keep_dims;
|
|
};
|
|
|
|
} // namespace mmlab
|
|
|
|
#endif // LAYER_TOPK_H
|