search

huggingface / pixparse

Pixel Parsing. A reproduction of OCR-free end-to-end document understanding models with open data
11 stars 3 forks source link

Make transforms configurable #25

Open molbap opened 11 months ago

molbap commented 11 months ago

in PR #24, in task_cruller_pretrain.py, transform set is hard-coded to "better".

        self.image_preprocess_train = create_transforms(
            'better',
            input_cfg=self.image_input_cfg,
            training=True,
            interpolation='bicubic',
            crop_margin=False,  # True?
            align_long_axis=False,
        )

We should make compatible a composition of transforms with a config file.