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

How to run inference #77

Open
wangrx33 opened this issue Jun 15, 2022 · 1 comment
Open

How to run inference #77

wangrx33 opened this issue Jun 15, 2022 · 1 comment

Comments

@wangrx33
Copy link

Hi there, I have a problem with the inference phase.
How to set the input_folder when I run the predict_simple.py?
Thanks

@282857341
Copy link
Owner

if ${predict}
then
cd /home/xychen/new_transformer/nnFormerFrame/DATASET/nnFormer_raw/nnFormer_raw_data/Task001_ACDC/
CUDA_VISIBLE_DEVICES=${cuda} nnFormer_predict -i imagesTs -o inferTs/${name} -m 3d_fullres -t ${task} -f 0 -chk model_best -tr nnFormerTrainerV2_${name}
python inference_acdc.py ${name}
fi

You can run the code above to run the inference. And the path of the input folder can be set as -i imagesTs

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

2 participants