Files
Grounded-SAM-2/demo/backend/server/data/resolver.py
Ren Tianhe 82e503604f [New Feature] Support SAM 2.1 (#59)
* support sam 2.1

* refine config path and ckpt path

* update README
2024-10-10 14:55:50 +08:00

19 lines
522 B
Python

# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
# This source code is licensed under the license found in the
# LICENSE file in the root directory of this source tree.
from typing import Iterable
def resolve_videos(node_ids: Iterable[str], required: bool = False):
"""
Resolve videos given node ids.
"""
from data.store import get_videos
all_videos = get_videos()
return [
all_videos[nid] if required else all_videos.get(nid, None) for nid in node_ids
]