Function render_batch

annotation_model/persistent_batch_render_auto.py:56–94

Runs Blender to render a batch of objects.

(object_files, target_path, output_dir, core_id)

    subprocess.run(command)

def render_batch(object_files, target_path, output_dir, core_id):
    """Runs Blender to render a batch of objects."""
    #core_id = get_core_id()
    #set_affinity(core_id)
    command = [
        "taskset",
        "-c",
        str(core_id),
        "/mnt/2077AI/3d_model_labeller/Multiview_3D_model_classifier/labeller_model/blender-4.2.1-linux-x64/blender",
        "-b",
        "-P",
        "persistent_blender_script.py",
        "--",
        "--object_path",
        target_path,
        "--output_dir",
        output_dir,
        "--engine",
        "CYCLES",
        "--scale",
        "0.8",
        "--num_images",
        "40",
        "--camera_dist",
        "1.2",
    ]

    env = os.environ.copy()
    env['OBJECT_BATCH'] = ','.join(object_files)

    log_dir = os.getcwd() + "/correct_logs/" + object_files[0][:-4]
    log_file_path = log_dir + "_log.log"
    error_file_path = log_dir + "_error.log"
    
    with open(log_file_path, 'a') as log_file, open(error_file_path, 'a') as error_file:
        # Run the command and suppress the output (stdout and stderr) to the log file
        subprocess.run(command, env=env, stdout=log_file, stderr=error_file)

    #subprocess.run(command, env=env)

def split_batches(object_files, num_batches):
    """Split the list of object files into smaller batches."""