Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -45,10 +45,12 @@ def detect(img,model):
|
|
45 |
parser.add_argument('--name', default='exp', help='save results to project/name')
|
46 |
parser.add_argument('--exist-ok', action='store_true', help='existing project/name ok, do not increment')
|
47 |
parser.add_argument('--trace', action='store_true', help='trace model')
|
48 |
-
opt = parser.parse_args()
|
49 |
|
50 |
if model == "yolov7":
|
51 |
-
|
|
|
|
|
|
|
52 |
img.save("Inference/test.jpg")
|
53 |
source, weights, view_img, save_txt, imgsz, trace = opt.source, opt.weights, opt.view_img, opt.save_txt, opt.img_size, opt.trace
|
54 |
save_img = True # save inference images
|
|
|
45 |
parser.add_argument('--name', default='exp', help='save results to project/name')
|
46 |
parser.add_argument('--exist-ok', action='store_true', help='existing project/name ok, do not increment')
|
47 |
parser.add_argument('--trace', action='store_true', help='trace model')
|
|
|
48 |
|
49 |
if model == "yolov7":
|
50 |
+
parser.add_argument('--img-size', type=int, default=640, help='inference size (pixels)')
|
51 |
+
if model == "yolov7-e6e" or "yolov7-e6":
|
52 |
+
parser.add_argument('--img-size', type=int, default=1280, help='inference size (pixels)')
|
53 |
+
opt = parser.parse_args()
|
54 |
img.save("Inference/test.jpg")
|
55 |
source, weights, view_img, save_txt, imgsz, trace = opt.source, opt.weights, opt.view_img, opt.save_txt, opt.img_size, opt.trace
|
56 |
save_img = True # save inference images
|