You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I have trained yolov3 (custom classes and grayscale images) and the inference seems to work good in python. I need to do inference in C++. For this reason I and using this work and the config file I have is for yolov3-tiny with channels=1. However, I have issues:
what(): shape '[255, 256, 1, 1]' is invalid for input of size 64992 while loading the weights. I use the same config file in python and I have no issue.
I figures that this is related to the output layer filters and I modified them to 24 (instead of 255) as I have 3 classes. I don't have this issue anymore but a new one
what(): Given groups=1, weight of size [16, 3, 3, 3], expected input[1, 1, 416, 416] to have 3 channels, but got 1 channels instead
I defined channels=1 in config during both training and inference process.
Hence, I am unable to run it in C++.
Please let me know if you know what could be the reason. Thanks!
The text was updated successfully, but these errors were encountered:
Hi,
I have trained yolov3 (custom classes and grayscale images) and the inference seems to work good in python. I need to do inference in C++. For this reason I and using this work and the config file I have is for yolov3-tiny with channels=1. However, I have issues:
I figures that this is related to the output layer filters and I modified them to 24 (instead of 255) as I have 3 classes. I don't have this issue anymore but a new one
I defined channels=1 in config during both training and inference process.
Hence, I am unable to run it in C++.
Please let me know if you know what could be the reason. Thanks!
The text was updated successfully, but these errors were encountered: