Feat/cityscapes tutorial (#580)

* Add Citscapes dataset adapter

* Add Cityscapes dataset adapter and cleanup

* Update datasets/__init__.py

* Correction in build_dataset arg

* Add tutorial for cityscapes semanticsegmentation

* Restore cityscacpescripts dependency

Author: tejasstanley

examples/tutorial_cityscapes_image_segmentation.ipynb

@@ -43,6 +43,7 @@
     "cityscapes_image_segmentation": CityscapesImageSegmentationDataset,
     "nuimages_image_segmentation": NuImagesSegmentationDataset,
     "nuimages_detection": NuImagesDetectionDataset,
+    "cityscapes_image_segmentation": CityscapesImageSegmentationDataset,
 }
 
 if CocoDataset is not None:

perceptionmetrics/datasets/__init__.py

@@ -349,7 +349,20 @@ def inference(self, tensor_in: torch.Tensor) -> torch.Tensor:
                 tensor_out = self.model(tensor_in.to(self.device))
 
             if isinstance(tensor_out, dict):
-                tensor_out = tensor_out["out"]
+                if "out" in tensor_out:
+                    tensor_out = tensor_out["out"]
+                elif "logits" in tensor_out:
+                    tensor_out = tensor_out["logits"]
+            elif hasattr(tensor_out, "logits"):
+                tensor_out = tensor_out.logits
+
+            if tensor_out.shape[-2:] != tensor_in.shape[-2:]:
+                tensor_out = torch.nn.functional.interpolate(
+                    tensor_out,
+                    size=tensor_in.shape[-2:],
+                    mode="bilinear",
+                    align_corners=False,
+                )
 
         return tensor_out