Return save_dir with prediction results (#3240)

2023-06-18 02:34:31 +02:00 · 2023-06-18 02:34:31 +02:00 · 9c44560ec4
commit 9c44560ec4
parent 15c90bd404
2 changed files with 6 additions and 2 deletions
--- a/ultralytics/yolo/engine/results.py
+++ b/ultralytics/yolo/engine/results.py
@ -101,6 +101,7 @@ class Results(SimpleClass):
        self.speed = {'preprocess': None, 'inference': None, 'postprocess': None}  # milliseconds per image
        self.names = names
        self.path = path
+        self.save_dir = None
        self._keys = ('boxes', 'masks', 'probs', 'keypoints')

    def __getitem__(self, idx):