コード例 #1
0
ファイル: executor.py プロジェクト: Netflix/vmaf
    def __init__(self,
                 assets,
                 logger,
                 fifo_mode=True,
                 delete_workdir=True,
                 result_store=None,
                 optional_dict=None,
                 optional_dict2=None,
                 ):
        """
        Use optional_dict for parameters that would impact result (e.g. model,
        patch size), and use optional_dict2 for parameters that would NOT
        impact result (e.g. path to data cache file).
        """

        TypeVersionEnabled.__init__(self)

        self.assets = assets
        self.logger = logger
        self.fifo_mode = fifo_mode
        self.delete_workdir = delete_workdir
        self.results = []
        self.result_store = result_store
        self.optional_dict = optional_dict
        self.optional_dict2 = optional_dict2

        self._assert_args()
        self._assert_assets()
コード例 #2
0
    def __init__(
        self,
        assets,
        logger,
        fifo_mode=True,
        delete_workdir=True,
        result_store=None,
        optional_dict=None,
        optional_dict2=None,
    ):
        """
        Use optional_dict for parameters that would impact result (e.g. model,
        patch size), and use optional_dict2 for parameters that would NOT
        impact result (e.g. path to data cache file).
        """

        TypeVersionEnabled.__init__(self)

        self.assets = assets
        self.logger = logger
        self.fifo_mode = fifo_mode
        self.delete_workdir = delete_workdir
        self.results = []
        self.result_store = result_store
        self.optional_dict = optional_dict
        self.optional_dict2 = optional_dict2

        self._assert_class()
        self._assert_args()
        self._assert_assets()
コード例 #3
0
ファイル: train_test_model.py プロジェクト: Alex-Song/vmaf
    def __init__(self, param_dict, logger=None):
        """
        :param param_dict: contains input parameters
        :param logger:
        :return:
        """
        TypeVersionEnabled.__init__(self)

        self.param_dict = param_dict
        self.logger = logger

        self.model_dict = {}
コード例 #4
0
    def __init__(self, param_dict, logger=None):
        """
        :param param_dict: contains input parameters
        :param logger:
        :return:
        """
        TypeVersionEnabled.__init__(self)

        self.param_dict = param_dict
        self.logger = logger

        self.model_dict = {}
コード例 #5
0
ファイル: perf_metric.py プロジェクト: taosenbai/vmaf
 def __init__(self, groundtruths, predictions):
     """
     Performance metric on quality metrics
     :param groundtruths: either list of real numbers (aggregate scores like
     MOS or DMOS or MLE), or list of list of real numbers (list of raw scores)
     :param predictions: list of real numbers
     :return:
     """
     TypeVersionEnabled.__init__(self)
     self.groundtruths = groundtruths
     self.predictions = predictions
     self._assert_args()
コード例 #6
0
ファイル: perf_metric.py プロジェクト: Netflix/vmaf
 def __init__(self, groundtruths, predictions):
     """
     Performance metric on quality metrics
     :param groundtruths: either list of real numbers (aggregate scores like
     MOS or DMOS or MLE), or list of list of real numbers (list of raw scores)
     :param predictions: list of real numbers
     :return:
     """
     TypeVersionEnabled.__init__(self)
     self.groundtruths = groundtruths
     self.predictions = predictions
     self._assert_args()
コード例 #7
0
ファイル: train_test_model.py プロジェクト: yuhjay/vmaf
    def __init__(self, param_dict, logger=None, optional_dict2=None):
        '''
        Put in optional_dict2 optionals that would not impact result, e.g.
        path to checkpoint file directories, or h5py file
        '''
        TypeVersionEnabled.__init__(self)
        self.param_dict = param_dict
        self.logger = logger
        self.optional_dict2 = optional_dict2

        self.model_dict = {}

        self._assert_args()
コード例 #8
0
ファイル: train_test_model.py プロジェクト: Netflix/vmaf
    def __init__(self, param_dict, logger=None, optional_dict2=None):
        '''
        Put in optional_dict2 optionals that would not impact result, e.g.
        path to checkpoint file directories, or h5py file
        '''
        TypeVersionEnabled.__init__(self)
        self.param_dict = param_dict
        self.logger = logger
        self.optional_dict2 = optional_dict2

        self.model_dict = {}

        self._assert_args()
コード例 #9
0
ファイル: executor.py プロジェクト: wanjinchang/vmaf
    def __init__(self,
                 assets,
                 logger,
                 fifo_mode=True,
                 delete_workdir=True,
                 result_store=None,
                 optional_dict=None,
                 ):

        TypeVersionEnabled.__init__(self)

        self.assets = assets
        self.logger = logger
        self.fifo_mode = fifo_mode
        self.delete_workdir = delete_workdir
        self.results = []
        self.result_store = result_store
        self.optional_dict = optional_dict

        self._assert_assets()
コード例 #10
0
    def executor_id(self):
        executor_id_ = TypeVersionEnabled.get_type_version_string(self)

        if self.optional_dict is not None and len(self.optional_dict) > 0:
            # include optional_dict info in executor_id for result store,
            # as parameters in optional_dict will impact result
            executor_id_ += '_{}'.format('_'.join(
                map(lambda k: '{k}_{v}'.format(k=k, v=self.optional_dict[k]),
                    sorted(self.optional_dict.keys()))))

        return executor_id_
コード例 #11
0
ファイル: executor.py プロジェクト: Netflix/vmaf
    def executor_id(self):
        executor_id_ = TypeVersionEnabled.get_type_version_string(self)

        if self.optional_dict is not None:
            # include optional_dict info in executor_id for result store,
            # as parameters in optional_dict will impact result
            executor_id_ += '_{}'.format(
                '_'.join(
                    map(lambda k: '{k}_{v}'.format(k=k,v=self.optional_dict[k]),
                        sorted(self.optional_dict.keys()))
                )
            )

        return executor_id_
コード例 #12
0
ファイル: train_test_model.py プロジェクト: yuhjay/vmaf
 def model_id(self):
     return TypeVersionEnabled.get_type_version_string(self)
コード例 #13
0
ファイル: executor.py プロジェクト: wanjinchang/vmaf
 def executor_id(self):
     return TypeVersionEnabled.get_type_version_string(self)
コード例 #14
0
ファイル: subjective_model.py プロジェクト: taosenbai/vmaf
 def __init__(self, dataset_reader):
     TypeVersionEnabled.__init__(self)
     self.dataset_reader = dataset_reader
コード例 #15
0
ファイル: train_test_model.py プロジェクト: Alex-Song/vmaf
 def model_id(self):
     return TypeVersionEnabled.get_type_version_string(self)
コード例 #16
0
ファイル: subjective_model.py プロジェクト: Netflix/vmaf
 def __init__(self, dataset_reader):
     TypeVersionEnabled.__init__(self)
     self.dataset_reader = dataset_reader