Spaces:
Sleeping
Sleeping
fixed bugs
Browse files
app.py
CHANGED
|
@@ -27,7 +27,7 @@ mae_model = MAEViT(
|
|
| 27 |
encoder_layers=2, #originally 2
|
| 28 |
encoder_heads=4, #originally 4
|
| 29 |
mlp_ratio=2.0, #originally 2.0
|
| 30 |
-
mask_ratio=0.
|
| 31 |
decoder_embed_dim=64, #originally 64
|
| 32 |
decoder_layers=2, #originally 2
|
| 33 |
decoder_heads=4, #originally 4
|
|
@@ -65,7 +65,7 @@ yolo_mapping = {
|
|
| 65 |
8: "Surprised"
|
| 66 |
}
|
| 67 |
|
| 68 |
-
def mae_reconstruct(image:Image, figure_name='figure/demo_temp.png', mask_ratio):
|
| 69 |
img = transform(image).unsqueeze(0)
|
| 70 |
img = img.to(device)
|
| 71 |
mae_model.mask_ratio = mask_ratio
|
|
|
|
| 27 |
encoder_layers=2, #originally 2
|
| 28 |
encoder_heads=4, #originally 4
|
| 29 |
mlp_ratio=2.0, #originally 2.0
|
| 30 |
+
mask_ratio=0.75,
|
| 31 |
decoder_embed_dim=64, #originally 64
|
| 32 |
decoder_layers=2, #originally 2
|
| 33 |
decoder_heads=4, #originally 4
|
|
|
|
| 65 |
8: "Surprised"
|
| 66 |
}
|
| 67 |
|
| 68 |
+
def mae_reconstruct(image:Image, figure_name='figure/demo_temp.png', mask_ratio=0.75):
|
| 69 |
img = transform(image).unsqueeze(0)
|
| 70 |
img = img.to(device)
|
| 71 |
mae_model.mask_ratio = mask_ratio
|