Tf.data.Dataset.filter(function) не может получить доступ к каталогу ⇐ Python

[Anonymous]

поэтому я пытаюсь сбалансировать набор данных, удалив лишние изображения из каждой метки.

Сначала я получаю количество для каждой метки и нахожу минимальное значение изображений для каждой метки, после этого нам нужно создать функцию фильтра для удаления лишних изображений.

def filter_dataset(изображение, метка): печать (label_counts) защита decrement_count (метка): label_counts[метка] -= 1 count = label_counts[метка] filter_condition = количество