Custom process chunk, new lsd scripts, new command line options #5
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
This pull requests adds the capability of defining a custom
process_chunkfunction 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 (seetensorflow_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.pytakes 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