Filter matplotlib>=3.7.2 tight layout warning (#3614)

2023-07-10 14:52:59 +02:00
parent 10eb0b6229
commit 391b7e67cf
2 changed files with 5 additions and 1 deletions
--- a/ultralytics/yolo/engine/exporter.py
+++ b/ultralytics/yolo/engine/exporter.py
@ -201,7 +201,7 @@ class Exporter:
        if self.args.half and (engine or onnx) and self.device.type != 'cpu':
            im, model = im.half(), model.half()  # to FP16

-        # Warnings
+        # Filter warnings
        warnings.filterwarnings('ignore', category=torch.jit.TracerWarning)  # suppress TracerWarning
        warnings.filterwarnings('ignore', category=UserWarning)  # suppress shape prim::Constant missing ONNX warning
        warnings.filterwarnings('ignore', category=DeprecationWarning)  # suppress CoreML np.bool deprecation warning