No detections while using model weights from training custom dataset #3579
Unanswered
cooldecola
asked this question in
Q&A
Replies: 1 comment 1 reply
-
I think the syntax is cfg.MODEL.WEIGHTS = "model_final.pth" not cfg.MODEL.WEIGHTS = ("model_final.pth") |
Beta Was this translation helpful? Give feedback.
1 reply
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
Uh oh!
There was an error while loading. Please reload this page.
-
Hi, had a custom dataset that I trained using Detectron2 and was happy with the AP value, now I'm trying to use the "model_final.pth" to run inference on a video, however whenever I do this I get no detections on any of the frames. However when I use pretrained weights from detectron2's model zoo it works. Why is this?
This is how I'm loading the weights:
cfg.MODEL.WEIGHTS = ("model_final.pth")
Beta Was this translation helpful? Give feedback.
All reactions