NVlabs / imaginaire

NVIDIA's Deep Imagination Team's PyTorch Library
Other
4.02k stars 449 forks source link

Single image output in inference? #91

Closed DanAmador closed 3 years ago

DanAmador commented 3 years ago

Good day,

is there a way to automatically get the output for a paired image inference as a single output of the resulting inference instead of the paired video with the labels and real image?

arunmallya commented 3 years ago

I suppose you can try editing the function doing the saving.