fixed transformation bug

Changed the order of interpolation to nearest-neighbour, fixing a bug that shifted the pixel values at the boundaries of masks
lowe-lab-ucl · Apr 8, 2022 · a5e3b12 · a5e3b12
1 parent 9cae310
commit a5e3b12
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/octopuslite/transform.py b/octopuslite/transform.py
@@ -14,7 +14,7 @@ def __call__(self, x, idx):
             return x
 
         tform = tf.AffineTransform(translation=self.transforms[idx, :2, 2])
-        return tf.warp(x, tform, preserve_range=True)
+        return tf.warp(x, tform, order = 0, preserve_range=True)
 
 
 def parse_transforms(path: os.PathLike) -> StackTransformer: