api_server.py: fix model path

3 years ago · d1ee2bc4f2
parent ed71c16912
commit d1ee2bc4f2
1 changed files with 1 additions and 1 deletions
--- 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)