Fix bug in val_onnx where grid strides were consumed by generator (#93) (#94)

corey-nm · web-flow · commit d2863d20e6f5 · 2022-08-22T10:54:25.000-04:00
diff --git 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))
     )

Original file line number	Diff line number	Diff line change
`@@ -155,7 +155,7 @@ def get_stride(yolo_pipeline: Pipeline) -> int:`
`155`	`155`	`if not isinstance(image_size, int):`
`156`	`156`	`image_size = image_size[0]`
`157`	`157`
`158`		`- grid_shapes = (`
	`158`	`+ grid_shapes = list(`
`159`	`159`	`get_tensor_dim_shape(model.graph.output[index], 2)`
`160`	`160`	`for index in range(1, len(model.graph.output))`
`161`	`161`	`)`