Skip to content

Fast inference for large files with several classes #2540#2545

Open
pooya-mohammadi wants to merge 2 commits intoMIC-DKFZ:masterfrom
pooya-mohammadi:master
Open

Fast inference for large files with several classes #2540#2545
pooya-mohammadi wants to merge 2 commits intoMIC-DKFZ:masterfrom
pooya-mohammadi:master

Conversation

@pooya-mohammadi
Copy link
Copy Markdown

When doing inference on large files with large number of classes the following code section blocks the code for up too 600 seconds:

            for c in range(data.shape[0]):
                reshaped_final[c] = resize_fn(data[c], new_shape, order, **kwargs)

in preprocessing.resampling.default_resampling.py and func: resample_data_or_seg.
If someone does not need the probabilities they can skip this section by only sending the segmentation array which is very fast.

Solving issue #2540

@pooya-mohammadi
Copy link
Copy Markdown
Author

Or the depicted code snippet can be done in a multiprocessing manner.

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