41 lines
810 B
Python
41 lines
810 B
Python
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
|