Git Product home page Git Product logo

adampi's Introduction

News

  • 2023/7/31 We release the complete training code for our method on the COCO dataset!
  • 2022/8/4 We release the code and the pretrained model for AdaMPI. Use it to create 3D photo from single in-the-wild 2D images!
  • 2022/7/22 We release the code and the pretrained inpainting model for the Warp-Back strategy.

AdaMPI

This is a PyTorch implementation of the following paper:

Single-View View Synthesis in the Wild with Learned Adaptive Multiplane Images, SIGGRAPH 2022.

Yuxuan Han, Ruicheng Wang and Jiaolong Yang

Project Page | Video | Paper

Abstract: This paper deals with the challenging task of synthesizing novel views for in-the-wild photographs. Existing methods have shown promising results leveraging monocular depth estimation and color inpainting with layered depth representations. However, these methods still have limited capability to handle scenes with complex 3D geometry. We propose a new method based on the multiplane image (MPI) representation. To accommodate diverse scene layouts in the wild and tackle the difficulty in producing high-dimensional MPI contents, we design a network structure that consists of two novel modules, one for plane depth adjustment and another for depth-aware color prediction. The former adjusts the initial plane positions using the RGBD context feature and an attention mechanism. Given adjusted depth values, the latter predicts the color and density for each plane separately with proper inter-plane interactions achieved via a feature masking strategy. To train our method, we construct large-scale stereo training data using only unconstrained single-view image collections by a simple yet effective warp-back strategy.

Document

This repo contain two parts:

  • The train and inference code for the AdaMPI. You can use it to generate 3D photo from single-view in-the-wild image input, as shown in the teaser gif. See AdaMPI.md for more details.

  • The clean and abstract implementation of the Warp-Back strategy, you can use it to generate multi-view training data to train your own view synthesis model (e.g. SynSin, WorldSheet) using only single-view image collections like COCO dataset. See Warp-Back.md for more details.

Contact

If you have any questions, please contact Yuxuan Han ([email protected]).

License and Citation

This repository can only be used for personal/research/non-commercial purposes. Please cite the following paper if this model helps your research:

@inproceedings{han2022single,
    author = {Han, Yuxuan and Wang, Ruicheng and Yang, Jiaolong},
    title = {Single-View View Synthesis in the Wild with Learned Adaptive Multiplane Images},
    booktitle = {ACM SIGGRAPH},
    year={2022}
}

Acknowledgments

  • The code for inference and training is heavily borrowed from MINE, we thank the author for their great effort.
  • We thank Varun Jampani and Meng-Li Shih for helpful suggestions and discussions to evaluate their method.

adampi's People

Contributors

easternjournalist avatar yxuhan avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo 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.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.