Skip to content

Conversation

@davidackerman
Copy link
Collaborator

@davidackerman davidackerman commented Jan 24, 2025

This pull requests adds the capability of defining a custom process_chunk function in the model config scripts that will be used to do custom chunk processing. This is useful if wanting to run multiple models on a single chunk (see tensorflow_acrlsd.py) or for postprocessing, or anywhere custom handling of chunk is desired.

Also added in this PR are two example tensorflow model scripts for predicting local shape descriptors (lsds) and affinities based on https://github.com/funkelab/lsd. tensorflow_lsd.py takes in the raw data as input, rescales it and predicts 10 lsds as output. The second script, tensorflow_acrlsd.py, runs the aforementioned model first, and uses its output plus the rescaled raw as input to another model to produce 3 affinities as output.

Instructions have also been added to the readme for tensorflow installation

Command line options have also been added to define the gpu queue and chargeback group.

@stuarteberg

@davidackerman davidackerman merged commit f6ca6c1 into main Jan 24, 2025
@davidackerman davidackerman deleted the custom_process_chunk branch January 30, 2025 16:03
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants