imagej / imagej/imagej-modelzoo

Modelzoo exporter command

Open
#1 8 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Dominant language
Java
Stars
6
Forks
4
PR merge metrics
No merged PRs in 30d

Description

We need a command & API to export trained models into modelzoo format.

## Use cases
### Export SavedModelBundle method
- input: `SavedModelBundle` directory (including `variables`, `saved_model.pb`)
- input: additional needed parameters (inputs, outputs, data type etc.)
### Add missing information to existing YAML specification
- input: A ZIP file with a `SavedModelBundle` and a `model.yaml` file

## Goal

The exporter should be able to write the YAML specification and ask the user for missing parameters (like author names, a doi to the training data etc.).

## Existing code

Writing the model is already implemented in Noise2Void in Fiji and should move from there to this repository, then N2V and others can use `imagej-modelzoo` as a library: https://github.com/juglab/N2V_fiji/blob/master/src/main/java/de/csbdresden/n2v/train/ModelSpecification.java

Tagging @esgomezm @ctr26 in case it might be possible to reuse code from DeepImageJ for the UI part (discussion started in [this issue](https://github.com/deepimagej/deepimagej-plugin/issues/15)). @tibuch

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

Start by reviewing the existing ModelSpecification.java implementation in N2V_fiji and the linked DeepImageJ UI discussion. Determine the API and command boundaries for SavedModelBundle and ZIP inputs, including missing parameters and YAML output. Done means a decided, reusable exporter design that supports the listed use cases.

Written by the indexing model from the issue text.

Assessment

Tech stack
java
Domain
backend-api-design, machine-learning
Issue type
Feature
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.