From ffa72fc9793ceabfed9d339aa9238758abe47979 Mon Sep 17 00:00:00 2001 From: Liubov Batanina Date: Tue, 24 Dec 2019 09:45:27 +0300 Subject: [PATCH] Refactoring --- samples/dnn/human_parsing.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/samples/dnn/human_parsing.py b/samples/dnn/human_parsing.py index 84d0663871..467a19a3b1 100644 --- a/samples/dnn/human_parsing.py +++ b/samples/dnn/human_parsing.py @@ -117,8 +117,7 @@ def postprocess(out): tail_list_rev.extend([tail_list[15], tail_list[14], tail_list[17], tail_list[16], tail_list[19], tail_list[18]]) tail_output_rev = np.stack(tail_list_rev, axis=0) tail_output_rev = np.flip(tail_output_rev, axis=2) - raw_output_all = np.mean(np.stack([head_output, tail_output_rev], axis=0), axis=0, keepdims=False) - raw_output_all = np.expand_dims(raw_output_all, axis=0) + raw_output_all = np.mean(np.stack([head_output, tail_output_rev], axis=0), axis=0, keepdims=True) raw_output_all = np.argmax(raw_output_all, axis=1) raw_output_all = raw_output_all.transpose(1, 2, 0) return raw_output_all