NeuFlowV2-ONNX / README.md
bukuroo's picture
Update README.md
6b92f2d verified
metadata
license: mit
pipeline_tag: image-to-image
tags:
  - onnx
  - opticalflow
  - ezonnx

For inference, please clone repo ezonnx.

from ezonnx import NeuFlowV2

model = NeuFlowV2("mixed")
output = model("image_0.jpg","image_1.jpg") # infer with a pair of str or cv2 image
print(output.flow)                          # flow map (H,W,2)
print(output.visualized_img)                # flow image (H,W,3)