akhaliq HF staff commited on
Commit
2f30fde
β€’
1 Parent(s): fc7e117

512 to 640

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ def detect(img,model):
29
  parser = argparse.ArgumentParser()
30
  parser.add_argument('--weights', nargs='+', type=str, default=model+".pt", help='model.pt path(s)')
31
  parser.add_argument('--source', type=str, default='Inference/', help='source') # file/folder, 0 for webcam
32
- parser.add_argument('--img-size', type=int, default=512, help='inference size (pixels)')
33
  parser.add_argument('--conf-thres', type=float, default=0.25, help='object confidence threshold')
34
  parser.add_argument('--iou-thres', type=float, default=0.45, help='IOU threshold for NMS')
35
  parser.add_argument('--device', default='', help='cuda device, i.e. 0 or 0,1,2,3 or cpu')
 
29
  parser = argparse.ArgumentParser()
30
  parser.add_argument('--weights', nargs='+', type=str, default=model+".pt", help='model.pt path(s)')
31
  parser.add_argument('--source', type=str, default='Inference/', help='source') # file/folder, 0 for webcam
32
+ parser.add_argument('--img-size', type=int, default=640, help='inference size (pixels)')
33
  parser.add_argument('--conf-thres', type=float, default=0.25, help='object confidence threshold')
34
  parser.add_argument('--iou-thres', type=float, default=0.45, help='IOU threshold for NMS')
35
  parser.add_argument('--device', default='', help='cuda device, i.e. 0 or 0,1,2,3 or cpu')