save metrics and frames to persistent storage

evaluate.py

@@ -74,8 +74,8 @@ def compute_fvd(item, gt_imgs, results):
     os.makedirs('temp/gt', exist_ok=True)
     os.makedirs('temp/result', exist_ok=True)
 
-    #save_mp4(gt_imgs, "temp/gt/gt.mp4")
-    #save_mp4(results, "temp/result/result.mp4")
+    save_mp4(gt_imgs, "temp/gt/gt.mp4")
+    save_mp4(results, "temp/result/result.mp4")
 
     evaluator = fvd.cdfvd('i3d', ckpt_path=None, device='cuda', n_real=1, n_fake=1)
     evaluator.compute_real_stats(evaluator.load_videos('temp/gt', data_type='video_folder'))
@@ -138,16 +138,16 @@ def get_score(item, image_paths, video_path, metrics, train_steps=100, inference
         #results = run(images, video_path, train_steps=100, inference_steps=10, fps=12, bg_remove=False, finetune=True)
         results, results_base = run_eval(images, video_path, train_steps=100, inference_steps=10, fps=12, modelId="fine_tuned_pcdms", img_width=1920, img_height=1080, bg_remove=False, resize_inputs=False)
 
-        os.makedirs('out/'+item, exist_ok=True)
+        os.makedirs('/data/out/'+item, exist_ok=True)
 
         for i, frame in enumerate(gt_frames):
-            frame.save("out/"+item+"/frame_"+str(i)+".png")
+            frame.save("/data/out/"+item+"/frame_"+str(i)+".png")
 
         for i, result in enumerate(results):
-            result.save("out/"+item+"/result_"+str(i)+".png")
+            result.save("/data/out/"+item+"/result_"+str(i)+".png")
             
         for i, result in enumerate(results_base):
-            result.save("out/"+item+"/base_"+str(i)+".png")
+            result.save("/data/out/"+item+"/base_"+str(i)+".png")
     
     ssim = []
     psnr = []
@@ -204,7 +204,7 @@ def get_score(item, image_paths, video_path, metrics, train_steps=100, inference
     
     #print(metrics)
 
-    with open('metrics.json', "w", encoding="utf-8") as json_file:
+    with open('/data/metrics.json', "w", encoding="utf-8") as json_file:
         json.dump(metrics, json_file, ensure_ascii=False, indent=4)
     
     
@@ -225,7 +225,7 @@ def run_evaluate():
     print("run_evaluate")
     snapshot_download(repo_id="acmyu/KeyframesAI-eval", local_dir="test", repo_type="dataset")
 
-    with open('metrics.json', 'r') as file:
+    with open('/data/metrics.json', 'r') as file:
         metrics = json.load(file)
 
     items = os.listdir('test')