search

Milan-BUAA / TSAN-brain-age-estimation

TSAN: Two-Stage-Age-Net, for brain age estimation from T1-weighted MRI data.
MIT License
59 stars 6 forks source link

Segmentation Fault Using Pretrained Model #12

Open JBarsotti opened 4 months ago

JBarsotti commented 4 months ago

Dear Dr. Yang,

Thank you for sharing the brain age TSCAN model you created on GitHub! It seems extremely interesting. I have been trying to use it on some brain scans in my possession, but I am getting a segmentation fault when calling the following code using the pretrained model:

output = model(input,male)

This is in the prediction.py file. I am using the pretrained model with the name TSAN_pretrain_model_20240402/ScaleDense_best_model.pth.tar.

liuziyang1106 commented 4 months ago

I'm happy to help you. Can you give me more error information？

JBarsotti commented 4 months ago

Wow! Thank you for such a prompt reply! :)

I'm not sure how much more information I can give about the error. I will attach a screenshot of the error message. I hope it helps some but I'm very bad at segmentation faults.

image

JBarsotti commented 4 months ago

Hi Dr. Yang,

I hope you are well!

I just wanted to check on the status of this issue. I know it hasn't been too long but was wondering whether you've had a chance to glance at it.

JBarsotti commented 3 months ago

I might have been able to figure this one out on my own!