Code error

[zhenhao-huang]

1.Data is augmented by three times, so line 123 of the train.py should be

positiveCount = len(positiveImageFiles)*3
negativeCount = len(negativeImageFiles)*3

2.The amount of FN and FP is reversed.

[amilkcar]

Hi @zhenhao-huang , what you did with this "2.The amount of FN and FP is reversed."

[zhenhao-huang]

I changed their location.

if y_test[i] == 0:
    if np.argmax(model_out[i, :]) == y_test[i]:
        TP = TP + 1
    else:
        FP = FP + 1
else:
    if np.argmax(model_out[i, :]) == y_test[i]:
        TN = TN + 1
    else:
        FN = FN + 1

[tkone2018]

@zhenhao-huang 你好，请问可以分享下你能够跑通的代码吗 包括推理一张图片的demo，谢谢