frigate
frigate copied to clipboard
blakeblackshear
YOLOX support for OpenVINO Detector
This PR is to add YOLOX support for the newly added OpenVINO detector. I added a new config parameter called model_type, which is ssd by default to match the default ssdlite_mobilenet_v2 model that comes with Frigate. Users can change this to yolox to use a yolox model instead. After the changes, I confirmed that the existing SSD OpenVINO support still works.
I have also added a text into the documents explaining how to use a yolox model, and what the limitations are.
Deploy Preview for frigate-docs canceled.
| Name | Link |
|---|---|
| Latest commit | cf70808c774b75762661267f46debda8c56c416c |
| Latest deploy log | https://app.netlify.com/sites/frigate-docs/deploys/63db2c1af4ea67000823a26b |
![netlify[bot] avatar](https://avatars.githubusercontent.com/in/13473?v=4 "Published by a pub.dev verified publisher"){kind=small}
I'll take a closer look and try to run this later today. Do you have any insight on the differences between this and the yolov4 models?
I'll take a closer look and try to run this later today. Do you have any insight on the differences between this and the yolov4 models?
I believe YOLOx is anchorless, while most other YOLO variants have anchors. Post processing the tensors would require a little more math due to anchors. What I have here for YOLOx should be 90% of the way there to support other YOLO variants as well. So maybe in the future, that could be explored.
@NateMeyer @NickM-27 are there any opens on this before it can be merged?
@NateMeyer @NickM-27 are there any opens on this before it can be merged?
First need to get checks passing
Alright 3rd time's the charm. If you can kick off the checks, it would be great.
@NateMeyer @NickM-27 let me know if there are any other opens preventing this merge. Thanks.