File size: 1,823 Bytes
fa9337c 57eea52 fa9337c 57eea52 fa9337c 57eea52 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 |
```markdown
# OCR with Hugging Face Transformers
```
This repository demonstrates how to perform Optical Character Recognition (OCR) using the Hugging Face Transformers library. The code in this repository utilizes a pretrained model for OCR on images.
## Prerequisites
Before you can run the code, you'll need to install the required libraries. You can do this with `pip`:
```python
pip install transformers
pip install pillow
```
## Usage
You can use the provided code to perform OCR on images. Here are the basic steps:
1. Import the necessary libraries:
```python
from transformers import VisionEncoderDecoderModel
from transformers import TrOCRProcessor, VisionEncoderDecoderModel
from PIL import Image
import requests
```
2. Load the pretrained OCR model and processor:
```python
model = VisionEncoderDecoderModel.from_pretrained("vanshp123/ocrmnist")
processor = TrOCRProcessor.from_pretrained('microsoft/trocr-base-stage1')
```
3. Load an image for OCR. You can replace `"/content/left_digit_section_4.png"` with the path to your image:
```python
image = Image.open("/content/left_digit_section_4.png").convert("RGB")
```
4. Process the image using the OCR processor and generate the text:
```python
pixel_values = processor(images=image, return_tensors="pt").pixel_values
generated_ids = model.generate(pixel_values)
generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
```
5. `generated_text` will contain the text recognized from the image.
## Example
You can use this code as a starting point for your OCR projects. It's important to adapt it to your specific use case and customize it as needed.
## License
This code uses models from the Hugging Face Transformers library, and you should review their licensing and usage terms for the pretrained models.
```
|