Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Vendor pretrainedmodels #1039

Open
wants to merge 4 commits into
base: main
Choose a base branch
from
Open

Vendor pretrainedmodels #1039

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
663117b
Vendor pretrainedmodels
adamjstewart Jan 18, 2025
0da5651
ruff
adamjstewart Jan 18, 2025
a59d486
Remove accidentally deleted code
adamjstewart Jan 18, 2025
6c68ae4
Compatibility with newer version of xception code
adamjstewart Jan 18, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 0 additions & 1 deletion docs/conf.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -100,7 +100,6 @@ def get_version():
"timm",
"cv2",
"PIL",
"pretrainedmodels",
"torchvision",
"segmentation_models_pytorch.encoders",
"segmentation_models_pytorch.utils",
Expand Down
8 changes: 6 additions & 2 deletions licenses/LICENSES.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
2 changes: 0 additions & 2 deletions pyproject.toml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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',
Expand Down
2 changes: 0 additions & 2 deletions requirements/minimum.old
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -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
Expand Down
2 changes: 0 additions & 2 deletions requirements/required.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -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
Expand Down
8 changes: 0 additions & 8 deletions segmentation_models_pytorch/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,5 +1,3 @@
import warnings

from . import datasets
from . import encoders
from . import decoders
Expand All @@ -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,
Expand Down
Loading
Loading