search

RobotEdh / Yolov-4

Yolo v4 using TensorFlow 2.x
110 stars 30 forks source link

ValueError: Cannot reshape a tensor with 2945760 elements to shape [2,76,76,3,36] (1247616 elements) #17

Open StevenW1999 opened 2 years ago

StevenW1999 commented 2 years ago

When training the model from scratch with 608x608 images i get the following error:

ValueError: Cannot reshape a tensor with 2945760 elements to shape [2,76,76,3,36] (1247616 elements) for '{{node custom_loss/Reshape_1}} = Reshape[T=DT_FLOAT, Tshape=DT_INT32](Yolo_v4/BN_138/FusedBatchNormV3, custom_loss/Reshape_1/shape)' with input shapes: [2,76,76,255], [5] and with input tensors computed as partial shapes: input[1] = [2,76,76,3,36].

I am not sure where it goes wrong, could you help me please? Thank you!