Skip to content

This repository contains code to generate stereo (Side by side) image from a single image.

Notifications You must be signed in to change notification settings

UX3D-nopper/stereo-image-generation

 
 

Repository files navigation

Stereo (Side by Side) Image Generation from Single Image

Utilizes AI to instantly fully convert 2D content into stereo 3D

image

Fig) Input image(Left) / Output(Middle) / 3D effect(Right)

Introduce

With the rapid of autostereoscopic 3D monitors, through the specialized optical lens and eye-tracking technology delivers users have an entirely new stereoscopic 3D visualization experience. However, it only works with 3D content inputs (Stereoscopic Images). Such as side-by-side images. But most image or video on the internet is 2D single view content. Making the technology difficult to popularize. In order to solve this problem, this project utilizes "Deep Learning" and "Computer Vision" to enable conversion of 2D content into stereo 3D content.

Changelog

  • [May 2023] Fixes regarding links and filenames
  • [Aug 2020] Release C++ and cython version
  • [Aug 2020] Initial release of stereo image generation base on MiDaS v2.0

Setup

  1. Download the model weights model-f45da743.pt and place the file in the root folder.

  2. Set up dependencies with CUDA:

    pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

    The code was tested with Cuda 10.1, Python 3.6.6, PyTorch 1.6.0, Torchvision 0.7.0 and OpenCV 3.4.0.12.

Usage

  1. Place input images in the folder example e.g. from here Male lion in Okonjima, Namibia

  2. Run the model:

    (Generate depth map from image)

    python depth_estimation_image.py

    (Generate depth map from camera)

    python depth_estimation_cam.py

    (Generate stereo image from image)

    python stereo_generation_image.py

    (Generate stereo image from camera)

    python stereo_generation_cam.py
  3. The resulting depth maps are written to the depth folder.

    The resulting stereo image are written to the stereo folder.

Acknowledgments

Our code builds upon Intel MiDaS

About

This repository contains code to generate stereo (Side by side) image from a single image.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 92.0%
  • C++ 5.2%
  • Cython 1.9%
  • CMake 0.9%