From 4c971fa3422a61b2b1915921964b1166f6ef1b42 Mon Sep 17 00:00:00 2001 From: Corey Lowman Date: Mon, 22 Aug 2022 10:39:43 -0400 Subject: [PATCH] Fix bug in val_onnx where grid strides were consumed by generator --- val_onnx.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/val_onnx.py b/val_onnx.py index f18bad612792..2df802f6f680 100644 --- a/val_onnx.py +++ b/val_onnx.py @@ -155,7 +155,7 @@ def get_stride(yolo_pipeline: Pipeline) -> int: if not isinstance(image_size, int): image_size = image_size[0] - grid_shapes = ( + grid_shapes = list( get_tensor_dim_shape(model.graph.output[index], 2) for index in range(1, len(model.graph.output)) )