Implement common post_processing (#11408)

* implement common post_processing

* fix formatting

* rename yolonas to post_process_yolonas
This commit is contained in:
Marc Altmann
2024-05-17 10:50:45 -06:00
committed by GitHub
parent 97f5ba0145
commit a70dd02788
3 changed files with 66 additions and 2 deletions
+1
View File
@@ -30,6 +30,7 @@ class InputTensorEnum(str, Enum):
class ModelTypeEnum(str, Enum):
ssd = "ssd"
yolox = "yolox"
yolonas = "yolonas"
class ModelConfig(BaseModel):