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