error in sample - Githubissues

KidsWithTokens / MedSegDiff

Medical Image Segmentation with Diffusion Model

MIT License

976 stars 145 forks source link

error in sample #147

Open wzx0720 opened 6 months ago

wzx0720 commented 6 months ago

First of all, thanks for your excellent work! I met a problem that when I run segmentation_sample.py, it only sample and infer 2 images. I wonder how can I solve this problem? Thanks a lot!

amitshakyaymsli commented 4 weeks ago

if you used the custom_dataset_loader.py file. You should modify the "self.name_list = image[:2]" to "self.name_list = image" similar with the labels.