RAFT/__init__.py

# from .demo import RAFT_infer
from .raft import RAFT