true_fn和false_fn输出的dtype类型不一致怎么办

qfqf123456

于 2021-03-10 12:14:16 发布

阅读量566

点赞数 2

分类专栏：深度学习 python报错分析文章标签： python 深度学习 tensorflow

本文链接：https://blog.youkuaiyun.com/qfqf123456/article/details/114632136

版权

在使用预训练的facenet模型进行验证时遇到类型不匹配错误，原始数据类型为uint8，模型期望为float32。通过将uint8转换为float32后，又出现与identity操作符的类型不一致问题。为解决此问题，采用TensorFlow的cast函数将identity的输出转换为float32。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

使用预训练的facenet模型在验证集上跑，中间报了个类型不匹配的错误：
原始：当前的话是一个uint8，一个是float32.

 image = tf.cond(get_control_flag(control[0], RANDOM_ROTATE),
                            lambda:tf.compat.v1.py_func(random_rotate_image, [image

最低0.47元/天解锁文章