INNER CODE UNIT · Python

query_features

yurijmikhalevich/rclip · rclip/main.py:316

    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] = []
        features: list[model.FeatureVector] = []
        for image in itertools.islice(images, self.SEARCH_BATCH_SIZE):
          helpers.raise_if_cancelled(cancel_event)
          batch_size += 1
          filepath = image["filepath"]
          if self._is_excluded(filepath, directory) or filepath in exclude_files:

View source record →