Skip to content
Snippets Groups Projects
Commit b98c9a9d authored by André Anjos's avatar André Anjos :speech_balloon:
Browse files

[engine.segment.dumper,viewer] Allow dumped annotations to be viewed

parent c2a11025
No related branches found
No related tags found
No related merge requests found
......@@ -63,10 +63,11 @@ def view(
target: numpy.typing.NDArray[numpy.bool_] = numpy.array(f["target"])
mask: numpy.typing.NDArray[numpy.bool_] = numpy.array(f["mask"])
pred: numpy.typing.NDArray[numpy.float32] = numpy.empty(
(0,), dtype=numpy.float32
)
if "prediction" in f:
pred: numpy.typing.NDArray[numpy.float32] = numpy.array(f["prediction"])
else:
pred = numpy.empty((0,), dtype=numpy.float32)
pred = numpy.array(f["prediction"])
image *= mask
target = numpy.logical_and(target, mask)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment