search

princeton-vl / SEA-RAFT

[ECCV2024 - Oral, Best Paper Award Candidate] SEA-RAFT: Simple, Efficient, Accurate RAFT for Optical Flow
BSD 3-Clause "New" or "Revised" License
327 stars 11 forks source link

How to export ONNX? #6

Closed newforrestgump001 closed 4 months ago

newforrestgump001 commented 4 months ago

Where can I the onnx model, have you exported it yet? Thank you a lot!

MemorySlices commented 4 months ago

We did not export SEA-RAFT models. However, SEA-RAFT models do not rely on custom layers so the standard onnx exporter (e.g. torch.onnx) should be enough.