Skip to content

Can you provide a pre-trained model for testing? #8

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

Open
anubisthetyrant opened this issue Oct 9, 2024 · 4 comments
Open

Can you provide a pre-trained model for testing? #8

anubisthetyrant opened this issue Oct 9, 2024 · 4 comments
Assignees

Comments

@anubisthetyrant
Copy link

Need a pre-trained model.

@vanossj
Copy link
Member

vanossj commented Oct 9, 2024

Thanks for your interest. The model was released on MHub.ai. I have updated the README

@vanossj vanossj closed this as completed Oct 9, 2024
@anubis-tyrant
Copy link

Any idea why it gets stuck in here?

  1. DicomImporter (0:00) 1/1
  2. NiftiConverter (0:03) 1/1
  3. NNUnetRunner [ ] 0/1 eta ~?
  4. BamfProcessorRunner
  5. DsegConverter
  6. DataOrganizer

Always gets stuck in the 3rd portion, and does not show any progress. Waited for 2 hours.
What is it doing?

@vanossj
Copy link
Member

vanossj commented Oct 16, 2024

Hmmm, I'm running into the same thing. @jithenece can you take a look?

Here's my code:
Downloaded an example CT scan from the TCGA-KIRC using idc-index

from idc_index import index
from pathlib import Path

Path("example_data").mkdir(exist_ok=True)

client = index.IDCClient()
client.download_from_selection(
    seriesInstanceUID=["1.3.6.1.4.1.14519.5.2.1.6450.4004.318185778053926832345567953536"],
    downloadDir="example_data",
)

Then ran the docker container with the following command

mkdir example_output
export in=$(pwd)/example_data
export out=$(pwd)/example_output

docker run --rm -t --gpus all -v $in:/app/data/input_data -v $out:/app/data/output_data mhubai/bamf_nnunet_ct_kidney

@vanossj
Copy link
Member

vanossj commented Oct 21, 2024

For the previous example, that would finish in about 10 minutes, but not produce output because of a different issue (referenced).

This example below runs successfully. If you are having issues, try running mhub with the --stop-on-error and --print flags and post them here.

Download example series

from idc_index import index
from pathlib import Path

Path("example_data").mkdir(exist_ok=True)

client = index.IDCClient()
client.download_from_selection(
    seriesInstanceUID=["1.3.6.1.4.1.14519.5.2.1.6450.4004.318185778053926832345567953536"],
    downloadDir="example_data",
)

run container

mkdir example_output
export in=$(pwd)/example_data
export out=$(pwd)/example_output

docker run --rm -t --gpus all -v $in:/app/data/input_data -v $out:/app/data/output_data mhubai/bamf_nnunet_ct_kidney

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

No branches or pull requests

3 participants