anuashok/ocr captcha v2 This model is a fine tuned version of microsoft/trocr base printed on your custom dataset. captchas like Training Summary CER (Character Error Rate) : 0.02025931928687196 Hyperparameters : Learning Rate : 1.1081459294764632e 05 Batch Size : 4 Num Epochs : 3 Warmup Ratio : 0.07863134774153628 Weight Decay : 0.06248152825021373 Num Beams : 6 Length Penalty : 0.5095100725173662 Usage python from transformers import VisionEncoderDecoderModel, TrOCRProcessor import torch from PIL import Image Load model and processor processor = TrOCRProcessor.from pretrained("anuashok/ocr captcha v2") model = VisionEncoderDecoderModel.from pretrained("anuashok/ocr captcha v2") Load image image = Image.open('path to your image.jpg').convert("RGB") Prepare image pixel values = processor(image, return tensors="pt").pixel values Generate text generated ids = model.generate(pixel values) generated text = processor.batch decode(generated ids, skip special tokens=True)[0] print(generated text)
We use cookies for essential functionality and analytics. You can accept or reject analytics cookies.Cookie policy