-
Notifications
You must be signed in to change notification settings - Fork 1.3k
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
Error while loading custom trained model #538
Comments
Have you faced this issue? I'm dealing with the same problem right now. Did you manage to find a solution? If so, could you please share it? I'd really appreciate any help you can offer! |
Hello @aziziselma so rightnow yolov9 int is not Autoshape compatible so I just transposed the image from (height, width, channels) to (channels, height, width). Although I was able to resolve this issue but I was getting a warning message stating that I won't be able to run inference with this model. The code that I have had written is:
So that's why I am now using yolov8-seg. If you get any solution for this problem please do share it with me. |
Thank you for your response. I am currently working on this issue as well. If I find a solution, I will definitely share it with you. |
Thanks!! All the Best |
I have trained my Yolov9 model for instance segmentation on my custom dataset, now when I am trying to load my model then I am getting an error.
The error I am getting is :
I even tried to use this code
But even this is not working
The text was updated successfully, but these errors were encountered: