From febc55d96a962368cd25bc1745da1df3ebafcf79 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 25 Dec 2018 13:21:02 +0100 Subject: [PATCH] updates --- detect.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/detect.py b/detect.py index cfd91665..61d7969d 100755 --- a/detect.py +++ b/detect.py @@ -67,7 +67,7 @@ def detect( # Get detections with torch.no_grad(): img = torch.from_numpy(img).unsqueeze(0).to(device) - # pred = torch.onnx._export(model, img, 'weights/model.onnx', verbose=True,); return # ONNX export + # pred = torch.onnx._export(model, img, 'weights/model.onnx', verbose=True); return # ONNX export pred = model(img) pred = pred[pred[:, :, 4] > conf_thres]