Optical Flow
-
RAFT: Recurrent All-Pairs Field Transforms for Optical Flow 돌려보기영상처리/3D computer vision 2023. 10. 30. 20:56
RAFT를 이용해 dense optical flow를 계산해보고자 한다. 아래와 같이, torchvision.models.optical_flow로부터 이미 train된 raft 모델을 가져와, image와 optical flow를 overlay 해보 았다. import numpy as np import torch import matplotlib.pyplot as plt import torchvision.transforms.functional as F import torchvision.transforms as T import cv2 from torchvision.models.optical_flow import raft_small from torchvision.models.optical_flow import..