Skip to content

Release update 0.3.2 #705

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.

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

Merged
merged 4 commits into from
Jan 7, 2023
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
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
2 changes: 1 addition & 1 deletion .flake8
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
[flake8]
max-line-length = 119
exclude =.git,__pycache__,docs/conf.py,build,dist,setup.py,tests
exclude =.git,__pycache__,docs/conf.py,build,dist,setup.py,tests,.venv
ignore = I101,I201,F401,F403,S001,D100,D101,D102,D103,D104,D105,D106,D107,D200,D205,D400,W504,D202,E203,W503,B006,D412
inline-quotes = "
6 changes: 3 additions & 3 deletions HALLOFFAME.md
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@ Please, follow these rules, when adding a solution to the "Hall of Fame":
[[description](https://www.kaggle.com/c/severstal-steel-defect-detection/discussion/114465#latest-659615)]

- 12th place.
[Pavel Yakubovskiy](https://www.linkedin.com/in/pavel-yakubovskiy/),
[Pavel Iakubovskii](https://www.linkedin.com/in/pavel-iakubovskii/),
[Ilya Dobrynin](https://www.linkedin.com/in/ilya-dobrynin-79a89b106/),
[Denis Kolpakov](https://www.linkedin.com/in/denis-kolpakov-ab3137197/)
[[description](https://www.kaggle.com/c/severstal-steel-defect-detection/discussion/114309#latest-661404)]
Expand Down Expand Up @@ -72,7 +72,7 @@ Please, follow these rules, when adding a solution to the "Hall of Fame":
[[code](https://github.com/khornlund/understanding-cloud-organization)]

- 55th place.
[Pavel Yakubovskiy](https://www.linkedin.com/in/pavel-yakubovskiy/)
[Pavel Iakubovskii](https://www.linkedin.com/in/pavel-iakubovskii/)
[[description](https://www.kaggle.com/c/understanding_cloud_organization/discussion/118019#latest-678626)]

## Other platforms
Expand All @@ -85,7 +85,7 @@ Please, follow these rules, when adding a solution to the "Hall of Fame":

### [Open Cities AI Challenge: Segmenting Buildings for Disaster Resilience](https://www.drivendata.org/competitions/60/building-segmentation-disaster-resilience/)
- 1st place.
[Pavel Yakubovskiy](https://www.linkedin.com/in/pavel-yakubovskiy/).
[Pavel Iakubovskii](https://www.linkedin.com/in/pavel-iakubovskii/).
[[code and description](https://github.com/qubvel/open-cities-challenge)]

### [Machine Learning based feature extraction of Electrical Substations from Satellite Data ](https://competitions.codalab.org/competitions/32132#learn_the_details)
Expand Down
2 changes: 1 addition & 1 deletion LICENSE
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
The MIT License

Copyright (c) 2019, Pavel Yakubovskiy
Copyright (c) 2019, Pavel Iakubovskii

Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
Expand Down
8 changes: 4 additions & 4 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ The main features of this library are:

- High level API (just two lines to create a neural network)
- 9 models architectures for binary and multi class segmentation (including legendary Unet)
- 119 available encoders (and 400+ encoders from [timm](https://github.com/rwightman/pytorch-image-models))
- 124 available encoders (and 500+ encoders from [timm](https://github.com/rwightman/pytorch-image-models))
- All encoders have pre-trained weights for faster and better convergence
- Popular metrics and losses for training routines

Expand Down Expand Up @@ -403,10 +403,10 @@ Note: In the official github repo the s0 variant has additional num_conv_branche

Pytorch Image Models (a.k.a. timm) has a lot of pretrained models and interface which allows using these models as encoders in smp, however, not all models are supported

- transformer models do not have ``features_only`` functionality implemented
- some models do not have appropriate strides
- not all transformer models have ``features_only`` functionality implemented that is required for encoder
- some models have inappropriate strides

Total number of supported encoders: 467
Total number of supported encoders: 549
- [table with available encoders](https://smp.readthedocs.io/en/latest/encoders_timm.html)

### 🔁 Models API <a name="api"></a>
Expand Down
4 changes: 2 additions & 2 deletions docs/conf.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,8 +23,8 @@
# -- Project information -----------------------------------------------------

project = 'Segmentation Models'
copyright = '{}, Pavel Yakubovskiy'.format(datetime.datetime.now().year)
author = 'Pavel Yakubovskiy'
copyright = '{}, Pavel Iakubovskii'.format(datetime.datetime.now().year)
author = 'Pavel Iakubovskii'

def get_version():
sys.path.append('../segmentation_models_pytorch')
Expand Down
233 changes: 199 additions & 34 deletions docs/encoders_timm.rst

Large diffs are not rendered by default.

2 changes: 2 additions & 0 deletions misc/generate_table_timm.py
Original file line number Diff line number Diff line change
Expand Up @@ -51,4 +51,6 @@ def make_table(data):

table = make_table(supported_models)
print(table)
with open("timm_encoders.txt", "w") as f:
print(table, file=f)
print(f"Total encoders: {len(supported_models.keys())}")
2 changes: 1 addition & 1 deletion segmentation_models_pytorch/__version__.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
VERSION = (0, 3, 1)
VERSION = (0, 3, 2)

__version__ = ".".join(map(str, VERSION))