FSnet

Repository for the paper "Light-Field Reconstruction and Depth Estimation from Focal Stack Images Using Convolutional Neural Networks" ICASSP 2020 [Paper].

We use the "ViewNet_train_invC.ipynb" to train the network estimating all-in-focus image from focal stack and use the "DIBR_train_invC.ipynb" for training depth estimation and light field reconstruction networks.

Name		Name	Last commit message	Last commit date
Latest commit History 12 Commits
LF_EPrecon		LF_EPrecon
DIBR_joint_train_invC.ipynb		DIBR_joint_train_invC.ipynb
DIBR_modules.ipynb		DIBR_modules.ipynb
DIBR_modules.py		DIBR_modules.py
DIBR_test_invC.ipynb		DIBR_test_invC.ipynb
DIBR_train.ipynb		DIBR_train.ipynb
DIBR_train_invC.ipynb		DIBR_train_invC.ipynb
Data_generation.ipynb		Data_generation.ipynb
README.md		README.md
Test-DIBR.ipynb		Test-DIBR.ipynb
ViewNet_train.ipynb		ViewNet_train.ipynb
ViewNet_train_invC.ipynb		ViewNet_train_invC.ipynb
data_utils.py		data_utils.py
generateLFh5.ipynb		generateLFh5.ipynb
generateSAI.ipynb		generateSAI.ipynb
paper_plot.ipynb		paper_plot.ipynb
test_invC.ipynb		test_invC.ipynb
train.ipynb		train.ipynb
train_invC.ipynb		train_invC.ipynb
unet_layers.py		unet_layers.py
visualize.py		visualize.py

macrohuang1993/FSnet

Folders and files

Latest commit

History

Repository files navigation

FSnet

About

Resources

Stars

Watchers

Forks

Languages