raise RuntimeError
if not self._shapes:
if self.__first_image is None:
_, self.__first_image, _ = self(idx=0)
self._shapes = {field: self.__first_image[field].shape
for field in self.output_fields}
return self._shapes
After Change
// 1) only read from the first image in list
// 2) not considering effects of random augmentation layers
// but time and modality dimensions should be correct
first_image = self.output_list[0]
self._shapes = {field: first_image[field].shape
for field in self.output_fields}
return self._shapes
