import os
import numpy as np
from skimage.io import imsave
import neural_renderer as nr
import pdb

renderer = nr.Renderer(camera_mode='look_at')

vertices, faces, textures = nr.load_obj(
        'straight.obj', load_texture=True, texture_size=16)

renderer.eye = nr.get_points_from_angles(2.7, 15.0, 0.0)
images = renderer.render(vertices[None, :, :], faces[None, :, :], textures[None, :, :, :, :, :]).permute(0,2,3,1).detach().cpu().numpy()

# images[np.isnan(images)] = 0.
# images[images < -1.0] = -1.0
# images[images > 1.0] = 1.0
imsave('pytorch_version_mug.png',images[0])