From bbb72659f27572c347e370bb31b89f5d882404da Mon Sep 17 00:00:00 2001 From: nemonameless Date: Thu, 21 Apr 2022 12:57:11 +0000 Subject: [PATCH] fix random_affine dsize tuple, test=document_fix --- ppdet/data/transform/operators.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppdet/data/transform/operators.py b/ppdet/data/transform/operators.py index a17f1c30b..8227b1db8 100644 --- a/ppdet/data/transform/operators.py +++ b/ppdet/data/transform/operators.py @@ -3149,7 +3149,7 @@ class Mosaic(BaseOperator): # warpAffine img = cv2.warpAffine( - img, M, dsize=input_dim, borderValue=(114, 114, 114)) + img, M, dsize=tuple(input_dim), borderValue=(114, 114, 114)) num_gts = len(labels) if num_gts > 0: -- GitLab