Skip to content

Commit 5bb2dd9

Browse files
committed
add main method for testing
1 parent 4c41326 commit 5bb2dd9

File tree

2 files changed

+1
-2
lines changed

2 files changed

+1
-2
lines changed

Diff for: src/main/java/io/bioimage/modelrunner/onnx/OnnxInterface.java

+1-1
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ public static void main(String args[]) throws LoadModelException, RunModelExcept
9292
OnnxInterface oi = new OnnxInterface();
9393
oi.loadModel(folderName, source);
9494

95-
RandomAccessibleInterval<FloatType> img = ArrayImgs.floats(new long[] {1, 3, 20000, 20000});
95+
RandomAccessibleInterval<FloatType> img = ArrayImgs.floats(new long[] {1, 1, 256, 256});
9696
Tensor<FloatType> tt = Tensor.build("input0", "bcyx", img);
9797
List<Tensor<?>> inps = new ArrayList<Tensor<?>>();
9898
inps.add(tt);

Diff for: src/main/java/io/bioimage/modelrunner/onnx/tensor/TensorBuilder.java

-1
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,6 @@
3232

3333
import net.imglib2.Cursor;
3434
import net.imglib2.RandomAccessibleInterval;
35-
import net.imglib2.blocks.PrimitiveBlocks;
3635
import net.imglib2.img.Img;
3736
import net.imglib2.type.Type;
3837
import net.imglib2.type.numeric.integer.ByteType;

0 commit comments

Comments
 (0)