test(unraid): add offline benchmark and 10 minute NDI soak

2026-08-15 09:52:38 -07:00
parent 38230f55a0
commit f5e71edc7f
5 changed files with 118 additions and 2 deletions
@@ -30,6 +30,7 @@ class Models:
        from mmdet.apis import init_detector
        from mmpose.apis import init_model
        self.detector = init_detector(self.det_config, self.det_checkpoint, device="cuda:0")
+        self.detector.test_cfg.nms.iou_threshold = float(os.getenv("NMS_IOU_THRESHOLD", ".60"))
        self.pose = init_model(self.pose_config, self.pose_checkpoint, device="cuda:0")

    @torch.inference_mode()