From d03d23bb6f1bdca74d80790b391e02e271774d9d Mon Sep 17 00:00:00 2001 From: KellerJordan Date: Sat, 10 Sep 2022 09:06:34 +0000 Subject: [PATCH] fix random translate augmentation using uninitialized memory --- ffcv/transforms/translate.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ffcv/transforms/translate.py b/ffcv/transforms/translate.py index e53e157e..33d280b5 100644 --- a/ffcv/transforms/translate.py +++ b/ffcv/transforms/translate.py @@ -30,12 +30,13 @@ def __init__(self, padding: int, fill: Tuple[int, int, int] = (0, 0, 0)): def generate_code(self) -> Callable: my_range = Compiler.get_iterator() pad = self.padding + fill = self.fill def translate(images, dst): n, h, w, _ = images.shape # y_coords = randint(low=0, high=2 * pad + 1, size=(n,)) # x_coords = randint(low=0, high=2 * pad + 1, size=(n,)) - # dst = fill + dst[:] = fill dst[:, pad:pad+h, pad:pad+w] = images for i in my_range(n):