Page MenuHomePhabricator

Numpy arrays not supported in vizLog.show_image_grid
Open, Needs TriagePublic

Description

Currently show_image_grid expects torch tensors.
If a numpy array should be displayed the array needs to be cast to torch tensors before:
images = torch.from_numpy(images)

Maybe it would be more convenient to have this check and cast inside the show_image_grid method

Event Timeline

wirkert added a project: Restricted Project.