Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -31,7 +31,7 @@ def Custom_detect(img,mode):
|
|
31 |
parser = argparse.ArgumentParser()
|
32 |
parser.add_argument('--weights', nargs='+', type=str, default=model+".pt", help='model.pt path(s)')
|
33 |
parser.add_argument('--source', type=str, default='Temp_files/', help='source')
|
34 |
-
parser.add_argument('--img-size', type=int, default=
|
35 |
parser.add_argument('--conf-thres', type=float, default=0.25, help='object confidence threshold')
|
36 |
parser.add_argument('--iou-thres', type=float, default=0.45, help='IOU threshold for NMS')
|
37 |
parser.add_argument('--device', default='', help='cuda device, i.e. 0 or 0,1,2,3 or cpu')
|
|
|
31 |
parser = argparse.ArgumentParser()
|
32 |
parser.add_argument('--weights', nargs='+', type=str, default=model+".pt", help='model.pt path(s)')
|
33 |
parser.add_argument('--source', type=str, default='Temp_files/', help='source')
|
34 |
+
parser.add_argument('--img-size', type=int, default=640, help='inference size (pixels)')
|
35 |
parser.add_argument('--conf-thres', type=float, default=0.25, help='object confidence threshold')
|
36 |
parser.add_argument('--iou-thres', type=float, default=0.45, help='IOU threshold for NMS')
|
37 |
parser.add_argument('--device', default='', help='cuda device, i.e. 0 or 0,1,2,3 or cpu')
|