glenn-jocher
commited on
Commit
•
81d3201
1
Parent(s):
177f39e
--save-txt store_true fix (#1323)
Browse files
detect.py
CHANGED
@@ -154,7 +154,7 @@ if __name__ == '__main__':
|
|
154 |
parser.add_argument('--iou-thres', type=float, default=0.45, help='IOU threshold for NMS')
|
155 |
parser.add_argument('--device', default='', help='cuda device, i.e. 0 or 0,1,2,3 or cpu')
|
156 |
parser.add_argument('--view-img', action='store_true', help='display results')
|
157 |
-
parser.add_argument('--save-txt', action='
|
158 |
parser.add_argument('--save-conf', action='store_true', help='save confidences in --save-txt labels')
|
159 |
parser.add_argument('--save-dir', type=str, default='runs/detect', help='directory to save results')
|
160 |
parser.add_argument('--name', default='', help='name to append to --save-dir: i.e. runs/{N} -> runs/{N}_{name}')
|
|
|
154 |
parser.add_argument('--iou-thres', type=float, default=0.45, help='IOU threshold for NMS')
|
155 |
parser.add_argument('--device', default='', help='cuda device, i.e. 0 or 0,1,2,3 or cpu')
|
156 |
parser.add_argument('--view-img', action='store_true', help='display results')
|
157 |
+
parser.add_argument('--save-txt', action='store_true', help='save results to *.txt')
|
158 |
parser.add_argument('--save-conf', action='store_true', help='save confidences in --save-txt labels')
|
159 |
parser.add_argument('--save-dir', type=str, default='runs/detect', help='directory to save results')
|
160 |
parser.add_argument('--name', default='', help='name to append to --save-dir: i.e. runs/{N} -> runs/{N}_{name}')
|