INNER CODE UNIT · Python

positive_queries

yurijmikhalevich/rclip · rclip/main.py:310

    positive_queries = [query] + positive_queries
    helpers.raise_if_cancelled(cancel_event)
    positive_features = self._model.compute_features_for_queries(positive_queries)
    helpers.raise_if_cancelled(cancel_event)
    negative_features = self._model.compute_features_for_queries(negative_queries)
    helpers.raise_if_cancelled(cancel_event)
    query_features = positive_features - negative_features

    # exclude images that were part of the query from the results
    exclude_files = {
      os.path.abspath(query) for query in positive_queries + negative_queries if helpers.is_file_path(query)
    }

    def iter_results() -> Iterator[RClip.SearchResult]:
      images = iter(self._db.get_image_vectors_by_dir_path(directory))
      while True:
        batch_size = 0
        filepaths: list[str] = []

View source record →