Explorar el Código

feat : web streaming

yjlim hace 2 años
padre
commit
632eff4616
Se han modificado 1 ficheros con 1 adiciones y 2 borrados
  1. 1 2
      yolov7-main/app_.py

+ 1 - 2
yolov7-main/app_.py

@@ -27,7 +27,7 @@ def stream():
     model = attempt_load(WEIGHT, map_location='cuda:0').half()
     # model = TracedModel(model, select_device('0'), img_size=640).half()
 
-    imgsz = 320
+    imgsz = 640
     stride = int(model.stride.max())  # model stride
     imgsz = check_img_size(imgsz, s=stride)  # check img_size
     device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
@@ -74,7 +74,6 @@ def stream():
             # Apply NMS
             pred = non_max_suppression(pred, 0.40, 0.45)
 
-
             # Process detections
             for i, det in enumerate(pred):  # detections per image
                 if True:  # batch_size >= 1