add missing var to python code

segment-anything-models-java · Oct 28, 2024 · 3bab7b8 · 3bab7b8
1 parent 25201b8
commit 3bab7b8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/main/java/ai/nets/samj/models/PythonMethods.java b/src/main/java/ai/nets/samj/models/PythonMethods.java
@@ -130,6 +130,7 @@ public class PythonMethods {
 			+ "        mask = np.array(mask)" +  System.lineSeparator()
 			+ "    " +  System.lineSeparator()
 			+ "    # Flatten the mask in row-major order" +  System.lineSeparator()
+			+ "    ww = mask.shape[1]" +  System.lineSeparator()
 			+ "    binary = mask.flatten()" +  System.lineSeparator()
 			+ "    " +  System.lineSeparator()
 			+ "    # Find positions where values change" +  System.lineSeparator()