search

voxelmorph / voxelmorph

Unsupervised Learning for Image Registration
Apache License 2.0
2.29k stars 581 forks source link

Training VoxelMroph with my own dataset #388

Open Neonsign247 opened 2 years ago

Neonsign247 commented 2 years ago

Thank you for your nice work and code. I'm having trouble utilizing your code to my own dataset.

Task (what are you trying to do/register?)

I'm trying to train new voxelmroph model that learns how to register from my own moving(following) 3D CT image to paired fixed(baseline) 3D CT image.

What have you tried

I read readme and some issues about 'list.txt' but I don't understand how to feed my paired moving and fixed data!

I tried list.txt like following, but the sequence of images from generator gives me random pair of images.

../data_ct/s001/baseline/nifti/ct.nii.gz ../data_ct/s001/followup/nifti/ct.nii.gz ../data_ct/s002/baseline/nifti/ct.nii.gz ../data_ct/s002/followup/nifti/ct.nii.gz ../data_ct/s003/baseline/nifti/ct.nii.gz ../data_ct/s003/followup/nifti/ct.nii.gz

I modified ./scripts/tf/train.py file to my training.

Please advise on how to feed paired moving/fixed image for training.

Thank you for reading.

solarflarefx commented 2 years ago

@Neonsign247 Have you solved this issue?