未验证 提交 8bdcd157 编写于 作者: S ShawnXsw 提交者: GitHub

fix dark postprocess in cpp infer deploy( #5751) (#5815)

上级 c1fe1dd2
......@@ -193,7 +193,7 @@ void get_final_preds(std::vector<float>& heatmap,
int px = int(coords[j * 2] + 0.5);
int py = int(coords[j * 2 + 1] + 0.5);
if(DARK && px > 1 && px < heatmap_width - 2){
if(DARK && px > 1 && px < heatmap_width - 2 && py > 1 && py < heatmap_height - 2){
dark_parse(heatmap, dim, coords, px, py, index, j);
}
else{
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册