未验证 提交 a70d9db9 编写于 作者: Z zhangyuqin1998 提交者: GitHub

reorder_prior_box (#52749)

* reorder_prior_box

* fix
上级 94afa5ab
...@@ -969,7 +969,7 @@ ...@@ -969,7 +969,7 @@
backward : pool3d_grad backward : pool3d_grad
- op : prior_box - op : prior_box
args : (Tensor input, Tensor image, float[] min_sizes, float[] aspect_ratios, float[] variances, float[] max_sizes = {}, bool flip=true, bool clip=true, float step_w=0.0, float step_h=0.0, float offset=0.5, bool min_max_aspect_ratios_order=false) args : (Tensor input, Tensor image, float[] min_sizes, float[] max_sizes = {}, float[] aspect_ratios = {}, float[] variances = {}, bool flip=true, bool clip=true, float step_w=0.0, float step_h=0.0, float offset=0.5, bool min_max_aspect_ratios_order=false)
output : Tensor(out), Tensor(var) output : Tensor(out), Tensor(var)
infer_meta : infer_meta :
func : PriorBoxInferMeta func : PriorBoxInferMeta
......
...@@ -2406,9 +2406,9 @@ inline void ExpandAspectRatios(const std::vector<float>& input_aspect_ratior, ...@@ -2406,9 +2406,9 @@ inline void ExpandAspectRatios(const std::vector<float>& input_aspect_ratior,
void PriorBoxInferMeta(const MetaTensor& input, void PriorBoxInferMeta(const MetaTensor& input,
const MetaTensor& image, const MetaTensor& image,
const std::vector<float>& min_sizes, const std::vector<float>& min_sizes,
const std::vector<float>& max_sizes,
const std::vector<float>& aspect_ratios, const std::vector<float>& aspect_ratios,
const std::vector<float>& variances, const std::vector<float>& variances,
const std::vector<float>& max_sizes,
bool flip, bool flip,
bool clip, bool clip,
float step_w, float step_w,
......
...@@ -381,9 +381,9 @@ void RepeatInterleaveWithTensorIndexInferMeta(const MetaTensor& x, ...@@ -381,9 +381,9 @@ void RepeatInterleaveWithTensorIndexInferMeta(const MetaTensor& x,
void PriorBoxInferMeta(const MetaTensor& input, void PriorBoxInferMeta(const MetaTensor& input,
const MetaTensor& image, const MetaTensor& image,
const std::vector<float>& min_sizes, const std::vector<float>& min_sizes,
const std::vector<float>& max_sizes,
const std::vector<float>& aspect_ratios, const std::vector<float>& aspect_ratios,
const std::vector<float>& variances, const std::vector<float>& variances,
const std::vector<float>& max_sizes,
bool flip, bool flip,
bool clip, bool clip,
float step_w, float step_w,
......
...@@ -24,9 +24,9 @@ void PriorBoxKernel(const Context& ctx, ...@@ -24,9 +24,9 @@ void PriorBoxKernel(const Context& ctx,
const DenseTensor& input, const DenseTensor& input,
const DenseTensor& image, const DenseTensor& image,
const std::vector<float>& min_sizes, const std::vector<float>& min_sizes,
const std::vector<float>& max_sizes,
const std::vector<float>& aspect_ratios, const std::vector<float>& aspect_ratios,
const std::vector<float>& variances, const std::vector<float>& variances,
const std::vector<float>& max_sizes,
bool flip, bool flip,
bool clip, bool clip,
float step_w, float step_w,
......
...@@ -111,9 +111,9 @@ void PriorBoxKernel(const Context& ctx, ...@@ -111,9 +111,9 @@ void PriorBoxKernel(const Context& ctx,
const DenseTensor& input, const DenseTensor& input,
const DenseTensor& image, const DenseTensor& image,
const std::vector<float>& min_sizes, const std::vector<float>& min_sizes,
const std::vector<float>& max_sizes,
const std::vector<float>& aspect_ratios, const std::vector<float>& aspect_ratios,
const std::vector<float>& variances, const std::vector<float>& variances,
const std::vector<float>& max_sizes,
bool flip, bool flip,
bool clip, bool clip,
float step_w, float step_w,
......
...@@ -23,9 +23,9 @@ void PriorBoxKernel(const Context& ctx, ...@@ -23,9 +23,9 @@ void PriorBoxKernel(const Context& ctx,
const DenseTensor& input, const DenseTensor& input,
const DenseTensor& image, const DenseTensor& image,
const std::vector<float>& min_sizes, const std::vector<float>& min_sizes,
const std::vector<float>& max_sizes,
const std::vector<float>& aspect_ratios, const std::vector<float>& aspect_ratios,
const std::vector<float>& variances, const std::vector<float>& variances,
const std::vector<float>& max_sizes,
bool flip, bool flip,
bool clip, bool clip,
float step_w, float step_w,
......
...@@ -25,9 +25,9 @@ void PriorBoxKernel(const Context& ctx, ...@@ -25,9 +25,9 @@ void PriorBoxKernel(const Context& ctx,
const DenseTensor& input, const DenseTensor& input,
const DenseTensor& image, const DenseTensor& image,
const std::vector<float>& min_sizes, const std::vector<float>& min_sizes,
const std::vector<float>& max_sizes,
const std::vector<float>& aspect_ratios, const std::vector<float>& aspect_ratios,
const std::vector<float>& variances, const std::vector<float>& variances,
const std::vector<float>& max_sizes,
bool flip, bool flip,
bool clip, bool clip,
float step_w, float step_w,
......
...@@ -20,9 +20,9 @@ KernelSignature PriorBoxOpArgumentMapping(const ArgumentMappingContext& ctx) { ...@@ -20,9 +20,9 @@ KernelSignature PriorBoxOpArgumentMapping(const ArgumentMappingContext& ctx) {
return KernelSignature("prior_box", return KernelSignature("prior_box",
{"Input", "Image"}, {"Input", "Image"},
{"min_sizes", {"min_sizes",
"max_sizes",
"aspect_ratios", "aspect_ratios",
"variances", "variances",
"max_sizes",
"flip", "flip",
"clip", "clip",
"step_w", "step_w",
......
...@@ -25,9 +25,9 @@ def python_prior_box( ...@@ -25,9 +25,9 @@ def python_prior_box(
input, input,
image, image,
min_sizes, min_sizes,
max_sizes=None,
aspect_ratios=[1.0], aspect_ratios=[1.0],
variances=[0.1, 0.1, 0.2, 0.2], variances=[0.1, 0.1, 0.2, 0.2],
max_sizes=None,
flip=False, flip=False,
clip=False, clip=False,
step_w=0, step_w=0,
......
...@@ -524,9 +524,9 @@ def prior_box( ...@@ -524,9 +524,9 @@ def prior_box(
input, input,
image, image,
min_sizes, min_sizes,
max_sizes,
aspect_ratios, aspect_ratios,
variance, variance,
max_sizes,
flip, flip,
clip, clip,
step_w, step_w,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册