Cpt.Captain/CREStereo-pytorch-nxt
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

api_server.py: fix model path

Browse Source
This commit is contained in:
Nils Koch 2022-05-31 11:57:12 +02:00
parent ed71c16912
commit d1ee2bc4f2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api_server.py
View File

@ -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)