Skip to content

Python implementation of convertion between equirectangular, cubemap and perspective. (equirect2cube, cube2equirect, equirect2perspec)

License

Notifications You must be signed in to change notification settings

sunset1995/py360convert

Repository files navigation

py360convert

Features of this project:

  • Conversion between cubemap and equirectangular:
  • Conversion between Equirectangular and planar:
  • Pure python implementation and depend only on numpy and scipy.
  • Vectorization implementation:
    • If opencv is installed, py360convert will automatically use it to accelerate computations (several times speedup).

Install

pip install py360convert

You can use the library with import py360convert and through the command line tool convert360.

Command line examples

The convert360 command line tool can be run like so. Use convert360 -h for details. The convert360 python script is also doubles as ab example code for how to use this as a package in your code.

convert360 e2c assets/example_input.png out.png --size 200
Input Equirectangular Output Cubemap

convert360 c2e assets/example_e2c.png out.png --width 800 --height 400
Input Cubemap Output Equirectangular

You can see the blurring artifacts in the polar region because the equirectangular in above figure are resampled twice (e2c then c2e).


convert360 e2p assets/example_input.png out.png --width 300 --height 300 --yaw 120 --pitch 23
Input Equirectangular Output Perspective

Doc

c2e(cubemap, h, w, cube_format='dice')

Convert the given cubemap to equirectangular.
Parameters:

  • cubemap: Numpy array or list/dict of numpy array (depend on cube_format).

  • h: Output equirectangular height.

  • w: Output equirectangular width.

  • mode:str: Interpolation method; typically bilinear or nearest. Valid options: "nearest", "linear", "bilinear", "biquadratic", "quadratic", "quad", "bicubic", "cubic", "biquartic", "quartic", "biquintic", "quintic".

  • cube_format: Options: 'dice' (default), 'horizon' or 'dict' or 'list'. Indicates the format of the given cubemap.

    • Say that each face of the cube is in shape of 256 (width) x 256 (height)

    • 'dice': a numpy array in shape of 1024 x 768 like below example

    • 'horizon': a numpy array in shape of 1536 x 256 like below example

    • 'list': a list with 6 elements each of which is a numpy array in the shape of 256 x 256. It's just converted from 'horizon' format with one line of code: np.split(cube_h, 6, axis=1).

    • 'dict': a dict with 6 elements with keys 'F', 'R', 'B', 'L', 'U', 'D' each of which is a numpy array in shape of 256 x 256.

    • Please refer to the source code if you still have any questions about the conversion between formats.

e2c(e_img, face_w=256, mode='bilinear', cube_format='dice')

Convert the given equirectangular to cubemap.
Parameters:

  • e_img: NDArray: Numpy array with shape [H, W, C].
  • face_w: int: The width of each cube face.
  • mode:str: See c2e.
  • cube_format:str: See c2e.

e2p(e_img, fov_deg, u_deg, v_deg, out_hw, in_rot_deg=0, mode='bilinear')

Take perspective image from given equirectangular. Parameters:

  • e_img: Numpy array with shape [H, W, C].
  • fov_deg: Field of view given in int or tuple (h_fov_deg, v_fov_deg).
  • u_deg: Horizontal viewing angle in range [-pi, pi]. (- Left / + Right).
  • v_deg: Vertical viewing angle in range [-pi/2, pi/2]. (- Down/ + Up).
  • out_hw: Output image (height, width) in tuple.
  • in_rot_deg: Inplane rotation.
  • mode: bilinear or nearest.

Example:

import numpy as np
from PIL import Image
import py360convert

cube_dice = np.array(Image.open('assets/demo_cube.png'))

# You can make conversion between supported cubemap format
cube_h = py360convert.cube_dice2h(cube_dice)  # the inverse is cube_h2dice
cube_dict = py360convert.cube_h2dict(cube_h)  # the inverse is cube_dict2h
cube_list = py360convert.cube_h2list(cube_h)  # the inverse is cube_list2h
print('cube_dice.shape:', cube_dice.shape)
print('cube_h.shape:', cube_h.shape)
print('cube_dict.keys():', cube_dict.keys())
print('cube_dict["F"].shape:', cube_dict["F"].shape)
print('len(cube_list):', len(cube_list))
print('cube_list[0].shape:', cube_list[0].shape)

Output:

cube_dice.shape: (768, 1024, 3)
cube_h.shape: (256, 1536, 3)
cube_dict.keys(): dict_keys(['F', 'R', 'B', 'L', 'U', 'D'])
cube_dict["F"].shape: (256, 256, 3)
len(cube_list): 6
cube_list[0].shape: (256, 256, 3)

About

Python implementation of convertion between equirectangular, cubemap and perspective. (equirect2cube, cube2equirect, equirect2perspec)

Topics

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages