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

1 次查看(过去 30 天)
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])."

回答(1 个)

Sourav Bairagya
Sourav Bairagya 2019-12-18
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 个评论
Susama Bagchi
Susama Bagchi 2019-12-22
编辑:Susama Bagchi 2019-12-22
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.

请先登录,再进行评论。

类别

Help CenterFile Exchange 中查找有关 Recognition, Object Detection, and Semantic Segmentation 的更多信息

产品


版本

R2019b

Community Treasure Hunt

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

Start Hunting!

Translated by