Upload test_mgie.py with huggingface_hub
Browse files- test_mgie.py +2 -1
test_mgie.py
CHANGED
|
@@ -95,8 +95,9 @@ SEED = 13331
|
|
| 95 |
|
| 96 |
data_path = '/home/zbz5349/WorkSpace/aigeeks/Qwen2.5-VL/magicbrush_dataset/genp2_4_single.json'
|
| 97 |
save_image = '/home/zbz5349/WorkSpace/aigeeks/ml-mgie/mix'
|
| 98 |
-
save_attention_map = "/home/zbz5349/WorkSpace/aigeeks/ml-mgie/
|
| 99 |
os.makedirs(save_image,exist_ok=True)
|
|
|
|
| 100 |
# 若有x个指令那么生成x(single) + x(mix) + 1(all)张图片
|
| 101 |
data = read_json(data_path)
|
| 102 |
for i in tqdm(range(1)):
|
|
|
|
| 95 |
|
| 96 |
data_path = '/home/zbz5349/WorkSpace/aigeeks/Qwen2.5-VL/magicbrush_dataset/genp2_4_single.json'
|
| 97 |
save_image = '/home/zbz5349/WorkSpace/aigeeks/ml-mgie/mix'
|
| 98 |
+
save_attention_map = "/home/zbz5349/WorkSpace/aigeeks/ml-mgie/single_attention_map"
|
| 99 |
os.makedirs(save_image,exist_ok=True)
|
| 100 |
+
os.makedirs(save_attention_map,exist_ok=True)
|
| 101 |
# 若有x个指令那么生成x(single) + x(mix) + 1(all)张图片
|
| 102 |
data = read_json(data_path)
|
| 103 |
for i in tqdm(range(1)):
|