pull/6943/head
andyjpaddle 2022-07-20 06:28:00 +00:00
parent a3a095150e
commit 991e194e70
2 changed files with 2 additions and 13 deletions

View File

@ -3,12 +3,12 @@ Global:
epoch_num: 8
log_smooth_window: 200
print_batch_step: 200
save_model_dir: /paddle/backup/visionlan/LA_v2
save_model_dir: ./output/rec/r45_visionlan
save_epoch_step: 1
# evaluation is run every 2000 iterations
eval_batch_step: [0, 2000]
cal_metric_during_train: True
pretrained_model: ./pretrained_model/LF_2_ocr
pretrained_model:
checkpoints:
save_inference_dir:
use_visualdl: True

View File

@ -157,17 +157,6 @@ class TextRecognizer(object):
padding_im[:, :, 0:resized_w] = resized_image
return padding_im
def resize_norm_img_svtr(self, img, image_shape):
imgC, imgH, imgW = image_shape
resized_image = cv2.resize(
img, (imgW, imgH), interpolation=cv2.INTER_LINEAR)
resized_image = resized_image.astype('float32')
resized_image = resized_image.transpose((2, 0, 1)) / 255
resized_image -= 0.5
resized_image /= 0.5
return resized_image
def resize_norm_img_vl(self, img, image_shape):
imgC, imgH, imgW = image_shape