From 7f7d099bb232c700386b32c5b07892a526a8ac44 Mon Sep 17 00:00:00 2001 From: ruiliu-ai Date: Thu, 23 Sep 2021 15:01:33 +0800 Subject: [PATCH] add evaluation --- evaluate.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/evaluate.py b/evaluate.py index ab2c3cb..5ff0b3b 100644 --- a/evaluate.py +++ b/evaluate.py @@ -173,7 +173,7 @@ def get_i3d_activations(batched_video, target_endpoint='Logits', flatten=True, g def get_frame_mask_list(args): #data_root = "./data/YouTubeVOS/" data_root = "./data/DATASET_DAVIS" - mask_dir = "./random_mask_stationary_w432_h240" + mask_dir = "./data/random_mask_stationary_w432_h240" mask_folder = sorted(os.listdir(mask_dir)) mask_list = [os.path.join(mask_dir, name) for name in mask_folder]