Skip to content

Commit c39de0c

Browse files
authored
Release update 0.3.2 (#705)
* Update timm encoders list * Update flake8 config * Update author name * Bump version
1 parent 478b7d2 commit c39de0c

File tree

8 files changed

+213
-46
lines changed

8 files changed

+213
-46
lines changed

.flake8

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
[flake8]
22
max-line-length = 119
3-
exclude =.git,__pycache__,docs/conf.py,build,dist,setup.py,tests
3+
exclude =.git,__pycache__,docs/conf.py,build,dist,setup.py,tests,.venv
44
ignore = I101,I201,F401,F403,S001,D100,D101,D102,D103,D104,D105,D106,D107,D200,D205,D400,W504,D202,E203,W503,B006,D412
55
inline-quotes = "

HALLOFFAME.md

+3-3
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ Please, follow these rules, when adding a solution to the "Hall of Fame":
3232
[[description](https://www.kaggle.com/c/severstal-steel-defect-detection/discussion/114465#latest-659615)]
3333

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

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

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

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

9191
### [Machine Learning based feature extraction of Electrical Substations from Satellite Data ](https://competitions.codalab.org/competitions/32132#learn_the_details)

LICENSE

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
The MIT License
22

3-
Copyright (c) 2019, Pavel Yakubovskiy
3+
Copyright (c) 2019, Pavel Iakubovskii
44

55
Permission is hereby granted, free of charge, to any person obtaining a copy
66
of this software and associated documentation files (the "Software"), to deal

README.md

+4-4
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ The main features of this library are:
2020

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

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

404404
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
405405

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

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

412412
### 🔁 Models API <a name="api"></a>

docs/conf.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,8 @@
2323
# -- Project information -----------------------------------------------------
2424

2525
project = 'Segmentation Models'
26-
copyright = '{}, Pavel Yakubovskiy'.format(datetime.datetime.now().year)
27-
author = 'Pavel Yakubovskiy'
26+
copyright = '{}, Pavel Iakubovskii'.format(datetime.datetime.now().year)
27+
author = 'Pavel Iakubovskii'
2828

2929
def get_version():
3030
sys.path.append('../segmentation_models_pytorch')

docs/encoders_timm.rst

+199-34
Large diffs are not rendered by default.

misc/generate_table_timm.py

+2
Original file line numberDiff line numberDiff line change
@@ -51,4 +51,6 @@ def make_table(data):
5151

5252
table = make_table(supported_models)
5353
print(table)
54+
with open("timm_encoders.txt", "w") as f:
55+
print(table, file=f)
5456
print(f"Total encoders: {len(supported_models.keys())}")
+1-1
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,3 @@
1-
VERSION = (0, 3, 1)
1+
VERSION = (0, 3, 2)
22

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

0 commit comments

Comments
 (0)