Update tools/demo_api.py

tools/demo_api.py

@@ -101,26 +101,8 @@ class Predictor(object):
             logger.info("Infer time: {:.4f}s".format(time.time() - t0))
         return outputs, img_info, pad
 
-    def visual(self, output, img_info, cls_conf=0.35):
-        ratio = img_info["ratio"]
-        img = img_info["raw_img"]
-        if output is None:
-            return img
-        output = output.cpu()
-
-        bboxes = output[:, 0:4]
-
-        # preprocessing: resize
-        bboxes /= ratio
-
-        cls = output[:, 6]
-        scores = output[:, 4] * output[:, 5]
-
-        vis_res = vis(img, bboxes, scores, cls, cls_conf, self.cls_names)
-        return vis_res
-
 def build_predictor(
-    exp_file, model_name, ckpt_path, device="cpu", fp16=False, fuse=False, trt=False, conf=0.1, nms=0.1, tsize=None
+    exp_file, model_name, ckpt_path, device="cpu", fp16=False, fuse=False, trt=False, conf=0, nms=0, tsize=None
 ):
     # load experiment
     exp = get_exp(exp_file, model_name)