pass degrading scales for image enhancement as a json file

This commit is contained in:
vahidrezanezhad 2024-05-28 10:01:17 +02:00
parent 5aa6ee0010
commit 29ddd4d909
2 changed files with 13 additions and 6 deletions

View file

@ -64,13 +64,17 @@ def pagexml2label(dir_xml,dir_out,type_output,config):
help="directory where original images will be written as labels.", help="directory where original images will be written as labels.",
type=click.Path(exists=True, file_okay=False), type=click.Path(exists=True, file_okay=False),
) )
def image_enhancement(dir_imgs, dir_out_images, dir_out_labels): @click.option(
#dir_imgs = './training_data_sample_enhancement/images' "--scales",
#dir_out_images = './training_data_sample_enhancement/images_gt' "-scs",
#dir_out_labels = './training_data_sample_enhancement/labels_gt' help="json dictionary where the scales are written.",
type=click.Path(exists=True, dir_okay=False),
)
def image_enhancement(dir_imgs, dir_out_images, dir_out_labels, scales):
ls_imgs = os.listdir(dir_imgs) ls_imgs = os.listdir(dir_imgs)
ls_scales = [ 0.5, 0.55, 0.6, 0.65, 0.7, 0.75, 0.8, 0.85, 0.9] with open(scales) as f:
scale_dict = json.load(f)
ls_scales = scale_dict['scales']
for img in tqdm(ls_imgs): for img in tqdm(ls_imgs):
img_name = img.split('.')[0] img_name = img.split('.')[0]

3
scales_enhancement.json Normal file
View file

@ -0,0 +1,3 @@
{
"scales" : [ 0.5, 0.55, 0.6, 0.65, 0.7, 0.75, 0.8, 0.85, 0.9]
}