Skip to content
master
Go to file
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 

README.md

Quadratum

Additional torchvision image transforms for practical usage.

Just for me. I needed it.

Note that all functors implemented here assumes an input image to be (H, W, C)-size np.uint8, ranged from 0 to 255.

  • Whiten: make all transparent pixels white.

  • Invert: invert RGB values.

  • Dominofy: limits the ratio of an image, like dominos.

  • Contain: contains an image into the given canvas (or box, whatever), just like, you know, the background-size: contain; thingy from CSS.


Installation

pip install quadratum

Usage

Similar to all the other transform functors:

from quadratum import transforms as qtrfm
from torchvision import transforms as vtrfm
transform = vtrfm.Compose([
    qtrfm.Whiten(),
    qtrfm.Dominofy(),
    qtrfm.Contain(256),
    vtrfm.ToPILImage(),
    vtrfm.CenterCrop(224),
    vtrfm.ToTensor(),
    vtrfm.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225]),
])

or, you can use some pre-defined transformers:

from skimage.io import imread
from quadratum.transformer import Transformer
transform = Transformer('resnet')
image = imread('image.png')  # image.shape => (640, 960, 4)
x = transform(image)  # x.size() => torch.Size([3, 224, 224])

The term "quadratum" means "square" in Latin. I wanted to make any noisy user-input images into fine-nice-good-well-godlike-heaven-deep-learning-applicable-preprocessed-square-images.

About

Additional torchvision image transforms for practical usage.

Topics

Resources

License

Packages

No packages published

Languages

You can’t perform that action at this time.