From 9bf18554c3e4b250ba7063876f0191f573ffb7a4 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 18 Nov 2022 23:46:15 +0100 Subject: [PATCH] Revert `--save-txt` to default False (#10213) * Revert `--save-txt` to default False Signed-off-by: Glenn Jocher * Update tutorial.ipynb Signed-off-by: Glenn Jocher Signed-off-by: Glenn Jocher --- classify/predict.py | 2 +- classify/tutorial.ipynb | 5 ++--- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/classify/predict.py b/classify/predict.py index 96508d633da8..a9104ed315ec 100644 --- a/classify/predict.py +++ b/classify/predict.py @@ -196,7 +196,7 @@ def parse_opt(): parser.add_argument('--imgsz', '--img', '--img-size', nargs='+', type=int, default=[224], help='inference size h,w') parser.add_argument('--device', default='', help='cuda device, i.e. 0 or 0,1,2,3 or cpu') parser.add_argument('--view-img', action='store_true', help='show results') - parser.add_argument('--save-txt', action='store_false', help='save results to *.txt') + parser.add_argument('--save-txt', action='store_true', help='save results to *.txt') parser.add_argument('--nosave', action='store_true', help='do not save images/videos') parser.add_argument('--augment', action='store_true', help='augmented inference') parser.add_argument('--visualize', action='store_true', help='visualize features') diff --git a/classify/tutorial.ipynb b/classify/tutorial.ipynb index e035a7bda40d..9e65e53d8736 100644 --- a/classify/tutorial.ipynb +++ b/classify/tutorial.ipynb @@ -118,8 +118,7 @@ "image 1/2 /content/yolov5/data/images/bus.jpg: 224x224 minibus 0.39, police van 0.24, amphibious vehicle 0.05, recreational vehicle 0.04, trolleybus 0.03, 3.9ms\n", "image 2/2 /content/yolov5/data/images/zidane.jpg: 224x224 suit 0.38, bow tie 0.19, bridegroom 0.18, rugby ball 0.04, stage 0.02, 4.1ms\n", "Speed: 0.3ms pre-process, 4.0ms inference, 1.5ms NMS per image at shape (1, 3, 224, 224)\n", - "Results saved to \u001b[1mruns/predict-cls/exp\u001b[0m\n", - "2 labels saved to runs/predict-cls/exp/labels\n" + "Results saved to \u001b[1mruns/predict-cls/exp\u001b[0m\n" ] } ], @@ -1475,4 +1474,4 @@ }, "nbformat": 4, "nbformat_minor": 0 -} \ No newline at end of file +}