Skip to content

Commit 69a2cb3

Browse files
committed
change os.chgdir by syspath append
1 parent bdac3a0 commit 69a2cb3

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

src/main/java/ai/nets/samj/models/EfficientViTSamJ.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,7 @@ public class EfficientViTSamJ extends AbstractSamJ {
8383
+ "from scipy.ndimage import binary_fill_holes" + System.lineSeparator()
8484
+ "import sys" + System.lineSeparator()
8585
+ "import os" + System.lineSeparator()
86-
+ "os.chdir(r'%s')" + System.lineSeparator()
86+
+ "sys.path.append(r'%s')" + System.lineSeparator()
8787
+ "from multiprocessing import shared_memory" + System.lineSeparator()
8888
+ "task.update('import sam')" + System.lineSeparator()
8989
+ "from efficientvit.models.efficientvit import EfficientViTSam, %s" + System.lineSeparator()

src/main/java/ai/nets/samj/models/Sam2.java

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -85,6 +85,9 @@ public class Sam2 extends AbstractSamJ {
8585
+ "from scipy.ndimage import label" + System.lineSeparator()
8686
+ "import sys" + System.lineSeparator()
8787
+ "import os" + System.lineSeparator()
88+
+ "import platform" + System.lineSeparator()
89+
+ "print(platform.platform())" + System.lineSeparator()
90+
+ "print(platform.machine())" + System.lineSeparator()
8891
+ "from multiprocessing import shared_memory" + System.lineSeparator()
8992
+ "from sam2.build_sam import build_sam2" + System.lineSeparator()
9093
+ "from sam2.sam2_image_predictor import SAM2ImagePredictor" + System.lineSeparator()

0 commit comments

Comments
 (0)