blender_render.py

# Run as: blender -b <filename> -P <this_script> -- <image_path>
import bpy, sys, os
import pdb 

#pdb.set_trace()
py_args = sys.argv[sys.argv.index("--") + 1:]
group_index = py_args.index('group')
if group_index > -1:
    group = py_args[group_index+1]
else:
    group = 15 #args.group

output_index = py_args.index('output')
if output_index > -1:
    output_path = py_args[output_index+1]
else:
    output_path = 'output'

# THE DATASET ROOT PATH
# uni pc
# base_path = '/media/lucap/big_data/datasets/repair'
# home pc 
base_path = '/home/palma/Unive/RePAIR/Datasets/RePAIR_dataset'

fragments_folder = os.path.join(base_path, f'group_{group}/facing_up')
files = os.listdir(fragments_folder)
objs = [file for file in files if file[-4:] == '.obj']
render_group_folder = os.path.join(os.getcwd(), output_path, f'group_{group}')
os.makedirs(render_group_folder, exist_ok=True)

for frag in objs:
    bpy.context.scene.render.filepath = os.path.join(render_group_folder, frag[:-4]+".png")
    f = os.path.join(fragments_folder, frag)
    # print(f)    
    if frag[-4:] == '.obj':
        bpy.ops.wm.obj_import(filepath=f)
    else:
        bpy.ops.import_mesh.ply(filepath=f)
    bpy.ops.render.render(write_still=True)
    #pdb.set_trace()
    
    bpy.data.objects[frag[:-4]].select_set(True)
    bpy.ops.object.delete()