This repository has been archived by the owner on Oct 12, 2022. It is now read-only.
Releases: Basars/models
Releases · Basars/models
v1.0 - Basars models
This release provides final results of the Basars project.
Naming Convention
stairs
model have conv3x3 (256, 128, 64, 32, 16) → conv1x1 (5)
upsamples
proj
model have conv3x3 (256, 128, 64, 16) → conv1x1 (5, 5)
upsamples
Model Definition
stairs
model:
model = Sequential(name='ViT-stairs', layers=[
VisionTransformer(input_shape=(224, 224, 3),
upsample_channels=(256, 128, 64, 32),
output_kernel_size=3, num_classes=16),
Conv2D(5, kernel_size=(1, 1), padding='same', activation='softmax', use_bias=False)
])
model.load_weights('basars-cls5-stairs.h5')
proj
model:
Sequential(name='ViT-proj', layers=[
VisionTransformer(input_shape=(224, 224, 3), num_classes=5),
Conv2D(5, kernel_size=(1, 1), padding='same', activation='softmax', use_bias=False)
])
model.load_weights('basars-cls5-proj.h5')