Minor error - seg_paths_train and seg_paths_eval not defined in else statements of main.py

mazurowski-lab / segmentation-guided-diffusion

[MICCAI 2024] Easy diffusion models (optionally with segmentation guidance) for medical images and beyond.

https://arxiv.org/abs/2402.05210

Other

110 stars 6 forks source link

Minor error - seg_paths_train and seg_paths_eval not defined in else statements of main.py #8

Closed lukalafaye closed 5 months ago

lukalafaye commented 5 months ago

You should move seg_paths_train = {} and seg_paths_eval = {} before the if/else statements in lines 93 and 103 of main.py. Otherwise the else statements will throw an undefined error for these dictionaries.

Thanks anyways for the project!

nickk124 commented 5 months ago

Thanks for the catch :) this has just been fixed in commit 404cebb69a51d714f152715958ce394e3e8f0258. Please let me know if you're still having problems!

Cheers