imagej / imagej/imagej-tensorflow

Upgrade to TensorFlow 2

Open
#29 0 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Dominant language
Java
Stars
26
Forks
15
PR merge metrics
No merged PRs in 30d

Description

I'll just use this issue to save some notes regarding the upgrade:

The new maven dependency for all operating systems (`tensorflow-core-platform-mkl-gpu` for GPU support though that didn't immediately work for me):
```

org.tensorflow
tensorflow-core-platform-mkl
0.2.0

```

Example of how to create Tensors from arrays with the new API:
```
float[] imgArray = new float[height * width* channel];
// .. assign img values to array
Shape shape = Shape.of(1, height, width, channel);
Tensor tensor = TFloat32.tensorOf(NdArrays.wrap(shape, DataBuffers.of(imgArray)));
```

Ping @tomburke-rse

Contributor guide

No contributing guide indexed for this repository

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

No source files or tests are named. Start by reviewing the Maven dependency and the Tensor creation example in the issue, then inspect the project for existing TensorFlow integration points. Done means the integration is upgraded to TensorFlow 2, including the dependency and affected tensor-creation usage.

Written by the indexing model from the issue text.

Assessment

Tech stack
java, tensorflow
Domain
machine-learning
Issue type
Refactor
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.