Merge pull request #1039 from adamjstewart/deps/pretrainedmodels

Vendor pretrainedmodels

Author: adamjstewart

docs/conf.py

@@ -100,7 +100,6 @@ def get_version():
     "timm",
     "cv2",
     "PIL",
-    "pretrainedmodels",
     "torchvision",
     "segmentation_models_pytorch.encoders",
     "segmentation_models_pytorch.utils",

licenses/LICENSES.md

@@ -13,14 +13,18 @@ The majority of the code is licensed under the [MIT License](LICENSE). However,
   * [segmentation_models_pytorch/encoders/mix_transformer.py](https://github.com/qubvel/segmentation_models.pytorch/blob/main/segmentation_models_pytorch/encoders/mix_transformer.py)
   * [LICENSE_nvidia](LICENSE_nvidia.md)
 
-
 - Apple License
   * Applies to the MobileOne encoder
   * [segmentation_models_pytorch/encoders/mobileone.py](https://github.com/qubvel/segmentation_models.pytorch/blob/main/segmentation_models_pytorch/encoders/mobileone.py)
   * [LICENSE_apple](LICENSE_apple.md)
 
 - BSD 3-Clause License
-  * Applies to the DeepLabV3 decoder
+  * Applies to several encoders and the DeepLabV3 decoder
+  * [segmentation_models_pytorch/encoders/_dpn.py](https://github.com/qubvel/segmentation_models.pytorch/blob/main/segmentation_models_pytorch/encoders/_dpn.py)
+  * [segmentation_models_pytorch/encoders/_inceptionresnetv2.py](https://github.com/qubvel/segmentation_models.pytorch/blob/main/segmentation_models_pytorch/encoders/_inceptionresnetv2.py)
+  * [segmentation_models_pytorch/encoders/_inceptionv4.py](https://github.com/qubvel/segmentation_models.pytorch/blob/main/segmentation_models_pytorch/encoders/_inceptionv4.py)
+  * [segmentation_models_pytorch/encoders/_senet.py](https://github.com/qubvel/segmentation_models.pytorch/blob/main/segmentation_models_pytorch/encoders/_senet.py)
+  * [segmentation_models_pytorch/encoders/_xception.py](https://github.com/qubvel/segmentation_models.pytorch/blob/main/segmentation_models_pytorch/encoders/_xception.py)
   * [segmentation_models_pytorch/decoders/deeplabv3/decoder.py](https://github.com/qubvel/segmentation_models.pytorch/blob/main/segmentation_models_pytorch/decoders/deeplabv3/decoder.py)
 
 - Apache-2.0 License

pyproject.toml

@@ -20,9 +20,7 @@ dependencies = [
     'huggingface-hub>=0.24',
     'numpy>=1.19.3',
     'pillow>=8',
-    'pretrainedmodels>=0.7.1',
     'safetensors>=0.3.1',
-    'six>=1.5',
     'timm>=0.9',
     'torch>=1.8',
     'torchvision>=0.9',

requirements/minimum.old

@@ -1,9 +1,7 @@
 huggingface-hub==0.24.0
 numpy==1.19.3
 pillow==8.0.0
-pretrainedmodels==0.7.1
 safetensors==0.3.1
-six==1.5.0
 timm==0.9.0
 torch==1.9.0
 torchvision==0.10.0

requirements/required.txt

@@ -1,9 +1,7 @@
 huggingface_hub==0.28.0
 numpy==2.2.2
 pillow==11.1.0
-pretrainedmodels==0.7.4
 safetensors==0.5.2
-six==1.17.0
 timm==1.0.14
 torch==2.6.0
 torchvision==0.21.0

segmentation_models_pytorch/__init__.py

@@ -1,5 +1,3 @@
-import warnings
-
 from . import datasets
 from . import encoders
 from . import decoders
@@ -24,12 +22,6 @@
 from typing import Optional as _Optional
 import torch as _torch
 
-# Suppress the specific SyntaxWarning for `pretrainedmodels`
-warnings.filterwarnings("ignore", message="is with a literal", category=SyntaxWarning)
-warnings.filterwarnings(
-    "ignore", message=r'"is" with \'str\' literal.*', category=SyntaxWarning
-)  # for python >= 3.12
-
 _MODEL_ARCHITECTURES = [
     Unet,
     UnetPlusPlus,