Video-Codec-Evaluation/evaluate.py at master · SINRG-Lab/Video-Codec-Evaluation · GitHub

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
import os, sys
sys.path.append(os.getcwd() + '/')
sys.path.append(os.getcwd() + '/codecs')
sys.path.append(os.getcwd() + '/scripts')

import glob
import configparser
from runvp9 import runvp9
from runh26x import runh26x
from runav1 import runav1
from quality import quality
from average import average


class Evaluate():
    def __init__(self):
        self.parse_config()
        self.ret = []

    def parse_config(self):
        config = configparser.ConfigParser()
        config.read('config.ini')

        #codec params
        self.preset_26x = config['CODEC']['preset_26x']
        self.preset_vpx = config['CODEC']['preset_vpx']
        self.codec      = config['CODEC']['codec']
        self.gop        = config['CODEC']['gop']
        self.crf        = config['CODEC']['crf']
        self.cbr        = config['CODEC']['cbr']
        self.mode       = config['CODEC']['mode']

        #Sequence
        self.frames     = config['SEQUENCE']['frames']
        self.fps        = config['SEQUENCE']['fps']
        self.width      = config['SEQUENCE']['width']
        self.height     = config['SEQUENCE']['height']
        self.num_seq    = config['SEQUENCE']['num_seq']

        #paths
        self.bin_out       = config['PATHS']['encode_out']
        self.results       = config['PATHS']['results']
        self.test_sequence = config['PATHS']['test_sequence']
        self.log_out       = config['PATHS']['log_out']

    def readfilename(self):
        files = glob.glob(self.test_sequence + self.height + '/*.yuv')
        for f in files:
            name = f.split('/')[-1]
            name = name.split(".")[0]
            self.ret.append(name)

    def cleanup(self):
       for f in glob.glob(self.bin_out + self.height + '/*.mp4'):
           os.remove(f)

       for f in glob.glob(self.results + self.height + '/*.json'):
           os.remove(f)

       for f in glob.glob(self.log_out + self.height + '/*.log'):
            os.remove(f)

    def encode_decode(self):
        if self.codec == 'libvpx-vp9':
            runvp9(self)
        elif self.codec == 'libx265' or self.codec == 'libx264':
            runh26x(self)
        elif self.codec == 'libaom-av1':
            runav1(self);
        else:
            pass;

    def measure_quality(self):
        quality(self)

    def average_all_sequence(self):
        average(self)


if __name__ == '__main__':
    eval = Evaluate()
    eval.cleanup()
    eval.encode_decode()
    eval.measure_quality()
    eval.average_all_sequence()