|
@@ -27,7 +27,7 @@ def stream():
|
|
|
model = attempt_load(WEIGHT, map_location='cuda:0').half()
|
|
|
|
|
|
|
|
|
- imgsz = 320
|
|
|
+ imgsz = 640
|
|
|
stride = int(model.stride.max())
|
|
|
imgsz = check_img_size(imgsz, s=stride)
|
|
|
device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
|
|
@@ -74,7 +74,6 @@ def stream():
|
|
|
|
|
|
pred = non_max_suppression(pred, 0.40, 0.45)
|
|
|
|
|
|
-
|
|
|
|
|
|
for i, det in enumerate(pred):
|
|
|
if True:
|