Source code for tbp.monty.frameworks.models.salience.strategies

# Copyright 2025 Thousand Brains Project
#
# Copyright may exist in Contributors' modifications
# and/or contributions to the work.
#
# Use of this source code is governed by the MIT
# license that can be found in the LICENSE file or at
# https://opensource.org/licenses/MIT.
from __future__ import annotations

from typing import Protocol

import numpy as np


[docs]class SalienceStrategy(Protocol):
    def __call__(self, rgba: np.ndarray, depth: np.ndarray) -> np.ndarray: ...


[docs]class UniformSalienceStrategy(SalienceStrategy):
    def __call__(self, rgba: np.ndarray, depth: np.ndarray) -> np.ndarray:
        return np.ones_like(depth)