Support skipping preprocess when exporting ONNX #193
Merged
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Using the following snippet will export a dynamic batch/shape ONNX model without pre-processing (aka
letterbox
in YOLOv5 andNestedTensor
in yolort).# 'yolov5s.pt' is downloaded from https://github.com/ultralytics/yolov5/releases/download/v5.0/yolov5s.pt python tools/export_model.py --checkpoint_path yolov5s.pt --skip_preprocess
https://github.com/zhiqwang/yolov5-rt-stack/blob/6cb22845a41bf4dd846a9bcad06c09d3fec8cd49/yolort/models/transform.py#L217-L251
Meanwhile we require the shape of the inputs to be
Nx3xHxW
, and as such this will close #39, close #45 and close #159.