-
Notifications
You must be signed in to change notification settings - Fork 62
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
Doesn't work with models trained with latest version of YoloV5 #14
Comments
@rajeshpt that is right - we will provide an update for yolov5v6.0, and in the meantime provide a hash of the version we have pinned here cc @maxhs2014 |
@Jacobsolawetz thanks for your quick response. Just as a suggestion, would it make sense not to keep a copy of the yolo source code and instead just use the torch.hub.load to load the model. That way this repo would work irrespective of yolo version changes. |
@rajeshpt interesting... can you use torch.hub.load with models that you have trained on your custom dataset? |
yes, you can. I am currently doing that and that is the recommended approach to avoid tight dependency with the yolo source and probably also from licensing perspective if one is developing a commercial product. |
@rajeshpt I'm trying to implement your fix, but I'm struggling to get it to run. Would you mind please uploading an example? |
Loading your custom model - for Inference - You can look at this for more details - ultralytics/yolov5#36 |
Is v6.0 still not supported? Thanks in advance! 😄 |
@JNaranjo-Alcazar it is still not supported, @maxhs2014 and I will take a look when we get a moment, and PRs always welcome :D |
Jacobsolawetz |
There is now a YOLOv5 v7.0, I'm guessing it is still not up to date to work with this? |
Models created/trained with latest version of YoloV5 don't load with this repo code. The YoloV5 code you have needs to be updated.
The text was updated successfully, but these errors were encountered: