Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Speed up feature extraction #40

Open
cooperlab opened this issue Oct 10, 2019 · 4 comments
Open

Speed up feature extraction #40

cooperlab opened this issue Oct 10, 2019 · 4 comments
Labels
enhancement New feature or request

Comments

@cooperlab
Copy link
Collaborator

Investigate how to speedup feature extraction pipelines. Examine GPU starvation and optimizing the reading and pre-processing of patches from WSI files.

@cooperlab cooperlab added the enhancement New feature or request label Oct 10, 2019
@Acmenwangtuo
Copy link

When I do this process, i use the gpu,why it is still very slow and only one cpu core run?

@Reasat
Copy link

Reasat commented Mar 16, 2021

In the feature generation step
docker run --runtime=nvidia -it --rm --name extractfeatures -v "$PWD":/"${PWD##*/}" cancerdatascience/hml_dataset_gpu:1.0 python scripts/FeatureExtraction.py --projectName "${PWD##*/}"

the FeatureExtraction.py script extracts features from one patch at a time. Significant speedup can be attained by loading the images in batches. I have patched up the file so that it loads batches of images.

https://gist.github.com/Reasat/76b53d6be24bceff4525e7ab92ca9ffd

Corresponding XML file
https://gist.github.com/Reasat/f674ca6616edfc94039bc8c449b14844
@cooperlab

@slee172
Copy link
Collaborator

slee172 commented Mar 16, 2021

@Reasat Thank you for the updates. We are working on a new version of the pipeline, so I'll see it again after the updates.

@cooperlab
Copy link
Collaborator Author

@Reasat thanks for looking at this. We acknowledge that the pipeline needs to be improved. We are working to introduce improvements that will speedup the 3 intensive processes - superpixel segmentation, boundary tracing, and feature extraction. We are done with the segmentation and tracing and should be done with the extraction portion soon. These improvements will allow feature extraction on multi-GPU systems.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

4 participants