TypeError: Cannot create initializer for non-floating point type . When running "train_tripletloss.py"

Question

我是张量流和模型训练的新手。我使用的是基于yolo和facenet的人脸识别算法。我现在正在尝试训练自己的模型。但是每次我这样做都会出错。 如果您能帮我解决，我将不胜感激。先感谢您。 这是代码的 link ： https://github.com/AzureWoods/faceRecognition-yolo-facenet/blob/master/train_tripletloss.py

这里是错误：

  File "C:\Users\user1\AppData\Roaming\Python\Python36\site-packages\tensorflow_core\contrib\layers\python\layers\initializers.py", line 120, in _initializer
    raise TypeError('Cannot create initializer for non-floating point type.')
TypeError: Cannot create initializer for non-floating point type.

Answer 1

尝试在 train_tripleloss.py 中添加 image = tf.to_float(image)，如下图所示。