From d1ee2bc4f2cd824e6a6fab71e684e2110a8f4491 Mon Sep 17 00:00:00 2001 From: Nils Koch Date: Tue, 31 May 2022 11:57:12 +0200 Subject: [PATCH] api_server.py: fix model path --- api_server.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api_server.py b/api_server.py index 18434f4..015c1b2 100644 --- a/api_server.py +++ b/api_server.py @@ -33,7 +33,7 @@ class IrImage(BaseModel): reference_pattern_path = '/home/nils/kinect_reference_cropped.png' reference_pattern = cv2.imread(reference_pattern_path) -model_path = "../train_log/models/latest.pth" +model_path = "train_log/models/latest.pth" device = torch.device('cuda:0') model = Model(max_disp=256, mixed_precision=False, test_mode=True)