from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

from .sample.car_pose import CarPoseDataset
from .dataset.kittihp import KITTIHP
from .dataset.nusceneshp import NUSCENESHP
from .dataset.lyfthp import LYFTHP

dataset_factory = {
  'kitti': KITTIHP,
  'nuscenes': NUSCENESHP,
  'lyft': LYFTHP
}

def get_dataset(dataset):
  class Dataset(dataset_factory[dataset], CarPoseDataset):
    pass
  return Dataset
  

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

from .sample.car_pose import CarPoseDataset
from .dataset.kittihp import KITTIHP
from .dataset.nusceneshp import NUSCENESHP
from .dataset.lyfthp import LYFTHP

dataset_factory = {
  'kitti': KITTIHP,
  'nuscenes': NUSCENESHP,
  'lyft': LYFTHP
}

def get_dataset(dataset):
  class Dataset(dataset_factory[dataset], CarPoseDataset):
    pass
  return Dataset
  
