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

11 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.