Vendor pretrainedmodels

qubvel-org · Jan 18, 2025 · 663117b · 663117b
1 parent 980aefe
commit 663117b
Show file tree

Hide file tree

Showing 16 changed files with 1,888 additions and 226 deletions.
diff --git a/docs/conf.py b/docs/conf.py
@@ -100,7 +100,6 @@ def get_version():
     "timm",
     "cv2",
     "PIL",
-    "pretrainedmodels",
     "torchvision",
     "segmentation_models_pytorch.encoders",
     "segmentation_models_pytorch.utils",

diff --git a/licenses/LICENSES.md b/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

diff --git a/pyproject.toml b/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',

diff --git a/requirements/minimum.old b/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

diff --git a/requirements/required.txt b/requirements/required.txt
@@ -1,9 +1,7 @@
 huggingface_hub==0.27.1
 numpy==2.2.1
 pillow==11.1.0
-pretrainedmodels==0.7.4
 safetensors==0.5.2
-six==1.17.0
 timm==1.0.13
 torch==2.5.1
 torchvision==0.20.1

diff --git a/segmentation_models_pytorch/__init__.py b/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,