daniel03c1 / masked_wavelet_nerf

MIT License
79 stars 5 forks source link

Compression - Decompression problem #6

Closed PhmHiu closed 1 year ago

PhmHiu commented 1 year ago

I'm trying to reproduce your result by running your code followed by your instructions. I trained the model and everything was fine. However, when I compress the model and decompress it to evaluate, the evaluation results are very low (1X PSNR compared to 3x PSNR of the original trained model). Can you check if anything was wrong? Thank you

benhenryL commented 1 year ago

Hi @PhmHiu , Thank you for your attention on our work! We found we omitted codes for applying Inverse DWT to model before the inference when we decompress the model. I just updated the code and please pull the updated codes. Thank you for pointing it out!

PhmHiu commented 1 year ago

Thank you very much! It's working now