glenn-jocher commited on
Commit
81d3201
1 Parent(s): 177f39e

--save-txt store_true fix (#1323)

Browse files
Files changed (1) hide show
  1. detect.py +1 -1
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='store_false', 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}')
 
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}')