Spaces:
Paused
Paused
fix eval script
Browse files- app.py +1 -1
- evaluate.py +1 -1
- main.py +1 -1
- requirements.txt +3 -0
app.py
CHANGED
|
@@ -60,7 +60,7 @@ with gr.Blocks() as demo:
|
|
| 60 |
)
|
| 61 |
|
| 62 |
eval_btn.click(
|
| 63 |
-
|
| 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 |
-
|
| 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
|