add test,founds bugs, fix it tomorrow

2025-01-14 01:08:44 +08:00
parent 6c4e6cb96f
commit 4b0f93cf6a
3 changed files with 84 additions and 2 deletions
@@ -33,6 +33,7 @@ class RknnSession:

        log.info(f"Loading RKNN model from {self.model_path} with {self.tpe} threads.")
        self.rknnpool = rknnPoolExecutor(rknnModel=self.model_path.as_posix(), TPEs=self.tpe, func=runInfrence)
+        log.info(f"Loaded RKNN model from {self.model_path} with {self.tpe} threads.")

    def __del__(self) -> None:
        self.rknnpool.release()