Spaces:
Running
on
Zero
Running
on
Zero
add paper
Browse files
app.py
CHANGED
@@ -491,7 +491,7 @@ def run_fn(
|
|
491 |
# resize the images before acquiring GPU
|
492 |
if "AlignedThreeModelAttnNodes" == model_name:
|
493 |
# dirty patch for the alignedcut paper
|
494 |
-
resolution = (
|
495 |
else:
|
496 |
resolution = RES_DICT[model_name]
|
497 |
images = [tup[0] for tup in images]
|
@@ -1021,7 +1021,7 @@ with demo:
|
|
1021 |
gr.Markdown('**Features are aligned across models and layers.** A linear alignment transform is trained for each model/layer, learning signal comes from 1) fMRI brain activation and 2) segmentation preserving eigen-constraints.')
|
1022 |
gr.Markdown('NCUT is computed on the concatenated graph of all models, layers, and images. Color is **aligned** across all models and layers.')
|
1023 |
gr.Markdown('')
|
1024 |
-
gr.Markdown("To see a good pattern, you will need to load 100 images. Running out of HuggingFace GPU Quota? Try [Demo](https://ncut-pytorch.readthedocs.io/en/latest/demo/) hosted at UPenn")
|
1025 |
gr.Markdown('---')
|
1026 |
with gr.Row():
|
1027 |
with gr.Column(scale=5, min_width=200):
|
|
|
491 |
# resize the images before acquiring GPU
|
492 |
if "AlignedThreeModelAttnNodes" == model_name:
|
493 |
# dirty patch for the alignedcut paper
|
494 |
+
resolution = (224, 224)
|
495 |
else:
|
496 |
resolution = RES_DICT[model_name]
|
497 |
images = [tup[0] for tup in images]
|
|
|
1021 |
gr.Markdown('**Features are aligned across models and layers.** A linear alignment transform is trained for each model/layer, learning signal comes from 1) fMRI brain activation and 2) segmentation preserving eigen-constraints.')
|
1022 |
gr.Markdown('NCUT is computed on the concatenated graph of all models, layers, and images. Color is **aligned** across all models and layers.')
|
1023 |
gr.Markdown('')
|
1024 |
+
gr.Markdown("To see a good pattern, you will need to load 100~1000 images. 100 images need 10sec for RTX4090. Running out of HuggingFace GPU Quota? Try [Demo](https://ncut-pytorch.readthedocs.io/en/latest/demo/) hosted at UPenn")
|
1025 |
gr.Markdown('---')
|
1026 |
with gr.Row():
|
1027 |
with gr.Column(scale=5, min_width=200):
|