Exception encountered when calling layer “tf.math.multiply_4“ (type TFOpLambda).

Exception encountered when calling layer “tf.math.multiply_4” (type TFOpLambda).

Dimensions must be equal, but are 2 and 80 for ‘{{node tf.math.multiply_2/Mul}} = Mul[T=DT_FLOAT](Placeholder, Placeholder_1)’ with input shapes: [?,19,19,5,2], [?,19,19,5,80].
解决:将#yolo_outputs = yolo_head(yolo_model.output, anchors, len(class_names))
改为:

yolo_outputs = yolo_head(yolo_model.output, anchors, len(class_names))
box_xy, box_wh, box_confidence, box_class_probs = yolo_outputs
yolo_outputs = (box_confidence, box_xy, box_wh, box_class_probs)

原因:维度不匹配不能乘,需要将参数更换顺序。

接着可能会报另一个错,如果没有请忽略
‘NoneType’ object has no attribute ‘run’
这需要重新运行之前的会话 tf.session运行之后我的问题就解决了

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值