After creating YOLO v2 with resnet50, while training the network, getting error "Invalid training data. The output size ([7 7 16]) of the last layer does not match the response size ([1 1 3])."

I followed the YOLO v2 object detection network as given in the below link:

https://in.mathworks.com/help/vision/ref/yolov2layers.html

Then I used the imagedatastore and augmented image datastore to modify the image size. After that, when I tried to train the network, got the error "Invalid training data. The output size ([7 7 16]) of the last layer does not match the response size ([1 1 3])."

Answer 1

[Sourav Bairagya]

From the error it seems that output of your network should have size [1 1 3], but it is having the size [7 7 16]. May be there is some error occured in the training input images during modification.

[Susama Bagchi]

If I only use the resnet50, with modified image size also it is working. But when I am adding YOLO V2 at the end of resnet50 as per the link, I am getting error. Even if I don't change the input layer size, then also output size mismatch error is there.

I hope, I can train the network after adding YOLO V2. If any other option, please let me know.