Update mtpnet_demo.py
Browse files- 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(
|
| 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)
|