diff --git a/apps/render_data.py b/apps/render_data.py index b9bbb763..563c03fb 100755 --- a/apps/render_data.py +++ b/apps/render_data.py @@ -144,9 +144,9 @@ def rotateBand2(x, R): return dst -def render_prt_ortho(out_path, folder_name, subject_name, shs, rndr, rndr_uv, angl_step=4, n_light=1, pitch=[0]): - cam = Camera(width=512, height=512) - cam.ortho_ratio = 0.4 +def render_prt_ortho(out_path, folder_name, subject_name, shs, rndr, rndr_uv, im_size, angl_step=4, n_light=1, pitch=[0]): + cam = Camera(width=im_size, height=im_size) + cam.ortho_ratio = 0.4 * (512 / im_size) cam.near = -100 cam.far = 100 cam.sanity_check() @@ -287,4 +287,4 @@ def render_prt_ortho(out_path, folder_name, subject_name, shs, rndr, rndr_uv, an if args.input[-1] == '/': args.input = args.input[:-1] subject_name = args.input.split('/')[-1][:-4] - render_prt_ortho(args.out_dir, args.input, subject_name, shs, rndr, rndr_uv, 1, 1, pitch=[0]) \ No newline at end of file + render_prt_ortho(args.out_dir, args.input, subject_name, shs, rndr, rndr_uv, args.size, 1, 1, pitch=[0]) \ No newline at end of file