Skip to content

Commit

Permalink
change inefficeint line of code
Browse files Browse the repository at this point in the history
  • Loading branch information
carlosuc3m committed Sep 2, 2023
1 parent 423f87f commit 9fbb8ce
Showing 1 changed file with 2 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,7 @@

import net.imglib2.img.Img;
import net.imglib2.img.ImgFactory;
import net.imglib2.img.array.ArrayImgFactory;
import net.imglib2.img.cell.CellImgFactory;
import net.imglib2.type.NativeType;
import net.imglib2.type.numeric.RealType;
Expand Down Expand Up @@ -84,7 +85,7 @@ public static void loadAndRunTf2() throws LoadEngineException, Exception {
// Load the corresponding model
Model model = loadModel(modelFolder, modelSource, engineInfo);
// Create an image that will be the backend of the Input Tensor
final ImgFactory< FloatType > imgFactory = new CellImgFactory<>( new FloatType(), 5 );
final ImgFactory< FloatType > imgFactory = new ArrayImgFactory<>( new FloatType() );
final Img< FloatType > img1 = imgFactory.create( 1, 512, 512, 1 );
// Create the input tensor with the nameand axes given by the rdf.yaml file
// and add it to the list of input tensors
Expand Down

0 comments on commit 9fbb8ce

Please sign in to comment.