MATLAB Answers

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])."

3 views (last 30 days)
Susama Bagchi
Susama Bagchi on 9 Dec 2019
Edited: Susama Bagchi on 22 Dec 2019
I followed the YOLO v2 object detection network as given in the below link:
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])."

  0 Comments

Sign in to comment.

Answers (1)

Sourav Bairagya
Sourav Bairagya on 18 Dec 2019
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.

  1 Comment

Susama Bagchi
Susama Bagchi on 22 Dec 2019
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.

Sign in to comment.

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!