-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathscript.py
More file actions
28 lines (19 loc) · 856 Bytes
/
script.py
File metadata and controls
28 lines (19 loc) · 856 Bytes
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
import os
searchdir = "D:/VSC/INFOCOM/FrameDrop-RL/models/ndarray"
for entry in os.listdir(searchdir):
if os.path.isfile(os.path.join(searchdir, entry)):
filename = os.path.basename(entry)
filename_without_type = filename[:-4]
conf_list = filename.split('_')
date = conf_list[0]
dataset = conf_list[2]
if date > "240401-140000":
if dataset == "JK":
test_dataset = ["JK-1"]
elif dataset == "SD":
test_dataset = ["SD-1"]
elif dataset == "JN":
test_dataset = ["JN"]
for data in test_dataset:
command = f"python run.py -video ../DATASET/test/{data}.mp4 -model models/ndarray/{filename} -mask f -f1 t"
os.system(command)