bubbliiiing / yolov4-tiny-tf2

这是一个YoloV4-tiny-tf2的源码,可以用于训练自己的模型。
MIT License
133 stars 47 forks source link

老师你好我想问一个关于ca的问题 #16

Open iceharlse opened 2 years ago

iceharlse commented 2 years ago

在使用您的caattention代码后出现下面问题 File "D:\github\yolov4-tf2\nets\attention.py", line 140, in x_cat_conv_split_h, x_cat_conv_split_w = Lambda(lambda x: tf.split(x, num_or_size_splits=[h, w], axis=2))( TypeError: Exception encountered when calling layer "lambda_3" (type Lambda).

Failed to convert elements of [None, None] to Tensor. Consider casting elements to a supported type. See https://www.tensorflow.org/api_docs/python/tf/dtypes for supported TF dtypes.

Call arguments received by layer "lambda_3" (type Lambda): • inputs=tf.Tensor(shape=(None, 1, None, 16), dtype=float32) • mask=None • training=False

如何解决

lvy1999s commented 2 years ago

老哥,解决了吗?我用ca也有问题