Блог

Главная - Вопросы по программированию - tf.boolean_mask не принимает аргумент axis

tf.boolean_mask не принимает аргумент axis

#python #tensorflow #keras #yolo

#python #тензорный поток #keras #yolo

Вопрос:

Вот мой код: 

  44     scores = tf.boolean_mask(box_class_scores,filtering_mask,axis=-1)
 45     boxes = tf.boolean_mask(boxes,filtering_mask,axis=-1)
 46     classes = tf.boolean_mask(box_classes,filtering_mask,axis=-1)

Ошибка, я получаю:

Ошибка типа: boolean_mask() получила неожиданный аргумент ключевого слова ‘axis’

tf.boolean_mask() Не принимает аргумент axis, но является допустимым аргументом, как видно из документации: https://www.tensorflow.org/api_docs/python/tf/boolean_mask

введите описание изображения здесь

Комментарии:

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

2. @SUSHANTHIWARY Вы проходите специализированный курс глубокого обучения? Я сталкиваюсь с той же проблемой уже 2 дня. Пожалуйста, помогите мне, как это решить

Ответ №1:

Если вы удалите axis аргумент из функции, он будет работать .Ошибка, которую вы получаете, будет устранена.

Ответ №2:

Как указал @syltruong, проверьте, соответствует ли используемая вами версия Keras документации. Прямо сейчас в документации используется версия 2.2.4.

Кстати, @Ganesh и @Sushant, на самом деле вам не нужно использовать параметр axis для этого конкретного фрагмента кода в специализации глубокого обучения.