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.
114 lines
4.7 KiB
114 lines
4.7 KiB
/**
|
|
* Copyright 2020 Huawei Technologies Co., Ltd
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#ifndef GE_GRAPH_PASSES_FOLDING_KERNEL_SSD_PRIOR_BOX_KERNEL_H_
|
|
#define GE_GRAPH_PASSES_FOLDING_KERNEL_SSD_PRIOR_BOX_KERNEL_H_
|
|
|
|
#include <memory>
|
|
#include <vector>
|
|
|
|
#include "inc/kernel.h"
|
|
|
|
namespace ge {
|
|
class SsdPriorboxKernel : public Kernel {
|
|
public:
|
|
/**
|
|
* Entry of the SsdPriorboxKernel optimizer
|
|
* @param [in] node: Input Node
|
|
* @return SUCCESS: node output compute success
|
|
* @return OTHERS: Execution failed
|
|
* @author
|
|
*/
|
|
Status Compute(const NodePtr &node, std::vector<GeTensorPtr> &v_output) override;
|
|
|
|
private:
|
|
/**
|
|
* Get specific op_desc attr value
|
|
* @param [in] op_desc: Input op_desc
|
|
* @param [in/out] img_width: img_width attr_value
|
|
* @param [in/out] img_height: img_height attr_value
|
|
* @param [in/out] step_h: step_h attr_value
|
|
* @param [in/out] step_w: step_w attr_value
|
|
* @param [in/out] layer_width: layer_width attr_value
|
|
* @param [in/out] layer_height: layer_height attr_value
|
|
* @return SUCCESS: node get attr value success
|
|
* @return OTHERS: Execution failed
|
|
* @author
|
|
*/
|
|
Status GetPriorSizeParam(const OpDescPtr &op_desc, int &img_width, int &img_height, float &step_w, float &step_h,
|
|
int &layer_width, int &layer_height);
|
|
/**
|
|
* Get specific op_desc size attr value,min_size_num etc.
|
|
* @param [in] op_desc: Input op_desc
|
|
* @param [in/out] offset: offset attr_value
|
|
* @param [in/out] clip: clip attr_value
|
|
* @return SUCCESS: get attr success
|
|
* @return OTHERS: Execution failed
|
|
* @author
|
|
*/
|
|
Status GetPriorOtherParam(const OpDescPtr &op_desc, float &offset, bool &clip);
|
|
/**
|
|
* Get specific op_desc list attr value,min_size_list etc.
|
|
* @param [in] op_desc: Input op_desc
|
|
* @param [in/out] min_size_list: min_size_list attr_value
|
|
* @param [in/out] max_size_list: max_size_list attr_value
|
|
* @param [in/out] aspect_ratio_list: aspect_ratio_list attr_value
|
|
* @param [in/out] variance_list: variance_list attr_value
|
|
* @param [in/out] clip: clip attr_value
|
|
* @return SUCCESS: get list attr success
|
|
* @return OTHERS: Execution failed
|
|
* @author
|
|
*/
|
|
Status GetPriorListParam(const OpDescPtr &op_desc, vector<float> &min_size_list, vector<float> &max_size_list,
|
|
vector<float> &aspect_ratio_list, vector<float> &variance_list);
|
|
/**
|
|
* set variance param to output_data.
|
|
* @param [in] variance: variance list
|
|
* @param [in] dim: output_data second channel offset
|
|
* @param [in] layer_height: layer_height
|
|
* @param [in] num_priors: num_priors
|
|
* @param [in/out] output_data: output_data
|
|
* @return SUCCESS: set variance success
|
|
* @return OTHERS: Execution failed
|
|
* @author
|
|
*/
|
|
Status SetVariance(const vector<float> &variance, const int dim, const int32_t layer_height,
|
|
const int32_t layer_width, const int num_priors, float *output_data);
|
|
/**
|
|
* get num priors and dim size.
|
|
* @param [in] aspect_ratios_size: aspect_ratio_list size
|
|
* @param [in] min_sizes_size: min_size_list size
|
|
* @param [in] max_sizes_size: max_size_list size
|
|
* @param [in] layer_width: layer_width
|
|
* @param [in] layer_height: layer_height
|
|
* @param [in/out] num_priors: num_priors
|
|
* @param [in/out] dim_size: dim_size
|
|
* @return SUCCESS: set variance success
|
|
* @return OTHERS: Execution failed
|
|
* @author
|
|
*/
|
|
Status GetNumPriorAndDimSize(uint32_t aspect_ratios_size, uint32_t min_sizes_size, uint32_t max_sizes_size,
|
|
int layer_width, int layer_height, int &num_priors, int &dim_size) const;
|
|
void DataCalulate(float x, float y, float box_x, float box_y, int img_x, int img_y, vector<float> &result);
|
|
std::unique_ptr<float[]> BoundaryCalulate(int dim_size, int layer_width, int layer_height, float step_width,
|
|
float step_height, int img_width, int img_height, float offset,
|
|
vector<float> min_sizes, vector<float> max_sizes,
|
|
vector<float> aspect_ratios);
|
|
};
|
|
} // namespace ge
|
|
|
|
#endif // GE_GRAPH_PASSES_FOLDING_KERNEL_SSD_PRIOR_BOX_KERNEL_H_
|