import os
import sys

import argparse
parser = argparse.ArgumentParser(description='')
parser.add_argument('--debug', action='store_true')
args = vars(parser.parse_args())

if args['debug']:
    os.environ["CUDA_VISIBLE_DEVICES"] = '0'

import math
import time
import torch
import random
import pytorch3d
import torchvision as tv
import numpy as np
import scipy.io as sio
import plotly.express as ex
from torch.utils.data import Dataset

sys.path.append(os.path.dirname(os.path.abspath(__file__)) + '/..')

from data.utils import Frame

class Dataset(Dataset):
    def __init__(self, folder):
        pass

    def __len__(self):
        return len(self.frames)


    def __getitem__(self, idx):
        frame : Frame = self.frames[idx]


if __name__ == '__main__':
    from torch.utils.data import DataLoader