tidalove commited on
Commit
f4aaa98
·
verified ·
1 Parent(s): 0a89fd3

Update tools/demo_api.py

Browse files
Files changed (1) hide show
  1. 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.3, nms=0.3, tsize=None
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)