-
Notifications
You must be signed in to change notification settings - Fork 6
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
How to obtain the pretrained pkls for dfvnet and aifnet #6
Comments
Hello dusk1037, Thanks a lot for your interest! You can run the training code to train your own models. It is very easy and straightforward, and I believe you can get a deeper understanding by doing this. For this project, the aberration-aware training (AAT) method is more important. You should not expect the model to function for different lenses since the aberrations are different! :-) Best, |
Thank you for replying, |
No. The model weights are highly related to the lens and the pixel size, which affect the 4D PSFs. That also means if you change the lens or the camera sensor, the pretrained model will not work. The pretrained weights in [20] and [22] consider lenses without any aberration, therefore will not work for the RF50mm lens we are using. I will upload our pretrained weights these days. Or you can run the training code to train the models by yourself. :-) |
Huge thanks for replying, |
Hello dusk1037, You can run If you have any questions, please feel free to ask. Hope you enjoy the research of deep learning and computational imaging!! |
I happened to encounter a Runtime error:CUDA out of memory when I tried to run the file "2_aber_aware_dff_aif.py", while the original traceback leads me to line 370 in ./dff/Aifnet.py, which consists of a annotation that says "FIXME: bug here", can it be resolved? |
Hello, This is a bug coming from the original AiFNet architecture. Reducing the stack size should resolve it. |
Thank you for your timely replying! |
No description provided.
The text was updated successfully, but these errors were encountered: