oneflow vision codebase
Project description
vision
Datasets, Transforms and Models specific to Computer Vision
Usage
ModelCreator
- Create model in a simple way
from flowvision.models import ModelCreator
model = ModelCreator.create_model('alexnet', pretrained=True)
the pretrained weight will be saved to ./checkpoints
- Supported model table
from flowvision.models import ModelCreator
ModelCreator.model_table()
Models
┏━━━━━━━━━━━━━━┳━━━━━━━━━━━━┓
┃ Name ┃ Pretrained ┃
┡━━━━━━━━━━━━━━╇━━━━━━━━━━━━┩
│ alexnet │ true │
│ vit_b_16_224 │ false │
│ vit_b_16_384 │ true │
│ vit_b_32_224 │ false │
│ vit_b_32_384 │ true │
│ vit_l_16_384 │ true │
│ vit_l_32_384 │ true │
└──────────────┴────────────┘
show all of the supported model in the table manner
- List models with pretrained weights
from flowvision.models import ModelCreator
ModelCreator.model_table(pretrained=True)
Models
┏━━━━━━━━━━━━━━┳━━━━━━━━━━━━┓
┃ Name ┃ Pretrained ┃
┡━━━━━━━━━━━━━━╇━━━━━━━━━━━━┩
│ alexnet │ true │
│ vit_b_16_384 │ true │
│ vit_b_32_384 │ true │
│ vit_l_16_384 │ true │
│ vit_l_32_384 │ true │
└──────────────┴────────────┘
- Search for model by Wildcard
from flowvision.models import ModelCreator
ModelCreator.model_table('vit*')
Models
┏━━━━━━━━━━━━━━┳━━━━━━━━━━━━┓
┃ Name ┃ Pretrained ┃
┡━━━━━━━━━━━━━━╇━━━━━━━━━━━━┩
│ vit_b_16_224 │ false │
│ vit_b_16_384 │ true │
│ vit_b_32_224 │ false │
│ vit_b_32_384 │ true │
│ vit_l_16_384 │ true │
│ vit_l_32_384 │ true │
└──────────────┴────────────┘
- Search for model with pretrained weights by Wildcard
from flowvision.models import ModelCreator
ModelCreator.model_table('vit*', pretrained=True)
Models
┏━━━━━━━━━━━━━━┳━━━━━━━━━━━━┓
┃ Name ┃ Pretrained ┃
┡━━━━━━━━━━━━━━╇━━━━━━━━━━━━┩
│ vit_b_16_384 │ true │
│ vit_b_32_384 │ true │
│ vit_l_16_384 │ true │
│ vit_l_32_384 │ true │
└──────────────┴────────────┘
Model Zoo
We did all our tests under the same setting, please check the model page here for more details.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
flowvision-0.0.2.tar.gz
(26.1 kB
view hashes)