def __init__(self, model_path, device='CPU', extensions=None, prob_threshold=0.6): ''' TODO: Use this to set your instance variables. ''' #raise NotImplementedError Model.__init__(self, model_path=model_path, device=device, extensions=extensions, prob_threshold=prob_threshold) self.model_name = 'Face Detection' self.model_path = model_path self.model_structure = model_path + '.xml' self.model_weights = model_path + '.bin'
def __init__(self, model_path, device='CPU', extensions=None, prob_threshold=0.6): ''' TODO: Use this to set your instance variables. ''' #raise NotImplementedError Model.__init__(self, model_path=model_path, device=device, extensions=extensions, prob_threshold=prob_threshold) self.model_name = 'Gaze_Estimation' self.model_path = model_path self.model_structure = model_path + '.xml' self.model_weights = model_path + '.bin' self.input_blob = [i for i in self.network.inputs.keys()] self.input_shape = self.network.inputs[self.input_blob[1]].shape self.output_blob = [i for i in self.network.outputs.keys()]