Comments (6)
I also want to see the training code.
from lenslesslearning.
since there is no code for training, I write the training code by myself using DiffuserCam Dataset, but the parameter le_admm.mu2 will get a gradient of nan during the training process, thus causing the failure in training. Iām not sure whether there are something wrong with my training code, so can you public your training code?
since there is no code for training, I write the training code by myself using DiffuserCam Dataset, but the parameter le_admm.mu2 will get a gradient of nan during the training process, thus causing the failure in training. Iām not sure whether there are something wrong with my training code, so can you public your training code?
do you have your training code?
from lenslesslearning.
I am also having this issue
from lenslesslearning.
Hi @zhangyingerjelly, @HkDzl, @arpanpoudel, we've released training code to reproduce this (and other features) here: https://github.com/LCAV/LenslessPiCam/blob/main/scripts/recon/train_unrolled.py
By default it uses this configuration (sorry if there's a lot going on there!)
This script is part of a broader package for lensless imaging (measurement, reconstruction algorithms, simulation, and evaluation): GitHub, documentation
It's a work in progress, but hope it helps and happy to support setup / get your feedback!
from lenslesslearning.
@ebezzam I have gone through your training recipe and saw that you had downsampled the images to 135,240 during training. Does this downsampling effect the reconstruction? considering the multiplexing property. I also saw that you had changed the image pair from BGR to RGB before training, will this affect the reconstruction?
from lenslesslearning.
@arpanpoudel thanks for looking into the code!
Yes by default we downsample by a factor 2 along each dimension, but if you set this factor to 1, you can keep the original resolution of the measurements. My impression is that the original authors already downsampled the data so that the dataset is more manageable (because the provided PSF has 4x the resolution of the measurements). But you're right, in general downsampling affects reconstruction, higher resolution measurements can allow higher resolution reconstructions but at a computational cost (larger FFTs). You can find some of our results here: Figure 5.4 compares different reconstruction approaches.
Regarding conversion from BGR to RGB, the original authors also do this but at the output. We do it before reconstruction but it doesn't make a difference as each color channel is handled independently during reconstruction.
from lenslesslearning.
Related Issues (5)
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
š Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. ššš
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ā¤ļø Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from lenslesslearning.