|
|
@ -74,6 +74,9 @@ class MultiClassNMSOp : public framework::OperatorWithKernel {
|
|
|
|
} else {
|
|
|
|
} else {
|
|
|
|
ctx->SetOutputDim("Out", {-1, box_dims[2] + 2});
|
|
|
|
ctx->SetOutputDim("Out", {-1, box_dims[2] + 2});
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!ctx->IsRuntime()) {
|
|
|
|
|
|
|
|
ctx->SetLoDLevel("Out", std::max(ctx->GetLoDLevel("BBoxes"), 1));
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
protected:
|
|
|
|
protected:
|
|
|
@ -493,6 +496,9 @@ class MultiClassNMS2Op : public MultiClassNMSOp {
|
|
|
|
} else {
|
|
|
|
} else {
|
|
|
|
ctx->SetOutputDim("Index", {-1, 1});
|
|
|
|
ctx->SetOutputDim("Index", {-1, 1});
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!ctx->IsRuntime()) {
|
|
|
|
|
|
|
|
ctx->SetLoDLevel("Index", std::max(ctx->GetLoDLevel("BBoxes"), 1));
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
};
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|