demo.py

# ---------------------------------------------------------------------
# Copyright (c) 2025 Qualcomm Technologies, Inc. and/or its subsidiaries.
# SPDX-License-Identifier: BSD-3-Clause
# ---------------------------------------------------------------------

from qai_hub_models.models._shared.sam2.demo import sam2_demo_main
from qai_hub_models.models.edgetam.model import (
    DEFAULT_MODEL_TYPE,
    MODEL_ASSET_VERSION,
    MODEL_ID,
    EdgeTAM,
)
from qai_hub_models.utils.asset_loaders import CachedWebModelAsset

IMAGE_ADDRESS = CachedWebModelAsset.from_asset_store(
    MODEL_ID, MODEL_ASSET_VERSION, "truck.jpg"
)


# Run EdgeTAM end-to-end model on given image.
# The demo will output image with segmentation mask applied for input points
def main(is_test: bool = False) -> None:
    sam2_demo_main(
        model_cls=EdgeTAM,
        model_id=MODEL_ID,
        default_image=IMAGE_ADDRESS,
        default_model_type=DEFAULT_MODEL_TYPE,
        is_test=is_test,
    )


if __name__ == "__main__":
    main()