|
import argparse |
|
import os |
|
import subprocess |
|
import sys |
|
from pathlib import Path |
|
|
|
from PIL import Image |
|
|
|
|
|
def render_asset( |
|
asset_path, |
|
output_dir, |
|
nviews=24, |
|
img_resolution=512, |
|
): |
|
""" |
|
Render given asset into output_dir and return the saved image paths. |
|
Assumes that blender is installed and is in your path. |
|
|
|
nviews : number of views to render |
|
img_resolution : resolution of each rendered view in pixels |
|
""" |
|
|
|
curr_file_path = __file__ |
|
curr_dir = os.path.dirname(curr_file_path) |
|
|
|
command = [ |
|
"blender", |
|
"--background", |
|
"-noaudio", |
|
"--python", |
|
f"{curr_dir}/blender_script.py", |
|
"--", |
|
"--object_path", |
|
asset_path, |
|
"--num_renders", |
|
str(nviews), |
|
"--output_dir", |
|
output_dir, |
|
"--render_resolution", |
|
str(img_resolution), |
|
"--transparent_background", |
|
"--engine", |
|
"CYCLES", |
|
] |
|
|
|
subprocess.run(command, check=True) |
|
|
|
|
|
images = [] |
|
|
|
for i in range(nviews): |
|
fp = os.path.abspath(os.path.join(output_dir, f"{i:03d}_textured.png")) |
|
images.append(fp) |
|
|
|
return images |
|
|
|
|
|
def save_gif(image_paths, outfile): |
|
images = [Image.open(img) for img in image_paths] |
|
if len(images) > 1: |
|
background = Image.new("RGBA", images[0].size, (255, 255, 255)) |
|
images = [ |
|
Image.alpha_composite(background, png).convert("RGB") for png in images |
|
] |
|
images[0].save( |
|
outfile, save_all=True, append_images=images[1:], duration=100, loop=0 |
|
) |
|
|
|
|
|
def render_turntable(obj_path, output_dir, output_name="turntable"): |
|
""" |
|
Render a turntable gif of the mesh. Assumes that blender is installed and is in your path. |
|
obj_path : path to the obj file |
|
output_dir : directory to save the gif. Final image will be saved as `turntable.gif` |
|
""" |
|
image_paths = render_asset(obj_path, output_dir) |
|
gif_turntable_outfile = Path(output_dir) / f"{output_name}.gif" |
|
save_gif(image_paths, gif_turntable_outfile) |
|
return gif_turntable_outfile |
|
|
|
|
|
if __name__ == "__main__": |
|
parser = argparse.ArgumentParser() |
|
parser.add_argument("-i", "--input") |
|
parser.add_argument("-o", "--output_dir") |
|
args = parser.parse_args(sys.argv[1:]) |
|
render_turntable(args.input, args.output_dir) |
|
|
