search

qubvel / segmentation_models.pytorch

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.
https://smp.readthedocs.io/
MIT License
9.14k stars 1.63k forks source link

ModuleNotFoundError: No module named 'segmentation_models_pytorch.fpn' #788

Closed alskdjfasdfsadf closed 10 months ago

alskdjfasdfsadf commented 1 year ago

Hello, Thanks so much for your work! There is a error:ModuleNotFoundError: No module named 'segmentation_models_pytorch.losses'. Would you help me to solve this problem? Thank you very much!

/bin/python3 /home/user/background_segmentation.py Traceback (most recent call last): File "/home/user/background_segmentation.py", line 11, in model = torch.load('/home/user/background_segmentation_model.pth', map_location=DEVICE) File "/home/user/.local/lib/python3.10/site-packages/torch/serialization.py", line 809, in load return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args) File "/home/user/.local/lib/python3.10/site-packages/torch/serialization.py", line 1172, in _load result = unpickler.load() File "/home/user/.local/lib/python3.10/site-packages/torch/serialization.py", line 1165, in find_class return super().find_class(mod_name, name) ModuleNotFoundError: No module named 'segmentation_models_pytorch.fpn'

github-actions[bot] commented 10 months ago

This issue is stale because it has been open 60 days with no activity. Remove stale label or comment or this will be closed in 7 days.

github-actions[bot] commented 10 months ago

This issue was closed because it has been stalled for 7 days with no activity.