PyTorch-Encoding

A CV toolkit for my papers.

MIT License

Downloads
1.6K
Stars
2K

Bot releases are hidden (Show)

PyTorch-Encoding - Wasabi Url Latest Release

Published by zhanghang1989 over 4 years ago

Change model zoo to wasabi to save model

PyTorch-Encoding - PyTorch Encoding v1.2.0

Published by zhanghang1989 over 4 years ago

ResNeSt

Split-Attention Network, A New ResNet Variant. It significantly boosts the performance of downstream models such as Mask R-CNN, Cascade R-CNN and DeepLabV3.

crop size PyTorch Gluon
ResNeSt-50 224 81.03 81.04
ResNeSt-101 256 82.83 82.81
ResNeSt-200 320 83.84 83.88
ResNeSt-269 416 84.54 84.53

Semantic Segmentation

Results on ADE20K

PyTorch-Encoding - PyTorch Encoding v1.0.1

Published by zhanghang1989 over 4 years ago

  1. Compatible with stable PyTorch 14..0.
PyTorch-Encoding - PyTorch Encoding v1.0.0

Published by zhanghang1989 almost 6 years ago

  1. Compatible with stable PyTorch 1.0.
  2. Inplace ABN with SyncBN.
  3. ImageNet training script
PyTorch-Encoding - PyTorch Encoding 0.4.3

Published by zhanghang1989 over 6 years ago

  1. Segmentation examples and pretrained models.
  2. Pypi install through pip install torch-encoding
PyTorch-Encoding - Encoding Layer & Sync BN

Published by zhanghang1989 almost 7 years ago

Make the compilation easier to the users.

PyTorch-Encoding - Encoding Layer

Published by zhanghang1989 about 7 years ago

Memory efficient implementation of Encoding Layer.

Package Rankings
Top 5.15% on Pypi.org
Badges
Extracted from project README
License: MIT Build Docs Unit Test PWC PWC
Related Projects