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

>>阅读原文<<