Update tools/demo_api.py
Browse files- tools/demo_api.py +1 -1
tools/demo_api.py
CHANGED
@@ -120,7 +120,7 @@ class Predictor(object):
|
|
120 |
return vis_res
|
121 |
|
122 |
def build_predictor(
|
123 |
-
exp_file, model_name, ckpt_path, device="cpu", fp16=False, fuse=False, trt=False, conf=0.
|
124 |
):
|
125 |
# load experiment
|
126 |
exp = get_exp(exp_file, model_name)
|
|
|
120 |
return vis_res
|
121 |
|
122 |
def build_predictor(
|
123 |
+
exp_file, model_name, ckpt_path, device="cpu", fp16=False, fuse=False, trt=False, conf=0.1, nms=0.1, tsize=None
|
124 |
):
|
125 |
# load experiment
|
126 |
exp = get_exp(exp_file, model_name)
|