INNER CODE UNIT · Python
iter_results
yurijmikhalevich/rclip · rclip/main.py:323
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:
continue
filepaths.append(filepath)
features.append(np.frombuffer(image["vector"], np.float32))
if not batch_size:
break
if not features:
continue