Fix updated image size not used (WongKinYiu#739)

Co-authored-by: magangloff <[email protected]>
ztjhz · 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