Expand bbox

app.py

@@ -46,6 +46,12 @@ DEFAULT_CHECKPOINT='_DATA/hamer_ckpts/checkpoints/hamer.ckpt'
 device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
 model_cfg = str(Path(DEFAULT_CHECKPOINT).parent.parent / 'model_config.yaml')
 model_cfg = get_config(model_cfg)
+# Override some config values, to crop bbox correctly
+if (model_cfg.MODEL.BACKBONE.TYPE == 'vit') and ('BBOX_SHAPE' not in model_cfg.MODEL):
+    model_cfg.defrost()
+    assert model_cfg.MODEL.IMAGE_SIZE == 256, f"MODEL.IMAGE_SIZE ({model_cfg.MODEL.IMAGE_SIZE}) should be 256 for ViT backbone"
+    model_cfg.MODEL.BBOX_SHAPE = [192,256]
+    model_cfg.freeze()
 model = HAMER.load_from_checkpoint(DEFAULT_CHECKPOINT, strict=False, cfg=model_cfg).to(device)
 model.eval()