Fix missing tensor_input (#14790)

This commit is contained in:
Nicolas Mowen 2024-11-04 12:04:33 -07:00 committed by GitHub
parent a13b9815f6
commit 553676aade
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -457,6 +457,7 @@ model:
width: 320 # <--- should match whatever was set in notebook
height: 320 # <--- should match whatever was set in notebook
input_pixel_format: bgr
input_tensor: nchw
path: /config/yolo_nas_s.onnx
labelmap_path: /labelmap/coco-80.txt
```