Skip to content

Commit 81df1a3

Browse files
authored
Update Yolo Adapter to Fix Feature Scale Detection (#1062)
As mentioned in issue #1061
1 parent 9453e7e commit 81df1a3

File tree

1 file changed

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

1 file changed

+1
-1
lines changed

tools/accuracy_checker/accuracy_checker/adapters/yolo.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -274,7 +274,7 @@ def get_anchors_offset(x, num, anchors):
274274
return int((num * 2) * (len(anchors) / (num * 2) - 1 - math.log2(x / 13)))
275275

276276
def parse_yolo_v3_results(prediction, threshold, w, h, det, layer_id):
277-
cells_x, cells_y = prediction.shape[1:]
277+
cells_y, cells_x = prediction.shape[1:]
278278
anchors = self.masked_anchors[layer_id] if self.masked_anchors else self.anchors
279279
num = len(anchors) // 2 if self.masked_anchors else self.num
280280
prediction = prediction.flatten()

0 commit comments

Comments
 (0)