Fix updated image size not used (WongKinYiu#739)

Co-authored-by: magangloff <[email protected]>
leeheewon-01 · Sep 12, 2022 · b1850c7 · b1850c7
1 parent 2e99efb
commit b1850c7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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