File tree Expand file tree Collapse file tree 1 file changed +2
-1
lines changed
Expand file tree Collapse file tree 1 file changed +2
-1
lines changed Original file line number Diff line number Diff line change @@ -2523,14 +2523,15 @@ def __call__(
25232523 img = convert_to_tensor (img , track_meta = get_track_meta ())
25242524 if lazy_ :
25252525 if self ._do_transform :
2526+ self .rand_affine_grid (sp_size , randomize = False , lazy = True )
25262527 affine = self .rand_affine_grid .get_transformation_matrix ()
25272528 else :
25282529 affine = convert_to_dst_type (torch .eye (len (sp_size ) + 1 ), img , dtype = self .rand_affine_grid .dtype )[0 ]
25292530 else :
25302531 if grid is None :
25312532 grid = self .get_identity_grid (sp_size , lazy_ )
25322533 if self ._do_transform :
2533- grid = self .rand_affine_grid (grid = grid , randomize = randomize , lazy = lazy_ )
2534+ grid = self .rand_affine_grid (grid = grid , randomize = False , lazy = lazy_ )
25342535 affine = self .rand_affine_grid .get_transformation_matrix ()
25352536 return affine_func ( # type: ignore
25362537 img ,
You can’t perform that action at this time.
0 commit comments