Question Use custom trained model with PeekingDuck
Hello to everybody
I wanted to ask how can I use a custom trained model through PeekingDuck. I've managed to train a custom model based on the yolo v4 tiny with Roboflow and got the new weights. How should I replace the default ones of the tiny yolo v4 with my new ones?
Thanks in advance
Hi there, thanks for the question! Noted that you are using Roboflow to train the custom model.
Let me have a short discussion with my team and get back to you! Cheers!
Yep this is it. I was actually trying to add a couple of classes to my yolo v4 tiny model so i've trained it with a few custom images in order to have better detection regarding these classes. But i'm not sure how to proceed with importing the trained model and use it through peeking duck instead of the default one.
Hi, in that case, you may have to create a custom node to inference your custom-trained model.
Here's a tutorial to get you started, however, if you have troubles, feel free to reach to me via messages, and we can connect further.