acmyu commited on
Commit
af05866
·
1 Parent(s): 341438f

fix eval script

Browse files
Files changed (4) hide show
  1. app.py +1 -1
  2. evaluate.py +1 -1
  3. main.py +1 -1
  4. requirements.txt +3 -0
app.py CHANGED
@@ -60,7 +60,7 @@ with gr.Blocks() as demo:
60
  )
61
 
62
  eval_btn.click(
63
- run_interpolate_frames, inputs=[], outputs=[eval_scores]
64
  )
65
 
66
 
 
60
  )
61
 
62
  eval_btn.click(
63
+ run_evaluate, inputs=[], outputs=[eval_scores]
64
  )
65
 
66
 
evaluate.py CHANGED
@@ -153,7 +153,7 @@ def get_files(directory_path):
153
 
154
 
155
  def run_evaluate():
156
- snapshot_download(repo_id="acmyu/KeyframesAI-eval", local_dir="test")
157
 
158
  items = os.listdir('test')
159
  items = ['test/woody'] #['sidewalk', 'aaa', 'azri', 'dead', 'frankgirl', 'kobold', 'ramona', 'renee', 'walk', 'woody']
 
153
 
154
 
155
  def run_evaluate():
156
+ snapshot_download(repo_id="acmyu/KeyframesAI-eval", local_dir="test", repo_type="dataset")
157
 
158
  items = os.listdir('test')
159
  items = ['test/woody'] #['sidewalk', 'aaa', 'azri', 'dead', 'frankgirl', 'kobold', 'ramona', 'renee', 'walk', 'woody']
main.py CHANGED
@@ -1256,7 +1256,7 @@ def run_eval(images, video_path, train_steps=100, inference_steps=10, fps=12, mo
1256
 
1257
  dwpose, rembg_session, pcdms_model, noise_scheduler, image_encoder_p, image_encoder_g, vae, unet = load_models()
1258
 
1259
- run_train(images, train_steps, modelId, bg_remove, resize_inputs)
1260
 
1261
  images = [img[0] for img in images]
1262
  in_img = images[0]
 
1256
 
1257
  dwpose, rembg_session, pcdms_model, noise_scheduler, image_encoder_p, image_encoder_g, vae, unet = load_models()
1258
 
1259
+ run_train_impl(images, train_steps, modelId, bg_remove, resize_inputs)
1260
 
1261
  images = [img[0] for img in images]
1262
  in_img = images[0]
requirements.txt CHANGED
@@ -21,3 +21,6 @@ gradio
21
  rembg[cpu]
22
  spaces
23
  matplotlib
 
 
 
 
21
  rembg[cpu]
22
  spaces
23
  matplotlib
24
+
25
+ lpips
26
+ pytorch-fid