search

dyollb / segmantic

Semantic segmentation and image-to-image translation based on AI
https://dyollb.github.io/segmantic/
MIT License
1 stars 3 forks source link

BUG: remove image_dir/label_dir from config in cross-validate #60

Closed dyollb closed 1 year ago

dyollb commented 1 year ago

https://github.com/dyollb/segmantic/blob/318931c677204bb4b9f11ec4d48f6b15c4298078/src/segmantic/seg/monai_unet.py#L768

setting data["dataset"] = str(dataset_path) is not overriding/overwriting keys image_dir and labels_dir. The training then uses those instead of the k-fold dataset.

how to fix: 

data.pop("image_dir")
data.pop("labels_dir")
Konohana0608 commented 1 year ago

implemented in new pull request.