keras-yolo3 icon indicating copy to clipboard operation
keras-yolo3 copied to clipboard

training tiny-yolov3 problems

Open mechealww2 opened this issue 6 years ago • 5 comments

Create Tiny YOLOv3 model with 6 anchors and 8 classes. C:\Users\LYK_pc\Anaconda3\lib\site-packages\keras\engine\saving.py:1140: UserWarning: Skipping loading of weights for layer conv2d_10 due to mismatch in shape ((1, 1, 512, 39) vs (255, 512, 1, 1)). weight_values[i].shape)) C:\Users\LYK_pc\Anaconda3\lib\site-packages\keras\engine\saving.py:1140: UserWarning: Skipping loading of weights for layer conv2d_10 due to mismatch in shape ((39,) vs (255,)). weight_values[i].shape)) C:\Users\LYK_pc\Anaconda3\lib\site-packages\keras\engine\saving.py:1140: UserWarning: Skipping loading of weights for layer conv2d_13 due to mismatch in shape ((1, 1, 256, 39) vs (255, 256, 1, 1)). weight_values[i].shape)) C:\Users\LYK_pc\Anaconda3\lib\site-packages\keras\engine\saving.py:1140: UserWarning: Skipping loading of weights for layer conv2d_13 due to mismatch in shape ((39,) vs (255,)). weight_values[i].shape))

Why there are mismatch?And I do not get good performance on my own dataset.Training stopped at val_loss =27... Anyone can help? Thks.

mechealww2 avatar Nov 17 '18 05:11 mechealww2

Hi, I am experiencing the same issue. Any news on that?

samuelefiorini avatar Mar 06 '19 17:03 samuelefiorini

Sorry,I do not find any solution. But i find keras yolov2 and ssd have better performance on my task.

mechealww2 avatar Mar 07 '19 09:03 mechealww2

Thanks @mechealww2, which implementation of yolov2 are you using?

samuelefiorini avatar Mar 07 '19 09:03 samuelefiorini

https://github.com/experiencor/keras-yolo2

mechealww2 avatar Mar 12 '19 01:03 mechealww2

The mismatch is because you are trying to train your own dataset with 8 classes. Last layer output depend on it because of equation "num_anchors*(num_classes+5)" which is in darknet model 3x(80+5)=255 and in your case 3x(8+5)=39

DaliborCimr avatar Jun 30 '22 12:06 DaliborCimr