def __init__(self,
config,
input_transform=None,
target_transform=None,
augment_data=True,
elastic_distortion=False,
cache=False,
phase=DatasetPhase.Train):
if isinstance(phase, str):
phase = str2datasetphase_type(phase)
# Use cropped rooms for train/val
data_root = config.scannet_path
if phase not in [DatasetPhase.Train, DatasetPhase.TrainVal]:
self.CLIP_BOUND = self.TEST_CLIP_BOUND
data_paths = read_txt(
os.path.join(data_root, self.DATA_PATH_FILE[phase]))
logging.info('Loading {}: {}'.format(self.__class__.__name__,
self.DATA_PATH_FILE[phase]))
super().__init__(data_paths,
data_root=data_root,
input_transform=input_transform,
target_transform=target_transform,
ignore_label=config.ignore_label,
return_transformation=config.return_transformation,
augment_data=augment_data,
elastic_distortion=elastic_distortion,
config=config)
def __init__(self,
config,
prevoxel_transform=None,
input_transform=None,
target_transform=None,
augment_data=True,
elastic_distortion=False,
cache=False,
phase=DatasetPhase.Train):
if isinstance(phase, str):
phase = str2datasetphase_type(phase)
if phase not in [DatasetPhase.Train, DatasetPhase.TrainVal]:
self.CLIP_BOUND = self.TEST_CLIP_BOUND
data_root = config.synthia_path
data_paths = read_txt(
osp.join('./splits/synthia4d', self.DATA_PATH_FILE[phase]))
data_paths = [d.split()[0] for d in data_paths]
logging.info('Loading {}: {}'.format(self.__class__.__name__,
self.DATA_PATH_FILE[phase]))
super().__init__(data_paths,
data_root=data_root,
input_transform=input_transform,
target_transform=target_transform,
ignore_label=config.ignore_label,
return_transformation=config.return_transformation,
augment_data=augment_data,
elastic_distortion=elastic_distortion,
config=config)
def __init__(self,
config,
prevoxel_transform=None,
input_transform=None,
target_transform=None,
augment_data=True,
elastic_distortion=False,
cache=False,
phase=DatasetPhase.Train):
if isinstance(phase, str):
phase = str2datasetphase_type(phase)
if phase not in [DatasetPhase.Train, DatasetPhase.TrainVal]:
self.CLIP_BOUND = self.TEST_CLIP_BOUND
data_root = config.synthia_path
data_paths = read_txt(osp.join(data_root, self.DATA_PATH_FILE[phase]))
data_paths = sorted([d.split()[0] for d in data_paths])
seq2files = defaultdict(list)
for f in data_paths:
seq_name = f.split(os.sep)[0]
seq2files[seq_name].append(f)
self.camera_path = config.synthia_camera_path
self.camera_intrinsic_file = config.synthia_camera_intrinsic_file
self.camera_extrinsics_file = config.synthia_camera_extrinsics_file
# Force sort file sequence for easier debugging.
file_seq_list = []
for key in sorted(seq2files.keys()):
file_seq_list.append(sorted(seq2files[key]))
logging.info('Loading {}: {}'.format(self.__class__.__name__,
self.DATA_PATH_FILE[phase]))
TemporalVoxelizationDataset.__init__(
self,
file_seq_list,
data_root=data_root,
input_transform=input_transform,
target_transform=target_transform,
ignore_label=config.ignore_label,
temporal_dilation=config.temporal_dilation,
temporal_numseq=config.temporal_numseq,
return_transformation=config.return_transformation,
augment_data=augment_data,
elastic_distortion=elastic_distortion,
config=config)
def __init__(self,
config,
prevoxel_transform=None,
input_transform=None,
target_transform=None,
cache=False,
augment_data=True,
elastic_distortion=False,
phase=DatasetPhase.Train):
if isinstance(phase, str):
phase = str2datasetphase_type(phase)
if phase not in [DatasetPhase.Train, DatasetPhase.TrainVal]:
self.CLIP_BOUND = self.TEST_CLIP_BOUND
data_root = config.data.stanford3d_path
if isinstance(self.DATA_PATH_FILE[phase], (list, tuple)):
data_paths = []
for split in self.DATA_PATH_FILE[phase]:
data_paths += read_txt(os.path.join(data_root, 'splits',
split))
else:
data_paths = read_txt(
os.path.join(data_root, 'splits', self.DATA_PATH_FILE[phase]))
if config.data.voxel_size:
self.VOXEL_SIZE = config.data.voxel_size
logging.info('voxel size: {}'.format(self.VOXEL_SIZE))
logging.info('Loading {} {}: {}'.format(self.__class__.__name__, phase,
self.DATA_PATH_FILE[phase]))
VoxelizationDataset.__init__(
self,
data_paths,
data_root=data_root,
prevoxel_transform=prevoxel_transform,
input_transform=input_transform,
target_transform=target_transform,
ignore_label=config.data.ignore_label,
return_transformation=config.data.return_transformation,
augment_data=augment_data,
elastic_distortion=elastic_distortion,
config=config)
def __init__(self,
config,
input_transform=None,
target_transform=None,
augment_data=True,
cache=False,
phase=DatasetPhase.Train):
if isinstance(phase, str):
phase = str2datasetphase_type(phase)
data_root = config.sunrgbd_path
data_paths = read_txt(os.path.join(data_root, self.DATA_PATH_FILE[phase]))
logging.info('Loading {}: {}'.format(self.__class__.__name__, self.DATA_PATH_FILE[phase]))
super().__init__(
data_paths,
data_root=data_root,
input_transform=input_transform,
target_transform=target_transform,
ignore_label=config.ignore_label,
return_transformation=config.return_transformation,
augment_data=augment_data,
config=config)
