def synthesize(model, folder):
from train_synth import synthesize
if model is None:
print('Please Enter the model path')
elif folder is None:
print(
'Please Enter the path of the folder you want to generate the targets for'
)
else:
print('Will generate the predictions at: ',
'/'.join(folder.split('/')[:-1]) + '/target_affinity')
print('Will generate the predictions at: ',
'/'.join(folder.split('/')[:-1]) + '/target_character')
print('Will generate the predictions at: ',
'/'.join(folder.split('/')[:-1]) + '/word_bbox')
os.makedirs('/'.join(folder.split('/')[:-1]) + '/target_affinity',
exist_ok=True)
os.makedirs('/'.join(folder.split('/')[:-1]) + '/target_character',
exist_ok=True)
os.makedirs('/'.join(folder.split('/')[:-1]) + '/word_bbox',
exist_ok=True)
synthesize.main(folder,
model_path=model,
base_path_character='/'.join(folder.split('/')[:-1]) +
'/target_character',
base_path_affinity='/'.join(folder.split('/')[:-1]) +
'/target_affinity',
base_path_bbox='/'.join(folder.split('/')[:-1]) +
'/word_bbox')
def train_synth(mode, model=None, folder=None):
"""
Training, Synthesizing, Testing using strong supervision on Synth-Text dataset
:param mode: 'train', 'test', 'synthesize'
:param model: Path to Model for Testing (only required if mode = 'test', 'synthesize'
:param folder: Path to folder to synthesize
:return: None
"""
mode = mode.lower()
if mode == 'train':
from train_synth import train
train.main()
elif mode == 'test':
from train_synth import test
if model is None:
print('Please Enter the model path')
else:
test.main(model)
elif mode == 'synthesize':
from train_synth import synthesize
if model is None:
print('Please Enter the model path')
elif folder is None:
print('Please Enter the path of the folder you want to generate the targets for')
else:
print('Will generate the predictions at: ', '/'.join(folder.split('/')[:-1])+'/target_affinity')
print('Will generate the predictions at: ', '/'.join(folder.split('/')[:-1])+'/target_character')
print('Will generate the predictions at: ', '/'.join(folder.split('/')[:-1]) + '/word_bbox')
os.makedirs('/'.join(folder.split('/')[:-1])+'/target_affinity', exist_ok=True)
os.makedirs('/'.join(folder.split('/')[:-1])+'/target_character', exist_ok=True)
os.makedirs('/'.join(folder.split('/')[:-1])+'/word_bbox', exist_ok=True)
synthesize.main(
folder,
model_path=model,
base_path_character='/'.join(folder.split('/')[:-1])+'/target_character',
base_path_affinity='/'.join(folder.split('/')[:-1])+'/target_affinity',
base_path_bbox='/'.join(folder.split('/')[:-1])+'/word_bbox',)
else:
print('Invalid Mode')
def synthesize(model, folder):
from train_synth import synthesize
if model is None:
print('Please Enter the model path')
elif folder is None:
print(
'Please Enter the path of the folder you want to generate the targets for'
)
else:
print('Will generate the Affinity Heatmap at: ',
'/'.join(folder.split('/')[:-1]) + '/affinity_heatmap')
print('Will generate the Character Heatmap at: ',
'/'.join(folder.split('/')[:-1]) + '/character_heatmap')
print('Will generate the Word Bbox at: ',
'/'.join(folder.split('/')[:-1]) + '/word_bbox')
print('Will generate the Character Bbox at: ',
'/'.join(folder.split('/')[:-1]) + '/character_bbox')
print('Will generate the Affinity Bbox at: ',
'/'.join(folder.split('/')[:-1]) + '/affinity_bbox')
print('Will generate the json annotations at: ',
'/'.join(folder.split('/')[:-1]) + '/json_annotations')
synthesize.main(
folder,
model_path=model,
base_path_character='/'.join(folder.split('/')[:-1]) +
'/character_heatmap',
base_path_affinity='/'.join(folder.split('/')[:-1]) +
'/affinity_heatmap',
base_path_bbox='/'.join(folder.split('/')[:-1]) + '/word_bbox',
base_path_char='/'.join(folder.split('/')[:-1]) +
'/character_bbox',
base_path_aff='/'.join(folder.split('/')[:-1]) + '/affinity_bbox',
base_path_json='/'.join(folder.split('/')[:-1]) +
'/json_annotations',
)