|
@@ -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
|