Skip to content

Generative View Synthesis: from Single-view Semantics to Novel-view Images, Habtegebrial et al. NeurIPS 2020

Notifications You must be signed in to change notification settings

tedyhabtegebrial/gvsnet

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

43 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Generative View Synthesis

This repository contains code accompanying the paper
Generative View Synthesis: From Single-view Semantics to Novel-view Images. Tewodros Habtegebrial , Varun Jampani , Orazio Gallo , Didier Stricker
Thirty-fourth Conference on Neural Information Processing Systems (NeurIPS-2020)
The project page can be found here

Dependencies

This code was develped with python3.6

scikit-image
torch # tested with versions 1.7.1, but it might work with older versions as well
tqdm

Instructions

Runnig the demo code
Pre-trained models

Download a pre-trained model for the CARLA dataset from here. Extract it to the folder pre_trained_models/carla

Download sample test dataset

Download 34 sample scenes (from the CARLA dataset) for demo purpose link. Extract the dataset to the folder datasets/carla_samples

Run the demo code
CUDA_VSIBLE_DEVICES=0 python demo.py \
    --dataset=carla_samples \
    --mode=demo \
    --movement_type=circle \
    --data_path=./datasets/carla_samples \
    --output_path=./output/carla_samples \
    --pre_trained_gvsnet=./pre_trained_models/carla/gvsnet_model.pt \
    --style_path=./data/sample_styles/carla_1.png \

Controlling style of the generated views

A style image can be pased with the following flag --style_path. If not given the color image of input view is used as a style image.

Training the GVSNet model

Datasets

Fow downloading the datasets used in our experiments please read instructions here datasets Please check the scripts folder training scripts and check options.py for a list of commandline arguments.

Traing the Semantic Uplifting Model

Recommended batch sizes and number of epochs

  • sun model batch_size=12 and above, num_epochs=30
cd scripts
./train_sun_carla

Traing the Layered Translation and Appearance Decoder Networks

Recommended batch sizes and number of epochs

  • ltn+adn batch_size=16 and above, num_epochs=20
cd scripts
./train_gvs_carla.sh

Acknowledgments: This repo builds upon the SPADE repository from NVIDIA

About

Generative View Synthesis: from Single-view Semantics to Novel-view Images, Habtegebrial et al. NeurIPS 2020

Topics

Resources

Stars

Watchers

Forks