New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
How to inference own yolo model ? #58
Comments
I think this is because your cfg and weights files don't match (number of weights in the output convolutional layers). The original "yolov3-tiny-416.cfg" is for detecting 80 classes. So its output convolutional layers (right before the "yolo" layers) have (80 + 5) * 3 = 255 filters. Say, if your custom model is for detecting only 1 class, number of filters in those layers should be modified to (1 + 5) * 3 = 18. So you could find all lines of:
and replace them with:
|
您好,我看您是台灣人,我想說用中文表達應該會比英文清楚一點 ( 抱歉 英文太爛 ... ) |
你訓練的model是YOLOv3還是YOLOv3-Tiny? 如果是YOLOv3的話,請將cfg和weights檔名中的 "-tiny" 刪掉再重試一次。 |
您好,感謝您的回覆 |
Try this @brian208579 as @jkjung-avt said |
@saisubramani thx for your suggestion if 'tiny' in model: to if 'tiny' in model: but i got the same error ... I followed the step from the comments below |
Can you share your yolov3-tinny-416.cfg file? So that i can debug whats the error.
|
@saisubramani ok , appreciate [net] learning_rate=0.001 [convolutional] [maxpool] [convolutional] [maxpool] [convolutional] [maxpool] [convolutional] [maxpool] [convolutional] [maxpool] [convolutional] [maxpool] [convolutional] ########### [convolutional] [convolutional] [convolutional] [yolo] [route] [convolutional] [upsample] [route] [convolutional] [convolutional] [yolo] |
@saisubramani I already solved my problem , the reason is I didn't add spaces at the bottom of cfg , thanks for your help @jkjung-avt @saisubramani |
its good @brian208579 |
@jkjung-avt
I try your thoughts in this Disqus post , and put my custom yolo weight & cfg in /yolov3_onnx.
Then excute python3 yolov3_to_onnx.py --model yolov3-416
I got the error below the figure
Can you give me some advice ?
Thanks you a lot ...
The text was updated successfully, but these errors were encountered: