apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

handle empty results

This commit is contained in:
Sean Sube 2024-01-05 21:23:05 -06:00
parent 058a0f1c0a
commit 1a8d538bfe
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
api/onnx_web/chain/result.py
View File

@ -328,13 +328,13 @@ class StageResult:
def size(self) -> Size:
if self.images is not None:
return Size(
max([image.width for image in self.images]),
max([image.height for image in self.images]),
max([image.width for image in self.images], default=0),
max([image.height for image in self.images], default=0),
)
elif self.arrays is not None:
return Size(
max([array.shape[0] for array in self.arrays]),
max([array.shape[1] for array in self.arrays]),
max([image.height for image in self.images], default=0),
max([image.height for image in self.images], default=0),
) # TODO: which fields within the shape are width/height?
else:
return Size(0, 0)