fix writing flow map bug in test

open-mmlab · Jan 26, 2022 · 42a6751 · 42a6751
1 parent 573d79c
commit 42a6751
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmflow/apis/test.py b/mmflow/apis/test.py
@@ -54,7 +54,7 @@ def single_gpu_test(
     if out_dir is not None:
         mmcv.mkdir_or_exist(out_dir)
         for i, r in enumerate(results):
-            write_flow(osp.join(out_dir, f'flow_{i:03d}.flo'), r)
+            write_flow(r, osp.join(out_dir, f'flow_{i:03d}.flo'))
 
     if show_dir is not None:
         mmcv.mkdir_or_exist(show_dir)