Skip to content

Commit 0f31cd6

Browse files
committed
fix jyhton script
1 parent bdcf7bb commit 0f31cd6

1 file changed

Lines changed: 12 additions & 6 deletions

File tree

.github/scripts/run_model.py

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -2,10 +2,11 @@
22
from ai.nets.samj.ij import SAMJ_Annotator
33

44
from java.util import ArrayList
5+
56
from time import time
67

78
from ij import IJ
8-
9+
from jarray import array
910
from net.imglib2.img.display.imagej import ImageJFunctions
1011

1112

@@ -15,20 +16,25 @@
1516
model = SAM2Tiny()
1617

1718
point_prompt = ArrayList()
18-
point_prompt.add((100, 100))
1919

20-
start_time = time()
2120

22-
mask = SAMJ_Annotator.samJReturnMask(model, wrapImg, point_prompt, None)
21+
my_seq = (100, 100)
22+
23+
arr1 = array(my_seq,'i')
2324

25+
26+
point_prompt.add(arr1)
27+
28+
start_time = time()
29+
mask = SAMJ_Annotator.samJReturnMask(model, wrapImg, point_prompt, None)
2430
end_time = time()
2531

2632
mask_sum = 0
2733
cursor = mask.cursor()
2834

2935
while cursor.hasNext():
30-
mask_sum += cursor.get()
31-
cursor.next()
36+
cursor.next()
37+
mask_sum += cursor.get().getRealDouble()
3238

3339
print("Total non-zero pixels: " + str(mask_sum))
3440
print("Total time: " + str(end_time - start_time))

0 commit comments

Comments
 (0)