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

Define global ckptdir in checkpoint.py and import from there in run_p… #214

Merged
merged 1 commit into from
Dec 8, 2022

Conversation

LeHenschel
Copy link
Member

…rediction.py and download_checkpoints.py to avoid mismatch and double download. Also fixed Dockerfile for FastSurferCNN only.

Previously, download_checkpoints.py and run_prediction.py would download the checkpoints into different directories! By default, the FastSurferCNN subdirectory would be the parent dir for download_checkpoints.py while the current working dir would be the parent dir for run_prediction.py.

To avoid this confusion:

  • checkpoints are now always downloaded into FastSurferCNN/checkpoints by default
  • path is completely defined via global variables AXI, SAG, COR in checkpoints.py
  • Issues with duplicate download when running run_prediction.py with the FastSurferCNN seg-only docker are avoided this way

…rediction.py and download_checkpoints.py to avoid mismatch and double download
@m-reuter m-reuter merged commit 3cc9bf3 into dev Dec 8, 2022
@m-reuter m-reuter deleted the fix/checkpoint_download branch December 8, 2022 09:18
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