This works extends SfSNet
Following is directory structure for all the experiments
├─── data_loading.py: data loading and pre-processing methods <br>
├─── generate_dataset_csv.py: generate csv given dataset directory <br>
├─── interpolate.py: Create interpolation result from provided image directory <br>
├─── main.py: Train SfSNet on mix data. Need to provide both CelebA and Synthetic dataset directory<br>
├─── models.py: Definition of all the models used. Skip-Net and SfSNet <br>
├─── shading.py: Shading generation method from Normal and Spherical Harmonics <br>
├─── train.py: Train and test rountines <br>
├─── utils.py: Help rountines <br>
├─── data/: sample small scale dataset to be used. Use ON_SERVER=False in main_* scripts to use this dataset <br>
├─── pretrained/: pretrained model provided by SfSNet Author- 'load_model_from_pretrained' loads this model's weights to our model<br>
usage: main.py [-h] [--batch_size N] [--epochs N] [--lr LR]
[--wt_decay W] [--no_cuda] [--seed S]
[--read_first READ_FIRST] [--details DETAILS]
[--load_pretrained_model LOAD_PRETRAINED_MODEL]
[--syn_data SYN_DATA] [--celeba_data CELEBA_DATA]
[--log_dir LOG_DIR] [--load_model LOAD_MODEL]
SfSNet - Residual
optional arguments:
-h, --help show this help message and exit
--batch_size N input batch size for training (default: 8)
--epochs N number of epochs to train (default: 10)
--lr LR learning rate (default: 0.001)
--wt_decay W SGD momentum (default: 0.0005)
--no_cuda disables CUDA training
--seed S random seed (default: 1)
--read_first READ_FIRST
read first n rows (default: -1) from the dataset
This is helpful to load part of the data. Note that, internally
we change this to sample randomly with seed value 100
--details DETAILS Explaination of the run
String provided will be written into root log directory
We perform many experiments and then get lost on the results and what was this experiment for.
This txt file will help us understand what was the purpose of this experiment
--load_pretrained_model LOAD_PRETRAINED_MODEL
Pretrained model path for SfSNet based model provided by author
--syn_data SYN_DATA Synthetic Dataset path directory high level containing csv files
--celeba_data CELEBA_DATA
CelebA Dataset path high level containing train, test folder and csv files
--log_dir LOG_DIR Log Path
Where to log and store results, model
--load_model LOAD_MODEL
load model from following directory
Example
CUDA_VISIBLE_DEVICES=0,1 python main.py --epochs 100 --lr 0.0002 --batch_size 8 --read_first 10000
--log_dir ./results/skip_net/exp4/ --details 'Skipnet with normals'
SfSNet generated shading is generated using Normal and Spherical Harmonics which does not captures the full illumination details. Due to geometric imperfection and spherical harmonics inaccuracy, generated shading is not near to perfect. Later, shading is used along with albedo to reconstruct the image. Following image shows traditional shading model which is based on Normal and Spherical Harmonics.
We propose to introduce new shading layer to capture more flexible and comprehensive illumination effect that is not modelled by 27 dimensional spherical harmonics like SfSNet. We capture this representation directly using image features and residual block being used in SfSNet for albedo and normal. Later, we add this representation into SfSNet based generated shading. This shading layer is residue of illumination missed by SfSNet. Figure 6 shows new illumination model we are proposing
Following is the flow as compared to SfSNet we are adding.
Illumination model based on Normal and Spherical Harmonics is not accurate and cannot model the illumination estimation. Hence, we are proposing new illumination model, which works on top of Spherical Harmonics and Normal based Measure issue with reconstructin faces using Shading, Albedo, Normal and Spherical Harmonics is little variations in normal and spherical harmonics causing different illumination.
Estimating Normal and Spherical Harmonics with slight error causes different illumination and expected. In order to overcome this problem, we propose new shading model (let's say residue) which works along with traditional illumination model and captures the illumination details missed by traditional model.
We propose different methods to capture this residue using following methods. We also propose new illumination models, first to correct shading using latent lighting and second to generating shading directly without use of Spherical Harmonics.
1_Shading_Correcting works on priciple built on top of tradition shading generation using normal and spherical harmonics We generate Shading and then rectify it using latent ligthing generated from Normal, Albedo and Image features.
2_Latent_Shading_Gen works on principle to avoid generation of Spherical Harmonics and instead generate shading directly using Normal, Albedo and Image features. And then use this latent lighting along with Normal to generate shading
This method predicts shading residue and adds into traditional shading model. New shading residue model is capturing details missed by traditional shading model.
This method is next step of Shading residue method which add manual control by adding predicted residue into Shading and subtracting from Albedo. We can refer this method to be baesd on Robinhood principle- which takes from albedo and gives to shading.
Problem with shading residue method is domain gap of Synthetic and Real albedo being generated. Hence, we use GAN to generate robust and albedo from synthetic domain which has real ground truth.
due to lack of albedo loss, intensity of albedo is pushed into residue
In method 5, we train GAN based albedo generation and residue network along with each other. In this method, We first train GAN based albedo generation and then we fix GAN based method and then only train residue network.
Method 7 is slight variation of method 6 with different Discriminator
