TypeError: reduce_sum() got an unexpected keyword argument 'reduction_indices'?

Question

在使用了我在 tf == 1 中写的这个函数后，我更新了 tensorflow 2.0。我遇到了下面提到的相同错误

    colors = tf.constant(img, dtype=tf.float32)
    model = tf.keras.models.model_from_json(json.load(open("model.json"))["model"], custom_objects={})
    model.load_weights("model_weights.h5")
    predictions = model.predict(colors, batch_size=32, verbose=0)
    # Output is one-hot vector for 9 class:["red","green","blue","orange","yellow","pink", "purple","brown","grey"]
    predictions = tf.one_hot(np.argmax(predictions, 1), 9)
    # Sum along the column, each entry indicates no of pixels
    res = tf.reduce_sum(predictions, reduction_indices= 0 ).numpy()
    # Threshold is 0.5 (accuracy is 96%) change threshold may cause accuracy decrease
    if res[0] / (sum(res[:-1]) + 1) > 0.5:
        return "red"
    elif res[1] / (sum(res[:-1]) + 1) > 0.5:
        return "green"
    elif res[2] / (sum(res[:-1]) + 1) > 0.5:
        return "blue"
    else:
        return "other"

错误信息如下 TypeError: reduce_sum() got an unexpected keyword argument 'reduction_indices'

Answer 1

我认为您的问题是 reduction_indices 在 Tensorflow 2.x 中已被弃用，所以请尝试这样做：

tf.reduce_sum(predictions, axis= 0)

这是等价的。

TypeError: reduce_sum() got an unexpected keyword argument 'reduction_indices'?

TypeError: reduce_sum() got an unexpected keyword argument 'reduction_indices'?

python

image-processing

computer-vision

tensorflow