Replace nosave and noval with save and val (#127)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-01-01 17:50:06 +01:00
parent 598f17a472
commit 63c7a74691
7 changed files with 74 additions and 77 deletions
--- a/ultralytics/yolo/v8/classify/predict.py
+++ b/ultralytics/yolo/v8/classify/predict.py
@ -43,7 +43,7 @@ class ClassificationPredictor(BasePredictor):

        # write
        text = '\n'.join(f'{prob[j]:.2f} {self.model.names[j]}' for j in top5i)
-        if self.save_img or self.args.view_img:  # Add bbox to image
+        if self.args.save or self.args.show:  # Add bbox to image
            self.annotator.text((32, 32), text, txt_color=(255, 255, 255))
        if self.args.save_txt:  # Write to file
            with open(f'{self.txt_path}.txt', 'a') as f: