joh-schb/PlantLDM

Improve visualization

Closed this issue · 1 comments

joh-schb commented

Make visualization faster and nicer

  • avoid for loop with vectorization
  • use pillow
  • Include option to save images to disk
joh-schb commented

The problem with $64 \times 64$ images is that they are too small in tensorboard. We would need to resize them in order to see them properly.

I added a function for getting the image, maybe we will change it later.