Parcourir la source

Update of write python script

Jérôme BUISINE il y a 4 ans
Parent
commit
1ecca35a5f
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  1. 1 1
      write_result_keras.py

+ 1 - 1
write_result_keras.py

@@ -51,7 +51,7 @@ def write_result(_scene_name, _data_file, _model_path, _n, _reconstructed_path,
     if not os.path.exists(reference_image_path):
         # call sub process to create 'reference' img
         print("Creation of reference image : ", reference_image_path)
-        subprocess.run(["python", "reconstruct_scene_mean.py", "--scene", _scene_name, "--n", 1000, "--image_name", reference_image_path.split('/')[-1]])
+        subprocess.run(["python", "reconstruct_scene_mean.py", "--scene", _scene_name, "--n", str(1000), "--image_name", reference_image_path.split('/')[-1]])
 
 
     # load the trained model