Function load_tensor

tests/visualize_cpp_masks.py:17–23

(path)

os.makedirs(out_dir, exist_ok=True)

def load_tensor(path):
    data = np.fromfile(path + ".bin", dtype=np.float32)
    if os.path.exists(path + ".shape"):
        with open(path + ".shape") as f:
            shape = [int(x) for x in f.read().strip().split(",") if x]
        return data.reshape(shape) if shape else data
    return data

for fi in [1, 2, 5, 10, 20, 30, 40, 50]:
    frame_path = os.path.join(video_dir, f"{fi:05d}.jpg")