Readd random bacgkround to doc3d p=1.0

Dawars · Sep 26, 2023 · de4cd08 · de4cd08
1 parent a852df0
commit de4cd08
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docmae/datamodule/mixed_module.py b/docmae/datamodule/mixed_module.py
@@ -50,7 +50,7 @@ def __init__(self, docaligner_dir: str, doc3d_dir: str, background_dir: str, bat
         self.train_transform = transforms.Compose(
             [
                 RandomResizedCropWithUV((288, 288), scale=(0.08, 1.0) if self.crop else (1.0, 1.0), antialias=True),
-                T.RandomApply([ReplaceBackground(self.background_dir, "train1")], p=0.25),
+                ReplaceBackground(self.background_dir, "train1"),
                 transforms.ToImageTensor(),
                 transforms.ToDtype(torch.float32),
             ]