nemo-retriever-ocr/cpp/README.md

Optimized Image Operations for PyTorch

Installation

python setup.py install

Usage

# It's important that you do this first
import torch
from pytorch_image_ops import color_transform, spatial_transform