Update README.md
Browse files
README.md
CHANGED
|
@@ -106,7 +106,7 @@ from transformers import AutoTokenizer
|
|
| 106 |
from internvl.model.internvl_chat import InternVLChatConfig, InternVisionModel
|
| 107 |
from utils import post_process, generate_similiarity_map, load_model, load_image
|
| 108 |
|
| 109 |
-
checkpoint = '/
|
| 110 |
image_path = './demo_images/0000000.png'
|
| 111 |
input_query = '11/12/2020'
|
| 112 |
out_dir = 'results'
|
|
@@ -146,11 +146,6 @@ attn_map = similarity.reshape(len(input_embedings), resized_size[0], resized_siz
|
|
| 146 |
|
| 147 |
"""generate map locally """
|
| 148 |
generate_similiarity_map(images, attn_map, target_aspect_ratio, all_bpe_strings, out_dir)
|
| 149 |
-
|
| 150 |
-
|
| 151 |
-
"""user command """
|
| 152 |
-
# python quick_start.py
|
| 153 |
-
|
| 154 |
```
|
| 155 |
|
| 156 |
### Evaluation on Vision Capability
|
|
|
|
| 106 |
from internvl.model.internvl_chat import InternVLChatConfig, InternVisionModel
|
| 107 |
from utils import post_process, generate_similiarity_map, load_model, load_image
|
| 108 |
|
| 109 |
+
checkpoint = 'xxx/checkpoint-370000'
|
| 110 |
image_path = './demo_images/0000000.png'
|
| 111 |
input_query = '11/12/2020'
|
| 112 |
out_dir = 'results'
|
|
|
|
| 146 |
|
| 147 |
"""generate map locally """
|
| 148 |
generate_similiarity_map(images, attn_map, target_aspect_ratio, all_bpe_strings, out_dir)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 149 |
```
|
| 150 |
|
| 151 |
### Evaluation on Vision Capability
|