|
37 | 37 | import io.bioimage.modelrunner.system.PlatformDetection;
|
38 | 38 | import io.bioimage.modelrunner.tensor.Tensor;
|
39 | 39 | import io.bioimage.modelrunner.tensor.shm.SharedMemoryArray;
|
| 40 | +import io.bioimage.modelrunner.tensorflow.v2.api050.shm.ShmBuilder; |
40 | 41 | import io.bioimage.modelrunner.tensorflow.v2.api050.tensor.ImgLib2Builder;
|
41 | 42 | import io.bioimage.modelrunner.tensorflow.v2.api050.tensor.TensorBuilder;
|
42 |
| -import io.bioimage.modelrunner.tensorflow.v2.api050.tensor.mappedbuffer.ImgLib2ToMappedBuffer; |
43 |
| -import io.bioimage.modelrunner.tensorflow.v2.api050.tensor.mappedbuffer.MappedBufferToImgLib2; |
44 | 43 | import io.bioimage.modelrunner.utils.CommonUtils;
|
45 | 44 | import io.bioimage.modelrunner.utils.Constants;
|
46 | 45 | import io.bioimage.modelrunner.utils.ZipUtils;
|
|
50 | 49 | import net.imglib2.util.Cast;
|
51 | 50 | import net.imglib2.util.Util;
|
52 | 51 |
|
53 |
| -import java.io.BufferedReader; |
54 | 52 | import java.io.File;
|
55 | 53 | import java.io.IOException;
|
56 |
| -import java.io.InputStreamReader; |
57 |
| -import java.io.RandomAccessFile; |
58 | 54 | import java.io.UnsupportedEncodingException;
|
59 | 55 | import java.net.URISyntaxException;
|
60 | 56 | import java.net.URL;
|
61 | 57 | import java.net.URLDecoder;
|
62 |
| -import java.nio.ByteBuffer; |
63 |
| -import java.nio.MappedByteBuffer; |
64 |
| -import java.nio.channels.FileChannel; |
65 | 58 | import java.nio.charset.StandardCharsets;
|
66 |
| -import java.nio.file.Files; |
67 |
| -import java.nio.file.Paths; |
68 | 59 | import java.security.ProtectionDomain;
|
69 |
| -import java.time.LocalDateTime; |
70 |
| -import java.time.format.DateTimeFormatter; |
71 | 60 | import java.util.ArrayList;
|
72 | 61 | import java.util.HashMap;
|
73 | 62 | import java.util.LinkedHashMap;
|
@@ -312,7 +301,7 @@ protected void runFromShmas(List<String> inputs, List<String> outputs) throws IO
|
312 | 301 | for (String ee : inputs) {
|
313 | 302 | Map<String, Object> decoded = Types.decode(ee);
|
314 | 303 | SharedMemoryArray shma = SharedMemoryArray.read((String) decoded.get(MEM_NAME_KEY));
|
315 |
| - TType inT = io.bioimage.modelrunner.tensorflow.v2.api030.shm.TensorBuilder.build(shma); |
| 304 | + TType inT = io.bioimage.modelrunner.tensorflow.v2.api050.shm.TensorBuilder.build(shma); |
316 | 305 | if (PlatformDetection.isWindows()) shma.close();
|
317 | 306 | inTensors.add(inT);
|
318 | 307 | String inputName = getModelInputName((String) decoded.get(NAME_KEY), c ++);
|
|
0 commit comments