Adds support for YOLO-NAS in OpenVino (#11645)

* update onnxruntime

* support for yolo-nas in openvino

* cleanup notebook

* update docs

* improve docs

* handle AUTO issue and update docs
This commit is contained in:
Blake Blackshear
2024-06-07 06:52:08 -05:00
committed by GitHub
parent 4e569ad644
commit 4313fd97aa
6 changed files with 279 additions and 46 deletions

View File

@@ -1,5 +1,6 @@
import logging
from abc import ABC, abstractmethod
from typing import List
import numpy as np
@@ -10,6 +11,7 @@ logger = logging.getLogger(__name__)
class DetectionApi(ABC):
type_key: str
supported_models: List[ModelTypeEnum]
@abstractmethod
def __init__(self, detector_config):