Update OpenVINO INT8 export (#7515)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
parent
4dc8c406f9
commit
30185e0d4c
16 changed files with 42 additions and 30 deletions
|
|
@ -182,16 +182,16 @@ import torch
|
|||
|
||||
# Model
|
||||
model = torch.hub.load('ultralytics/yolov5', 'custom', 'yolov5s.pt')
|
||||
'yolov5s.torchscript ') # TorchScript
|
||||
'yolov5s.onnx') # ONNX Runtime
|
||||
'yolov5s_openvino_model') # OpenVINO
|
||||
'yolov5s.engine') # TensorRT
|
||||
'yolov5s.mlmodel') # CoreML (macOS Only)
|
||||
'yolov5s_saved_model') # TensorFlow SavedModel
|
||||
'yolov5s.pb') # TensorFlow GraphDef
|
||||
'yolov5s.tflite') # TensorFlow Lite
|
||||
'yolov5s_edgetpu.tflite') # TensorFlow Edge TPU
|
||||
'yolov5s_paddle_model') # PaddlePaddle
|
||||
model = torch.hub.load('ultralytics/yolov5', 'custom', 'yolov5s.torchscript ') # TorchScript
|
||||
model = torch.hub.load('ultralytics/yolov5', 'custom', 'yolov5s.onnx') # ONNX Runtime
|
||||
model = torch.hub.load('ultralytics/yolov5', 'custom', 'yolov5s_openvino_model') # OpenVINO
|
||||
model = torch.hub.load('ultralytics/yolov5', 'custom', 'yolov5s.engine') # TensorRT
|
||||
model = torch.hub.load('ultralytics/yolov5', 'custom', 'yolov5s.mlmodel') # CoreML (macOS Only)
|
||||
model = torch.hub.load('ultralytics/yolov5', 'custom', 'yolov5s_saved_model') # TensorFlow SavedModel
|
||||
model = torch.hub.load('ultralytics/yolov5', 'custom', 'yolov5s.pb') # TensorFlow GraphDef
|
||||
model = torch.hub.load('ultralytics/yolov5', 'custom', 'yolov5s.tflite') # TensorFlow Lite
|
||||
model = torch.hub.load('ultralytics/yolov5', 'custom', 'yolov5s_edgetpu.tflite') # TensorFlow Edge TPU
|
||||
model = torch.hub.load('ultralytics/yolov5', 'custom', 'yolov5s_paddle_model') # PaddlePaddle
|
||||
|
||||
# Images
|
||||
img = 'https://ultralytics.com/images/zidane.jpg' # or file, Path, PIL, OpenCV, numpy, list
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue