insight-platform / Savant

Python Computer Vision & Video Analytics Framework With Batteries Included
https://savant-ai.io
Apache License 2.0
566 stars 45 forks source link

Implement Custom YOLO v5 Module #218

Closed maoztamir closed 1 year ago

maoztamir commented 1 year ago

I am trying to create an environment that uses a simple custom yolov5 object detection model, the model is working on a stand-alone deep stream environment. I have used the traffic meter sample as the baseline, however, I still have some issues implementing this example. The main issue is that the .wts failed to load when loading the module, do you have any advice?

dorgun commented 1 year ago

@maoztamir, hi!

Are you facing any issues with loading the weights when running the "traffic_meter" example? Please provide an error log. If possible, attach the complete module log.

bwsw commented 1 year ago

Closing due to no activity.