Merge pull request #64 from bkvie/patch-5

Update run_inference.py

Author: ClementPinard

datasets/mpisintel.py

@@ -29,7 +29,7 @@ def make_dataset(dataset_dir, split, dataset_type='clean'):
         img1 = os.path.join(img_dir, scene_dir, '{}_{:04d}.png'.format(prefix, frame_nb))
         img2 = os.path.join(img_dir, scene_dir, '{}_{:04d}.png'.format(prefix, frame_nb + 1))
         flow_map = os.path.join(flow_dir,flow_map)
-        if not (os.path.isfile(os.path.join(dataset_dir,img1)) or os.path.isfile(os.path.join(dataset_dir,img2))):
+        if not (os.path.isfile(os.path.join(dataset_dir,img1)) and os.path.isfile(os.path.join(dataset_dir,img2))):
             continue
         images.append([[img1,img2],flow_map])
 

run_inference.py

@@ -105,7 +105,7 @@ def main():
         for suffix, flow_output in zip(['flow', 'inv_flow'], output):
             filename = save_path/'{}{}'.format(img1_file.namebase[:-1], suffix)
             if args.output_value in['vis', 'both']:
-                rgb_flow = flow2rgb(args.div_flow * flow_output, max_value=args.max_flow)
+                rgb_flow = flow2rgb(args.div_flow * flow_output[0], max_value=args.max_flow)
                 to_save = (rgb_flow * 255).astype(np.uint8).transpose(1,2,0)
                 imwrite(filename + '.png', to_save)
             if args.output_value in ['raw', 'both']: