V3D / sgm /data /joint3d.py
heheyas
init
cfb7702
raw
history blame
235 Bytes
import torch
from torch.utils.data import Dataset
default_sub_data_config = {}
class Joint3D(Dataset):
def __init__(self, sub_data_config: dict) -> None:
super().__init__()
self.sub_data_config = sub_data_config