Skip to content

Resolve "segmentation-models-pytorch" dependency for iclearn

When using the Torch model from iclearn on a different project there was an issue with the segmentation-models-pytorch library not being installed. Adding segmentation-models-pytorch to the dependencies in pyproject.toml might be the next course of action, or we could remove its inclusion entirely from model_collection.py.

I might be executing the importing incorrectly with:

from iclearn.models.torch.model import TorchModel

but even then it would probably be better to avoid such problems in the future as much as possible.

Edited by Diarmuid Coleman