MedSegDiff error in sample

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!

Dec 21 '23 02:12 wzx0720

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.

Jun 12 '24 08:06 amitshakyaymsli

why is it like that in the code? doesn't seem to make sense to get just the 2 first elements.

Oct 15 '24 18:10 pedropesserl

why is it like that in the code? doesn't seem to make sense to get just the 2 first elements.

@pedropesserl I think you can try what amitshakyaymsli mentioned before：

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.

Oct 16 '24 02:10 wzx0720