This commit is contained in:
pikaliov
2026-07-07 19:34:52 +03:00
parent 05355c6a7b
commit 4e76326c4f
5 changed files with 171 additions and 14 deletions

View File

@@ -292,6 +292,8 @@ def main(args):
num_workers=args.num_workers,
image_size=args.image_size,
build_test=False, # оценка идёт через multi-positive eval ниже
mutually_exclusive=args.mutually_exclusive,
seed=args.seed,
)
# Multi-positive eval (протокол §6.2): уникальная галерея + positive-карты.
@@ -485,7 +487,7 @@ def parse_args():
p.add_argument("--text_levels", nargs="+", default=["level1"])
p.add_argument("--train_meta", default="cross-area-drone2sate-train.json")
p.add_argument("--test_meta", default="cross-area-drone2sate-test.json")
p.add_argument("--image_size", type=int, default=384)
p.add_argument("--image_size", type=int, default=256)
p.add_argument("--num_workers", type=int, default=8)
# Model
@@ -496,6 +498,11 @@ def parse_args():
p.add_argument("--shared_dim", type=int, default=512)
p.add_argument("--inject_mona", action="store_true", default=True)
p.add_argument("--mona_bottleneck", type=int, default=64)
p.add_argument(
"--mutually_exclusive", action=argparse.BooleanOptionalAction, default=True,
help="Батчи без общих позитивных тайлов (§6.4); --no-mutually_exclusive "
"→ обычный random shuffle.",
)
# Training
p.add_argument("--epochs", type=int, default=50)