diff --git a/detect.py b/detect.py index 09bb7311..3b2d17ea 100644 --- a/detect.py +++ b/detect.py @@ -167,7 +167,7 @@ if __name__ == '__main__': parser.add_argument('--half', action='store_true', help='half precision FP16 inference') parser.add_argument('--device', default='', help='device id (i.e. 0 or 0,1) or cpu') parser.add_argument('--view-img', action='store_true', help='display results') - parser.add_argument('--save-txt', action='store_true', help='display results') + parser.add_argument('--save-txt', action='store_true', help='save results to *.txt') parser.add_argument('--classes', nargs='+', type=int, help='filter by class') parser.add_argument('--agnostic-nms', action='store_true', help='class-agnostic NMS') opt = parser.parse_args() diff --git a/utils/utils.py b/utils/utils.py index 5647702e..51d4127b 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -830,7 +830,7 @@ def print_mutation(hyp, results, bucket=''): def apply_classifier(x, model, img, im0): # applies a second stage classifier to yolo outputs - + im0 = [im0] if isinstance(im0, np.ndarray) else im0 for i, d in enumerate(x): # per image if d is not None and len(d): d = d.clone()