Skip to content

Commit 8bf812a

Browse files
committed
enh(run_inference): Use sam2.1
1 parent d14f749 commit 8bf812a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/osm_ai_helper/run_inference.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ def run_inference(
3535
output_dir: str,
3636
lat_lon: Tuple[float, float],
3737
margin: int = 1,
38-
sam_model: str = "facebook/sam2-hiera-small",
38+
sam_model: str = "facebook/sam2.1-hiera-small",
3939
selector: str = "leisure=swimming_pool",
4040
zoom: int = 18,
4141
):
@@ -52,7 +52,7 @@ def run_inference(
5252
margin (int, optional): Number of tiles around the location.
5353
Defaults to 1.
5454
sam_model (str, optional): [SAM2](https://github.com/facebookresearch/sam2) model to use.
55-
Defaults to "facebook/sam2-hiera-small".
55+
Defaults to "facebook/sam2.1-hiera-small".
5656
selector (str, optional): OpenStreetMap selector.
5757
Defaults to "leisure=swimming_pool".
5858
zoom (int, optional): Zoom level.

0 commit comments

Comments
 (0)