remove device from segmentation tools

mixaal · Sep 12, 2024 · 4d3df0a · 4d3df0a
1 parent a0fbdc1
commit 4d3df0a
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/utils.rs b/src/utils.rs
@@ -2,9 +2,7 @@ use tch::{IValue, Tensor};
 
 use crate::{image::ImageCHW, BBox, SegBBox, SegmentationResult};
 
-pub struct SegmentationTools {
-    device: tch::Device,
-}
+pub struct SegmentationTools {}
 pub struct DetectionTools {}
 
 impl SegmentationTools {