forked from cvg/glue-factory
-
Notifications
You must be signed in to change notification settings - Fork 0
/
copyEval.py
54 lines (41 loc) · 2.03 KB
/
copyEval.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
import os
import shutil
source_dir = "/homes/tp4618/Documents/bitbucket/SuperGlueThesis/external/glue-factory/evaluations"
target_dir = "/homes/tp4618/Documents/bitbucket/SuperGlueThesis/external/glue-factory/pretrainedEvaluations"
filename = "finnEval_poselib.txt"
def copy_eval_results():
for folder_name in os.listdir(source_dir):
folder_path = os.path.join(source_dir, folder_name)
if os.path.isdir(folder_path):
file_path = os.path.join(folder_path, filename)
if os.path.exists(file_path):
target_folder_path = os.path.join(target_dir, folder_name)
target_file_path = os.path.join(target_folder_path, filename)
if os.path.exists(target_file_path):
raise FileExistsError(f"File {filename} already exists in {target_folder_path}. Aborting to prevent overwriting.")
os.makedirs(target_folder_path, exist_ok=True)
shutil.copy2(file_path, target_folder_path)
print(f"Copied {filename} from {folder_name} to {target_folder_path}")
if __name__ == "__main__":
# copy_eval_results()
hasFinn = []
noFinn = []
for folder_name in os.listdir(target_dir):
folder_path = os.path.join(target_dir, folder_name)
if os.path.isdir(folder_path):
file_path = os.path.join(folder_path, filename)
if os.path.exists(file_path):
hasFinn.append(folder_name)
else:
noFinn.append(folder_name)
print("Has Finn:", hasFinn)
print("No Finn:", noFinn)
print("\n\n")
authorConfigs = "/homes/tp4618/Documents/bitbucket/SuperGlueThesis/external/authorGlueFactory/gluefactory/configs"
authorFolders = os.listdir(authorConfigs)
print("Author folders:", authorFolders)
for folder in noFinn:
if folder.split('.')[0] in authorFolders:
print(f"python -m gluefactory.eval.finnEval --conf {folder} --overwrite")
else:
print(f"Folder {folder} not found in author configs.")