search

LiUzHiAn / hf2vad

MIT License
121 stars 23 forks source link

about visualization code #13

Open Smallwhite999 opened 2 years ago

Smallwhite999 commented 2 years ago

Hello, author, your work is very good and meaningful. Can you give the visualization related code in the visualization process, such as the visual comparison between reconstruction and prediction, such as Figure 6 and Figure 11, thanks!

LiUzHiAn commented 2 years ago

matplotlib.pyplot would help.

Smallwhite999 commented 2 years ago

matplotlib.pyplot would help.

If you can, can you share your visualization code? In order to further understand and learn your paper, thank you very much！

LiUzHiAn commented 2 years ago

Basically, you just need to visualize the reconstruction or prediction results with respect to the input data.

For the flows, please check https://github.com/LiUzHiAn/hf2vad/blob/0f34301dd291562d9377621d5c1fcc6550f571fc/utils/flow_utils.py#L74-L111 for help and do some basic image oprations. And for the images, just calculate the MSE or MAE, then plot them using pyplot. Thanks.

wolfworld6 commented 2 years ago

Basically, you just need to visualize the reconstruction or prediction results with respect to the input data.

For the flows, please check

https://github.com/LiUzHiAn/hf2vad/blob/0f34301dd291562d9377621d5c1fcc6550f571fc/utils/flow_utils.py#L74-L111

for help and do some basic image oprations. And for the images, just calculate the MSE or MAE, then plot them using pyplot. Thanks.

when I run the code，I got this error “numpy.core._exceptions.MemoryError: Unable to allocate 426. TiB for an array with shape (117107452696000,) and data type float32”，how is this possible？ I have checked that the .npy shape is (240, 360, 2)。