mGPT/render/blender/sampler.py

import numpy as np

def get_frameidx(*, mode, nframes, exact_frame, frames_to_keep):
    if mode == "sequence":
        frameidx = np.linspace(0, nframes - 1, frames_to_keep)
        frameidx = np.round(frameidx).astype(int)
        frameidx = list(frameidx)
    elif mode == "frame":
        index_frame = int(exact_frame*nframes)
        frameidx = [index_frame]
    elif mode == "video":
        frameidx = range(0, nframes)
    else:
        raise ValueError(f"Not support {mode} render mode")
    return frameidx