Skip to content

Commit 4c29eec

Browse files
authored
AC: remove workaround for NHWC layout for yolo (#3125)
1 parent 682ce9a commit 4c29eec

File tree

1 file changed

+1
-3
lines changed
  • tools/accuracy_checker/openvino/tools/accuracy_checker/adapters

1 file changed

+1
-3
lines changed

tools/accuracy_checker/openvino/tools/accuracy_checker/adapters/yolo.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -368,7 +368,7 @@ def parameters(cls):
368368
),
369369
'output_format': StringField(
370370
choices=['BHW', 'HWB'], optional=True,
371-
description="Set output layer format"
371+
description="Set output layer format", default='BHW',
372372
),
373373
'multiple_labels': BoolField(
374374
optional=True, default=False,
@@ -456,8 +456,6 @@ def process(self, raw, identifiers, frame_meta):
456456
detections = {'labels': [], 'scores': [], 'x_mins': [], 'y_mins': [], 'x_maxs': [], 'y_maxs': []}
457457
input_shape = list(meta.get('input_shape', {'data': (1, 3, 416, 416)}).values())[0]
458458
nchw_layout = input_shape[1] == 3
459-
if self.output_format is None:
460-
self.output_format = 'BHW' if nchw_layout else 'HWB'
461459
self.processor.width_normalizer = input_shape[3 if nchw_layout else 2]
462460
self.processor.height_normalizer = input_shape[2 if nchw_layout else 1]
463461
for layer_id, p in enumerate(prediction):

0 commit comments

Comments
 (0)