pytorch-image 1.0.0

Minimalistic library designed to be an interface for the model input with a lightweight `Image` class.

pytorch-image

Minimalistic library designed to be an interface for the model input with a lightweight Image class.

from my_model import MyModel, Image

image = Image.from_numpy(numpy_image)
predictions = model(image)

It also offers a simple interface for essential image operations such as loading, resizing, augmenting, and saving.

Installation

pip install pytorch-image

poetry add pytorch-image

Usage

from pytorch_image import Image
import albumentations


image = Image.open("path/to/image.png")

image.augment(albumentations.HorizontalFlip())

image.torch()

Image.from_numpy
Image.from_torch
Image.from_pil_image
Image.open