def augment_single(self, data, gt_txt):
import calamari_ocr.thirdparty.ocrodeg as ocrodeg
data = ocrodeg.random_pad(data, (0, data.shape[1] * 5))
for sigma in [2, 5]:
noise = ocrodeg.bounded_gaussian_noise(data.shape, sigma, 3.0)
data = ocrodeg.distort_with_noise(data, noise)
return ocrodeg.printlike_multiscale(data, inverted=True), gt_txt
def augment_single(self, data, gt_txt):
import calamari_ocr.thirdparty.ocrodeg as ocrodeg
original_dtype = data.dtype
data = data.astype(np.float)
m = data.max()
data = data / (1 if m == 0 else m)
data = ocrodeg.random_pad(data, (0, data.shape[1] * 2))
for sigma in [2, 5]:
noise = ocrodeg.bounded_gaussian_noise(data.shape, sigma, 3.0)
data = ocrodeg.distort_with_noise(data, noise)
data = ocrodeg.printlike_multiscale(data, blur=1, inverted=True)
data = (data * 255 / data.max()).astype(original_dtype)
return data, gt_txt
def augment_single(self, data, gt_txt):
import calamari_ocr.thirdparty.ocrodeg as ocrodeg
original_dtype = data.dtype
data = data.astype(np.float)
m = data.max()
data = data / (1 if m == 0 else m)
data = ocrodeg.random_pad(data, (0, data.shape[1] * 2))
for sigma in [2, 5]:
noise = ocrodeg.bounded_gaussian_noise(data.shape, sigma, 3.0)
data = ocrodeg.distort_with_noise(data, noise)
data = ocrodeg.printlike_multiscale(data, blur=1, inverted=True)
data = (data * 255 / data.max()).astype(original_dtype)
return data, gt_txt