first commit

2025-05-13 09:03:38 +08:00
commit b98753bfbb
121 changed files with 8665 additions and 0 deletions
--- a/app_inference.py
+++ b/app_inference.py
@@ -0,0 +1,75 @@
+from PytorchBoot.application import PytorchBootApplication
+from runners.global_points_inferencer import GlobalPointsInferencer
+from runners.global_and_local_points_inferencer import GlobalAndLocalPointsInferencer
+from runners.local_points_inferencer import LocalPointsInferencer
+from runners.inference_server import InferencerServer
+from runners.evaluate_uncertainty_guide import EvaluateUncertaintyGuide
+@PytorchBootApplication("global_points_inference")
+class GlobalPointsInferenceApp:
+    @staticmethod
+    def start():
+        ''' 
+            call default or your custom runners here, code will be executed 
+        automatically when type "pytorch-boot run" or "ptb run" in terminal
+        
+            example:
+                Trainer("path_to_your_train_config").run()
+                Evaluator("path_to_your_eval_config").run()
+        '''       
+        GlobalPointsInferencer("./configs/local/global_only_inference_config.yaml").run()
+
+@PytorchBootApplication("global_and_local_points_inference")
+class GlobalAndLocalPointsInferenceApp:
+    @staticmethod
+    def start():
+        ''' 
+            call default or your custom runners here, code will be executed 
+        automatically when type "pytorch-boot run" or "ptb run" in terminal
+        
+            example:
+                Trainer("path_to_your_train_config").run()
+                Evaluator("path_to_your_eval_config").run()
+        '''       
+        GlobalAndLocalPointsInferencer("./configs/local/global_pts_and_local_pts_pose.yaml").run()
+
+@PytorchBootApplication("local_points_inference")
+class LocalPointsInferenceApp:
+    @staticmethod
+    def start():
+        ''' 
+            call default or your custom runners here, code will be executed 
+        automatically when type "pytorch-boot run" or "ptb run" in terminal
+        
+            example:
+                Trainer("path_to_your_train_config").run()
+                Evaluator("path_to_your_eval_config").run()
+        '''       
+        LocalPointsInferencer("./configs/local/local_only_inference_config.yaml").run()
+
+@PytorchBootApplication("server")
+class InferenceServerApp:
+    @staticmethod
+    def start():
+        ''' 
+            call default or your custom runners here, code will be executed 
+        automatically when type "pytorch-boot run" or "ptb run" in terminal
+        
+            example:
+                Trainer("path_to_your_train_config").run()
+                Evaluator("path_to_your_eval_config").run()
+        '''       
+        InferencerServer("./configs/server/server_inference_server_config.yaml").run()
+
+@PytorchBootApplication("evaluate_uncertainty_guide")
+class EvaluateUncertaintyGuideApp:
+    @staticmethod
+    def start():
+        ''' 
+            call default or your custom runners here, code will be executed 
+        automatically when type "pytorch-boot run" or "ptb run" in terminal
+        
+            example:
+                Trainer("path_to_your_train_config").run()
+                Evaluator("path_to_your_eval_config").run()
+        '''       
+        EvaluateUncertaintyGuide("./configs/local/uncertainty_guide_evaluation_config.yaml").run()