2067cb7429ffd95efc3d9ace255b234831615d39,robosat/tools/train.py,,get_dataset_loaders,#Any#Any#,189

Before Change

    image_transform = Compose(
        [
            ConvertImageMode("RGB"),
            Resize(target_size, Image.BILINEAR),
            CenterCrop(target_size),
            ImageToTensor(),
            Normalize(mean=mean, std=std),

After Change

    mean, std = dataset["stats"]["mean"], dataset["stats"]["std"]

    transform = JointCompose(
        [
            JointTransform(ConvertImageMode("RGB"), ConvertImageMode("P")),
            JointTransform(Resize(target_size, Image.BILINEAR), Resize(target_size, Image.NEAREST)),
            JointTransform(CenterCrop(target_size), CenterCrop(target_size)),
            JointRandomVerticalFlip(0.5),
            JointRandomHorizontalFlip(0.5),
            JointRandomRotation(0.5, 90),
            JointRandomRotation(0.5, 180),
            JointRandomRotation(0.5, 270),
            JointTransform(ImageToTensor(), MaskToTensor()),
            JointTransform(Normalize(mean=mean, std=std), None),
        ]
    )

    train_dataset = SlippyMapTilesConcatenation(
        [os.path.join(path, "training", "images")], os.path.join(path, "training", "labels"), transform

Instances