Fix updated image size not used (#739)

Co-authored-by: magangloff <71378343+magangloff@users.noreply.github.com>
2022-09-12 16:45:16 +02:00 · 2022-09-12 16:45:16 +02:00 · b1850c7dca
commit b1850c7dca
parent 2e99efb27f
1 changed files with 1 additions and 1 deletions
--- a/test.py
+++ b/test.py
@ -59,7 +59,7 @@ def test(data,
        imgsz = check_img_size(imgsz, s=gs)  # check img_size
        
        if trace:
-            model = TracedModel(model, device, opt.img_size)
+            model = TracedModel(model, device, imgsz)

    # Half
    half = device.type != 'cpu' and half_precision  # half precision only supported on CUDA