diff --git a/deploy/python/infer.py b/deploy/python/infer.py index bc503941cdc86c4553ec1db4b2c4c72fc7d8f0ab..f573c5f06604553f206c3346811145a21eb89bf8 100644 --- a/deploy/python/infer.py +++ b/deploy/python/infer.py @@ -152,7 +152,10 @@ class Detector(object): def postprocess(self, inputs, result): # postprocess output of predictor np_boxes_num = result['boxes_num'] - if sum(np_boxes_num) <= 0: + assert isinstance(np_boxes_num, np.ndarray), \ + '`np_boxes_num` should be a `numpy.ndarray`' + + if np_boxes_num.sum() <= 0: print('[WARNNING] No object detected.') result = {'boxes': np.zeros([0, 6]), 'boxes_num': np_boxes_num} result = {k: v for k, v in result.items() if v is not None} @@ -188,7 +191,7 @@ class Detector(object): shape: [N, im_h, im_w] ''' # model prediction - np_boxes, np_masks = None, None + np_boxes_num, np_boxes, np_masks = np.array([0]), None, None for i in range(repeats): self.predictor.run() output_names = self.predictor.get_output_names() @@ -428,7 +431,7 @@ class Detector(object): if not os.path.exists(self.output_dir): os.makedirs(self.output_dir) out_path = os.path.join(self.output_dir, video_out_name) - fourcc = cv2.VideoWriter_fourcc(* 'mp4v') + fourcc = cv2.VideoWriter_fourcc(*'mp4v') writer = cv2.VideoWriter(out_path, fourcc, fps, (width, height)) index = 1 while (1):