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)