mirrors
/
mmcv
mirror of https://github.com/open-mmlab/mmcv.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Update optflow.py (#335)

pull/341/head
lizz 2020-06-12 23:36:09 +08:00 committed by GitHub
parent 67a26da917
commit 6988de82b4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mmcv/visualization/optflow.py
View File

@ -55,8 +55,6 @@ def flow2rgb(flow, color_wheel=None, unknown_thr=1e6):
dx /= max_rad dx /= max_rad
dy /= max_rad dy /= max_rad
[h, w] = dx.shape
rad = np.sqrt(dx**2 + dy**2) rad = np.sqrt(dx**2 + dy**2)
angle = np.arctan2(-dy, -dx) / np.pi angle = np.arctan2(-dy, -dx) / np.pi