def add_data_parameters()

in options/train_options.py [0:0]

• 51 lines of code
• 1 McCabe index (conditional complexity)

    def add_data_parameters(self):
        dataset_params = self.parser.add_argument_group("data")
        dataset_params.add_argument("--dataset", type=str, default="mp3d")
        dataset_params.add_argument(
            "--use_semantics", action="store_true", default=False
        )
        dataset_params.add_argument(
            "--config",
            type=str,
            default="/private/home/ow045820/projects/habitat/habitat-api/configs/tasks/pointnav_rgbd.yaml",
        )
        dataset_params.add_argument(
            "--current_episode_train", type=int, default=-1
        )
        dataset_params.add_argument(
            "--current_episode_val", type=int, default=-1
        )
        dataset_params.add_argument("--min_z", type=float, default=0.5)
        dataset_params.add_argument("--max_z", type=float, default=10.0)
        dataset_params.add_argument("--W", type=int, default=256)
        dataset_params.add_argument(
            "--images_before_reset", type=int, default=1000
        )
        dataset_params.add_argument(
            "--image_type",
            type=str,
            default="both",
            choices=(
                "both",
                "translation",
                "rotation",
                "outpaint",
                "fixedRT_baseline",
            ),
        )
        dataset_params.add_argument("--max_angle", type=int, default=45)
        dataset_params.add_argument(
            "--use_z", action="store_true", default=False
        )
        dataset_params.add_argument(
            "--use_inv_z", action="store_true", default=False
        )
        dataset_params.add_argument(
            "--use_rgb_features", action="store_true", default=False
        )
        dataset_params.add_argument(
            "--use_alpha", action="store_true", default=False
        )
        dataset_params.add_argument(
            "--normalize_image", action="store_true", default=False
        )