[Tests/Fix] Use NHWC format to compare golden data
authorDongju Chae <dongju.chae@samsung.com>
Wed, 26 May 2021 08:19:21 +0000 (17:19 +0900)
committer채동주/On-Device Lab(SR)/Staff Engineer/삼성전자 <dongju.chae@samsung.com>
Wed, 26 May 2021 09:26:13 +0000 (18:26 +0900)
This patch uses NHWC format to compare golden data.

Signed-off-by: Dongju Chae <dongju.chae@samsung.com>
tests/utils/ne_test_utils.cc

index 34a777a82efaa7074b6ea8f9e0b4cefbcfa26d7b..a41b4b528dc2c27eedb79c3708399fbecccb5f8d 100644 (file)
@@ -674,15 +674,19 @@ UtilTRIV2::set_data_info (npubin_meta *meta, uint32_t model_id) {
   tensors_data_info info_in;
   tensors_data_info info_out;
 
+  /**
+   * The output data with TRIV2 format may have trash values at the tail.
+   * So, layout converting is unavoidable for golden data comparison.
+   */
   info_in.num_info = meta->input_seg_num;
   for (uint32_t idx = 0; idx < info_in.num_info; idx++) {
-    info_in.info[idx].layout = DATA_LAYOUT_TRIV2;
+    info_in.info[idx].layout = DATA_LAYOUT_NHWC;
     info_in.info[idx].type = DATA_TYPE_QASYMM8;
   }
 
   info_out.num_info = meta->output_seg_num;
   for (uint32_t idx = 0; idx < info_out.num_info; idx++) {
-    info_out.info[idx].layout = DATA_LAYOUT_TRIV2;
+    info_out.info[idx].layout = DATA_LAYOUT_NHWC;
     info_out.info[idx].type = DATA_TYPE_QASYMM8;
   }