mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-05-05 13:07:44 +03:00
gs
This commit is contained in:
parent
91be6073d3
commit
a50377f22a
@ -682,6 +682,29 @@ model:
|
|||||||
|
|
||||||
Note that the labelmap uses a subset of the complete COCO label set that has only 80 objects.
|
Note that the labelmap uses a subset of the complete COCO label set that has only 80 objects.
|
||||||
|
|
||||||
|
#### YOLOx
|
||||||
|
|
||||||
|
[YOLOx](https://github.com/Megvii-BaseDetection/YOLOX) models are supported, but not included by default.
|
||||||
|
|
||||||
|
After placing the downloaded onnx model in your config folder, you can use the following configuration:
|
||||||
|
|
||||||
|
```yaml
|
||||||
|
detectors:
|
||||||
|
onnx:
|
||||||
|
type: onnx
|
||||||
|
|
||||||
|
model:
|
||||||
|
model_type: yolox
|
||||||
|
width: 416 # <--- should match the imgsize set during model export
|
||||||
|
height: 416 # <--- should match the imgsize set during model export
|
||||||
|
input_tensor: nchw_denorm
|
||||||
|
input_dtype: float
|
||||||
|
path: /config/model_cache/yolox_tiny.onnx
|
||||||
|
labelmap_path: /labelmap/coco-80.txt
|
||||||
|
```
|
||||||
|
|
||||||
|
Note that the labelmap uses a subset of the complete COCO label set that has only 80 objects.
|
||||||
|
|
||||||
#### RF-DETR
|
#### RF-DETR
|
||||||
|
|
||||||
[RF-DETR](https://github.com/roboflow/rf-detr) is a DETR based model. The ONNX exported models are supported, but not included by default. See [the models section](#downloading-rf-detr-model) for more information on downloading the RF-DETR model for use in Frigate.
|
[RF-DETR](https://github.com/roboflow/rf-detr) is a DETR based model. The ONNX exported models are supported, but not included by default. See [the models section](#downloading-rf-detr-model) for more information on downloading the RF-DETR model for use in Frigate.
|
||||||
@ -962,6 +985,10 @@ The input image size in this notebook is set to 320x320. This results in lower C
|
|||||||
|
|
||||||
### Downloading YOLO Models
|
### Downloading YOLO Models
|
||||||
|
|
||||||
|
#### YOLOx
|
||||||
|
|
||||||
|
YOLOx models can be downloaded [from the YOLOx repo](https://github.com/Megvii-BaseDetection/YOLOX/tree/main/demo/ONNXRuntime).
|
||||||
|
|
||||||
#### YOLOv3, YOLOv4, and YOLOv7
|
#### YOLOv3, YOLOv4, and YOLOv7
|
||||||
|
|
||||||
To export as ONNX:
|
To export as ONNX:
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user