Fix: use bilinear (for 4D) instead of linear

maudzung · Jul 23, 2020 · 1968a46 · 1968a46
1 parent 5b36aa9
commit 1968a46
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/data_process/kitti_dataset.py b/src/data_process/kitti_dataset.py
@@ -227,7 +227,7 @@ def collate_fn(self, batch):
         # Resize images to input shape
         imgs = torch.stack(imgs)
         if self.img_size != cnf.BEV_WIDTH:
-            imgs = F.interpolate(imgs, size=self.img_size, mode="linear", align_corners=True)
+            imgs = F.interpolate(imgs, size=self.img_size, mode="bilinear", align_corners=True)
         self.batch_count += 1
 
         return paths, imgs, targets