PyTorch繪製光流flow

這裏給出兩種方法的code，第一種是用hsv來轉化，第二種是直接繪製。python 比較tricky的地方多是思路上先生成hsv圖，由於hsv圖中第一個h維度，恰好用各類不一樣的顏色來表達角度，這個和flow中的角度是一致的。而flow的絕對值長度恰好又能被s和v維度表達（其實h和v只有一個自由度）web def arctan(x, y): y_x = abs(y/x) angle

>>阅读原文<<