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

조회 수: 7(최근 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.

댓글을 달려면 로그인하십시오.

Community Treasure Hunt

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

Start Hunting!

Translated by