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

reorder_prior_box (#52749)

* reorder_prior_box

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