소스 검색

feat : web streaming

yjlim 2 년 전
부모
커밋
632eff4616
1개의 변경된 파일1개의 추가작업 그리고 2개의 파일을 삭제
  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