From de4cd087f6e82991d9a757bcd05a866b6b2fc95e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?D=C3=A1vid=20Komorowicz?= Date: Tue, 26 Sep 2023 14:12:04 +0200 Subject: [PATCH] Readd random bacgkround to doc3d p=1.0 --- docmae/datamodule/mixed_module.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docmae/datamodule/mixed_module.py b/docmae/datamodule/mixed_module.py index 65ace0c..5a9477e 100644 --- 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), ]