erlinersi commited on
Commit
515a7cd
·
1 Parent(s): 6852510

Update mtpnet_demo.py

Browse files
Files changed (1) hide show
  1. mtpnet_demo.py +1 -1
mtpnet_demo.py CHANGED
@@ -35,7 +35,7 @@ def detect(path, task, thickness, color, alpha):
35
  global mtpnet
36
  if mtpnet is None:
37
  ort.set_default_logger_severity(4)
38
- mtpnet = ort.InferenceSession('../weights/' + opt.weights)
39
 
40
  # Set Dataloader
41
  dataset = LoadImages(opt.source, img_size=opt.img_size)
 
35
  global mtpnet
36
  if mtpnet is None:
37
  ort.set_default_logger_severity(4)
38
+ mtpnet = ort.InferenceSession(opt.weights)
39
 
40
  # Set Dataloader
41
  dataset = LoadImages(opt.source, img_size=opt.img_size)