def init()

in sample-apps/custom-model/code/application.py [0:0]

• 25 lines of code
• 3 McCabe index (conditional complexity)

    def init(self, parameters, inputs, outputs):
        """Initializes the application's attributes with parameters from the interface, and default values."""
        try:
            self.threshold = parameters.threshold
            self.person_index = parameters.person_index
            self.frame_num = 0
            self.inference_time_ms = 0
            self.inference_time_max = 0
            self.frame_time_ms = 0
            self.frame_time_max = 0
            self.epoch_frames = 150
            self.epoch_start = time.time()
            self.colours = np.random.rand(32, 3)
            self.buffered_media = {}
            self.buffered_image = {}
            # Load model
            logger.info("Loading model: " + parameters.model_name)
            self.model = panoramasdk.model()
            self.model.open(parameters.model_name, 1)
            os.environ['TVM_TENSORRT_USE_FP16'] = '1'
            # Create input and output arrays
            class_info = self.model.get_output(0)
            self.class_array = np.empty(class_info.get_dims(), dtype=class_info.get_type())
            logger.info("Initialization complete")
            return True

        except Exception as e:
            logger.error("Exception: {}".format(e))
            return False