diff --git a/models/common.py b/models/common.py index 6ab82ab..7eca56a 100644 --- a/models/common.py +++ b/models/common.py @@ -629,7 +629,7 @@ class Detections: im = Image.fromarray(im.astype(np.uint8)) if isinstance(im, np.ndarray) else im # from np if pprint: - LOGGER.info(s.rstrip(', ')) + print(s.rstrip(', ')) if show: im.show(self.files[i]) # show if save: @@ -646,8 +646,7 @@ class Detections: def print(self): self.display(pprint=True) # print results - LOGGER.info(f'Speed: %.1fms pre-process, %.1fms inference, %.1fms NMS per image at shape {tuple(self.s)}' % - self.t) + print(f'Speed: %.1fms pre-process, %.1fms inference, %.1fms NMS per image at shape {tuple(self.s)}' % self.t) def show(self, labels=True): self.display(show=True, labels=labels) # show results