search

Doodleverse / segmentation_zoo

A collection of geoscientific image segmentation models
MIT License
8 stars 3 forks source link

CustomSatUnet Model is missing from doodleverse_utils #21

Closed 2320sharon closed 1 year ago

2320sharon commented 1 year ago

In scripts/select_model_and_batch_process_folder.py if the MODEL variable read from the config file is 'satunet' the code attempts to create a custom_satunet model from doodleverseutils doodleverse_utils/model_imports.py, but this model doesn't exist.

After digging through doodleverse_utils/model_imports.py I found a model defined as simple_satunet so I'm going to replace the call for custom_satunet with simple_satunet for now.

dbuscombe-usgs commented 1 year ago

Thanks for identifying this. I think custom_satunet does not exist. It is only simple_satunet that exists. So, I believe your fix can be made permanent