updates

74290825 · Glenn Jocher · e23b1a3d · 74290825
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

detect.py detect.py +1 -1

未找到文件。
--- a/detect.py
+++ b/detect.py
@@ -60,7 +60,7 @@ def detect(
        img = torch.from_numpy(img).unsqueeze(0).to(device)
        if ONNX_EXPORT:
            torch.onnx.export(model, img, 'weights/model.onnx', verbose=True)
-            return  # ONNX export
+            return
        pred = model(img)
        pred = pred[pred[:, :, 4] > conf_thres]  # remove boxes < threshold