diff --git a/tools/infer_det.py b/tools/infer_det.py index ce16da8dc5fffb3f5fdc633aeb00a386a2d60d4f..5ce1f2ffc4e936488801feec5034f5c8ef45df89 100755 --- a/tools/infer_det.py +++ b/tools/infer_det.py @@ -52,7 +52,7 @@ def draw_det_res(dt_boxes, config, img, img_name, save_path): cv2.imwrite(save_path, src_im) logger.info("The detected Image saved in {}".format(save_path)) - +@paddle.no_grad() def main(): global_config = config['Global']