File tree Expand file tree Collapse file tree 3 files changed +4
-5
lines changed
src/main/java/io/bioimage/modelrunner/tensorflow/v2/api050 Expand file tree Collapse file tree 3 files changed +4
-5
lines changed Original file line number Diff line number Diff line change 125
125
<!-- NB: Deploy releases to the SciJava Maven repository. -->
126
126
<releaseProfiles >sign,deploy-to-scijava</releaseProfiles >
127
127
128
- <dl-modelrunner .version>0.4.1 </dl-modelrunner .version>
128
+ <dl-modelrunner .version>0.5.0 </dl-modelrunner .version>
129
129
<tensorflow-core .version>0.5.0</tensorflow-core .version>
130
130
<scijava .jvm.build.version>[11,)</scijava .jvm.build.version>
131
131
<scijava .jvm.version>11</scijava .jvm.version>
Original file line number Diff line number Diff line change @@ -274,9 +274,9 @@ private void unzipTfWeights(ModelDescriptor descriptor) throws LoadModelExceptio
274
274
if (new File (modelFolder , "tf_weights.zip" ).isFile ()) {
275
275
System .out .println ("Unzipping model..." );
276
276
ZipUtils .unzipFolder (modelFolder + File .separator + "tf_weights.zip" , modelFolder );
277
- } else if ( descriptor .getWeights ().getSupportedDLFrameworks ()
277
+ } else if ( descriptor .getWeights ().getAllSuportedWeightNames ()
278
278
.contains (EngineInfo .getBioimageioTfKey ()) ) {
279
- String source = descriptor .getWeights ().getSupportedWeights ().stream ()
279
+ String source = descriptor .getWeights ().gettAllSupportedWeightObjects ().stream ()
280
280
.filter (ww -> ww .getFramework ().equals (EngineInfo .getBioimageioTfKey ()))
281
281
.findFirst ().get ().getSource ();
282
282
source = DownloadModel .getFileNameFromURLString (source );
Original file line number Diff line number Diff line change 22
22
package io .bioimage .modelrunner .tensorflow .v2 .api050 .tensor ;
23
23
24
24
import io .bioimage .modelrunner .tensor .Utils ;
25
+ import io .bioimage .modelrunner .utils .CommonUtils ;
25
26
import net .imglib2 .Cursor ;
26
27
import net .imglib2 .RandomAccessibleInterval ;
27
- import net .imglib2 .blocks .PrimitiveBlocks ;
28
28
import net .imglib2 .img .Img ;
29
29
import net .imglib2 .type .Type ;
30
30
import net .imglib2 .type .numeric .integer .IntType ;
35
35
import net .imglib2 .util .Util ;
36
36
import net .imglib2 .view .Views ;
37
37
38
- import java .nio .ByteBuffer ;
39
38
import java .util .Arrays ;
40
39
41
40
import org .tensorflow .Tensor ;
You can’t perform that action at this time.
0 commit comments