From a13a43c4fee7ec40aba8c0b53c5464682d93c3d9 Mon Sep 17 00:00:00 2001 From: Rory Clear <45880510+roryclear@users.noreply.github.com> Date: Sun, 27 Apr 2025 01:45:58 +0100 Subject: [PATCH] yolo 416 to 640 res (#10047) --- examples/webgpu/yolov8/compile.py | 2 +- examples/webgpu/yolov8/index.html | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/webgpu/yolov8/compile.py b/examples/webgpu/yolov8/compile.py index 667b5ac75a..0e88557f56 100644 --- a/examples/webgpu/yolov8/compile.py +++ b/examples/webgpu/yolov8/compile.py @@ -12,7 +12,7 @@ if __name__ == "__main__": yolo_infer = YOLOv8(w=0.25, r=2.0, d=0.33, num_classes=80) state_dict = safe_load(get_weights_location(yolo_variant)) load_state_dict(yolo_infer, state_dict) - prg, inp_sizes, out_sizes, state = export_model(yolo_infer, Device.DEFAULT.lower(), Tensor.randn(1,3,416,416), model_name="yolov8") + prg, inp_sizes, out_sizes, state = export_model(yolo_infer, Device.DEFAULT.lower(), Tensor.randn(1,3,640,640), model_name="yolov8") dirname = Path(__file__).parent safe_save(state, (dirname / "net.safetensors").as_posix()) with open(dirname / f"net.js", "w") as text_file: diff --git a/examples/webgpu/yolov8/index.html b/examples/webgpu/yolov8/index.html index dafa04afdd..3c84de9fe7 100644 --- a/examples/webgpu/yolov8/index.html +++ b/examples/webgpu/yolov8/index.html @@ -111,7 +111,7 @@