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

api_server.py: fix model path

Browse Source
main
Nils Koch 3 years ago
parent ed71c16912
commit d1ee2bc4f2
1 changed files with 1 additions and 1 deletions
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Show Stats Download Patch File Download Diff File
  1. 2
      api_server.py

2
api_server.py
Unescape View File

@ -33,7 +33,7 @@ class IrImage(BaseModel):
reference_pattern_path = '/home/nils/kinect_reference_cropped.png' reference_pattern_path = '/home/nils/kinect_reference_cropped.png'
reference_pattern = cv2.imread(reference_pattern_path) 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') device = torch.device('cuda:0')
model = Model(max_disp=256, mixed_precision=False, test_mode=True) model = Model(max_disp=256, mixed_precision=False, test_mode=True)

Write Preview
Loading…
Cancel
Save