-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathrunner.py
More file actions
130 lines (95 loc) · 7.38 KB
/
Copy pathrunner.py
File metadata and controls
130 lines (95 loc) · 7.38 KB
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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
# import os
# os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
import evaluator
model_dir = "/mnt/WindowsDev/Timor/Study/PreTainedModels/OpenDVCW/"
dataset_dir = "/mnt/WindowsDev/Timor/Study/DataSets/"
label = "Proposed-haar200k"
output_path = "/mnt/WindowsDev/Timor/Study/TEZA/eval_h266/"
num_of_p_frames = 7
# test_dir = "/home/ubu-admin/Developer/OpenDVCW/workdir"
test_dir = "/workspaces/OpenDVCW/workdir"
h264_workdir= test_dir + "/h264/test"
h265_workdir= test_dir + "/h265/test"
h266_workdir= test_dir + "/h266/test"
# tave_path="/home/ubu-admin/Developer/OpenDVCW/cpp_encoder/build/tave"
tave_path="/workspaces/OpenDVCW/cpp_encoder/build/tave"
workdir = test_dir + "/eval"
# model_list = [
# model_dir + "HAAR/model_save_checkpoints_wavelets_haar_Lmbd_256_nfilt_256_epcs_15_stps_10000_I_QP_42_240x240_CosineDecay_20220705-061459/",
# model_dir + "HAAR/model_save_checkpoints_wavelets_haar_Lmbd_1024_nfilt_256_epcs_15_stps_10000_I_QP_37_240x240_CosineDecay_20220704-225023/",
# model_dir + "HAAR/model_save_checkpoints_wavelets_haar_Lmbd_4096_nfilt_256_epcs_15_stps_10000_I_QP_32_240x240_CosineDecay_20220704-152812/",
# model_dir + "HAAR/model_save_checkpoints_wavelets_haar_Lmbd_16384_nfilt_256_epcs_15_stps_10000_I_QP_27_240x240_CosineDecay_20220704-044114/",
# model_dir + "HAAR/model_save_checkpoints_wavelets_haar_Lmbd_65536_nfilt_256_epcs_15_stps_10000_I_QP_22_240x240_CosineDecay_20220703-214625/"
# ]
# model_list = [
# model_dir + "DB2/model_save_checkpoints_wavelets_db2_Lmbd_256_nfilt_256_epcs_15_stps_60000_I_QP_42_240x240_CosineDecay_20220729-022643/",
# model_dir + "DB2/model_save_checkpoints_wavelets_db2_Lmbd_1024_nfilt_256_epcs_15_stps_60000_I_QP_37_240x240_CosineDecay_20220727-000239/",
# model_dir + "DB2/model_save_checkpoints_wavelets_db2_Lmbd_4096_nfilt_256_epcs_15_stps_60000_I_QP_32_240x240_CosineDecay_20220725-042633/",
# model_dir + "DB2/model_save_checkpoints_wavelets_db2_Lmbd_16384_nfilt_256_epcs_15_stps_60000_I_QP_27_240x240_CosineDecay_20220721-093040/",
# model_dir + "DB2/model_save_checkpoints_wavelets_db2_Lmbd_65536_nfilt_256_epcs_15_stps_60000_I_QP_22_240x240_CosineDecay_20220719-155911/"
# ]
# model_list = [
# model_dir + "SYM3/model_save_checkpoints_wavelets_sym3_Lmbd_256_nfilt_256_epcs_15_stps_60000_I_QP_42_240x240_CosineDecay_20221001-130850/",
# model_dir + "SYM3/model_save_checkpoints_wavelets_sym3_Lmbd_1024_nfilt_256_epcs_15_stps_60000_I_QP_37_240x240_CosineDecay_20220930-212030/",
# model_dir + "SYM3/model_save_checkpoints_wavelets_sym3_Lmbd_4096_nfilt_256_epcs_15_stps_60000_I_QP_32_240x240_CosineDecay_20220930-053044/",
# model_dir + "SYM3/model_save_checkpoints_wavelets_sym3_Lmbd_16384_nfilt_256_epcs_15_stps_60000_I_QP_27_240x240_CosineDecay_20220929-131432/",
# model_dir + "SYM3/model_save_checkpoints_wavelets_sym3_Lmbd_65536_nfilt_256_epcs_15_stps_60000_I_QP_22_240x240_CosineDecay_20220928-182917/"
# ]
# model_list = [
# model_dir + "COIF1/model_save_checkpoints_wavelets_coif1_Lmbd_256_nfilt_256_epcs_15_stps_60000_I_QP_42_240x240_CosineDecay_20221005-112002/",
# model_dir + "COIF1/model_save_checkpoints_wavelets_coif1_Lmbd_1024_nfilt_256_epcs_15_stps_60000_I_QP_37_240x240_CosineDecay_20221004-192207/",
# model_dir + "COIF1/model_save_checkpoints_wavelets_coif1_Lmbd_4096_nfilt_256_epcs_15_stps_60000_I_QP_32_240x240_CosineDecay_20221004-033121/",
# model_dir + "COIF1/model_save_checkpoints_wavelets_coif1_Lmbd_16384_nfilt_256_epcs_15_stps_60000_I_QP_27_240x240_CosineDecay_20221003-115147/",
# model_dir + "COIF1/model_save_checkpoints_wavelets_coif1_Lmbd_65536_nfilt_256_epcs_15_stps_60000_I_QP_22_240x240_CosineDecay_20221002-194858/"
# ]
# model_list = [
# model_dir + "HAAR60k/model_save_checkpoints_wavelets_haar_Lmbd_256_nfilt_256_epcs_15_stps_60000_I_QP_42_240x240_CosineDecay_20221203-191019/",
# model_dir + "HAAR60k/model_save_checkpoints_wavelets_haar_Lmbd_1024_nfilt_256_epcs_15_stps_60000_I_QP_35_240x240_CosineDecay_20221204-145313/",
# model_dir + "HAAR60k/model_save_checkpoints_wavelets_haar_Lmbd_4096_nfilt_256_epcs_15_stps_60000_I_QP_32_240x240_CosineDecay_20221202-182053/",
# model_dir + "HAAR60k/model_save_checkpoints_wavelets_haar_Lmbd_16384_nfilt_256_epcs_15_stps_60000_I_QP_27_240x240_CosineDecay_20221202-054609/",
# model_dir + "HAAR60k/model_save_checkpoints_wavelets_haar_Lmbd_65536_nfilt_256_epcs_15_stps_60000_I_QP_22_240x240_CosineDecay_20221201-171832/"
# ]
model_list = [
model_dir + "HAAR200k/model_save_checkpoints_wavelets_haar_Lmbd_256_nfilt_256_epcs_5_stps_200000_I_QP_42_240x240_CosineDecay_20221207-161406/",
model_dir + "HAAR200k/model_save_checkpoints_wavelets_haar_Lmbd_1024_nfilt_256_epcs_5_stps_200000_I_QP_35_240x240_CosineDecay_20221207-022051/",
model_dir + "HAAR200k/model_save_checkpoints_wavelets_haar_Lmbd_4096_nfilt_256_epcs_5_stps_200000_I_QP_32_240x240_CosineDecay_20221206-123302/",
model_dir + "HAAR200k/model_save_checkpoints_wavelets_haar_Lmbd_16384_nfilt_256_epcs_5_stps_200000_I_QP_27_240x240_CosineDecay_20221205-224634/",
model_dir + "HAAR200k/model_save_checkpoints_wavelets_haar_Lmbd_65536_nfilt_256_epcs_5_stps_200000_I_QP_22_240x240_CosineDecay_20221205-085653/"
]
def run(input_seq_path, h264_bit_rate, h265_bit_rate, h266_bit_rate):
eva = evaluator.Evaluator(bs=1, height=240, width=240, channels=3,
input_seq_path=input_seq_path, workdir=workdir,
model_list=model_list, num_of_p_frames=num_of_p_frames, prefix="im",
suffix=".png", bin_suffix=".bin", decom_prefix="decom",
tave_path=tave_path,
h264_workdir=h264_workdir, h264_bitrate_list=h264_bit_rate,
h265_workdir=h265_workdir, h265_bitrate_list=h265_bit_rate,
h266_workdir=h266_workdir, h266_bitrate_list=h266_bit_rate,
proposed_label=label)
eva.eval(tave_run=True)
eva.plot_graph(fig_name=output_path + input_seq_path.split("/")[-2] + label + ".svg")
eva.save_csv(output_path + input_seq_path.split("/")[-2] + label + ".csv")
# HoneyBee_1920x1080_120fps_420_8bit_YUV_RAW for db2
input_seq_path = dataset_dir + "HoneyBee_1920x1080_120fps_420_8bit_YUV_RAW/"
h264_bit_rate = [1e6, 1.2e6, 5e6, 15e6, 30e6]
h265_bit_rate = [1e6, 3e6, 7e6, 10e6, 20e6]
h266_bit_rate = [1e6, 1e6, 2e6, 5e6, 10e6]
run(input_seq_path, h264_bit_rate, h265_bit_rate, h266_bit_rate)
# Beauty_1920x1080_120fps_420_8bit_YUV_RAW for db2
input_seq_path = dataset_dir + "Beauty_1920x1080_120fps_420_8bit_YUV_RAW/"
h264_bit_rate = [0.5e6, 1.5e6, 3e6, 5e6, 6e6]
h265_bit_rate = [0.5e6, 1.5e6, 3e6, 5e6, 6e6]
h266_bit_rate = [0.2e6, 0.4e6, 1.5e6, 4e6, 8e6]
run(input_seq_path, h264_bit_rate, h265_bit_rate, h266_bit_rate)
# Bosphorus_1920x1080_120fps_420_8bit_YUV_raw for db2
input_seq_path = dataset_dir + "Bosphorus_1920x1080_120fps_420_8bit_YUV_raw/"
h264_bit_rate = [0.8e6, 3e6, 6e6, 8e6, 11e6]
h265_bit_rate = [0.5e6, 1.5e6, 4e6, 8e6, 11e6]
h266_bit_rate = [0.1e6, 1.4e6, 2e6, 4e6, 5e6]
run(input_seq_path, h264_bit_rate, h265_bit_rate, h266_bit_rate)
# ShakeNDry_1920x1080_120fps_420_8bit_YUV_RAW for db2
input_seq_path = dataset_dir + "ShakeNDry_1920x1080_120fps_420_8bit_YUV_RAW/"
h264_bit_rate = [0.5e6, 2e6, 5e6, 7e6, 9e6]
h265_bit_rate = [0.5e6, 2e6, 5e6, 7e6, 9e6]
h266_bit_rate = [0.1e6, 0.4e6, 2e6, 5e6, 12e6]
run(input_seq_path, h264_bit_rate, h265_bit_rate, h266_bit_rate)