Adapt to OSS DGP v1.0 (#184)

docker/Dockerfile

@@ -128,7 +128,7 @@ RUN git clone https://github.com/TRI-ML/dgp.git
 ENV PYTHONPATH="/workspace/dgp:$PYTHONPATH"
 
 # Override DGP wandb with required version
-RUN pip install wandb==0.8.21 pyquaternion
+RUN pip install wandb==0.8.21 pyquaternion xarray diskcache tenacity pycocotools
 
 # Expose Port for jupyter (8888)
 EXPOSE 8888
@@ -141,4 +141,4 @@ WORKDIR /workspace/${PROJECT}
 # Copy project source last (to avoid cache busting)
 WORKDIR /workspace/${PROJECT}
 COPY . /workspace/${PROJECT}
-ENV PYTHONPATH="/workspace/${PROJECT}:$PYTHONPATH"
+ENV PYTHONPATH="/workspace/${PROJECT}:$PYTHONPATH"

packnet_sfm/datasets/dgp_dataset.py

@@ -6,7 +6,7 @@
 
 from dgp.datasets.synchronized_dataset import SynchronizedSceneDataset
 from dgp.utils.camera import Camera, generate_depth_map
-from dgp.utils.geometry import Pose
+from dgp.utils.pose import Pose
 
 from packnet_sfm.utils.misc import make_list
 from packnet_sfm.utils.types import is_tensor, is_numpy, is_list
@@ -202,7 +202,7 @@ def get_filename(self, sample_idx, datum_idx):
             Filename for the datum in that sample
         """
         scene_idx, sample_idx_in_scene, datum_indices = self.dataset.dataset_item_index[sample_idx]
-        scene_dir = self.dataset.get_scene_directory(scene_idx)
+        scene_dir = self.dataset.dataset_metadata.directory
         filename = self.dataset.get_datum(
             scene_idx, sample_idx_in_scene, datum_indices[datum_idx]).datum.image.filename
         return os.path.splitext(os.path.join(os.path.basename(scene_dir),