Method update

model_manager.py:71–98

(self)

        return os.path.join(TRT_MODEL_DIR, f"{model_name}_weights_map.json")

    def update(self):
        trt_engines = [
            trt_file
            for trt_file in os.listdir(TRT_MODEL_DIR)
            if trt_file.endswith(".trt")
        ]

        tmp_all_models = copy.deepcopy(self.all_models)
        for cc, base_models in tmp_all_models.items():
            for base_model, models in base_models.items():
                tmp_config_list = {}
                for model_config in models:
                    if model_config["filepath"] not in trt_engines:
                        info(
                            "Model config outdated. {} was not found".format(
                                model_config["filepath"]
                            )
                        )
                        continue
                    tmp_config_list[model_config["filepath"]] = model_config

                tmp_config_list = list(tmp_config_list.values())
                if len(tmp_config_list) == 0:
                    self.all_models[cc].pop(base_model)
                else:
                    self.all_models[cc][base_model] = models

        self.write_json()

    def __del__(self):
        self.update()