Skip to content
This repository has been archived by the owner on Oct 12, 2022. It is now read-only.

Releases: Basars/models

v1.0 - Basars models

14 Feb 00:42
Compare
Choose a tag to compare

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')