Skip to content

Pytorch transforms based on numpy arrays instead of PIL Image

Notifications You must be signed in to change notification settings

gn64/pytorch-nptransforms

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

14 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Pytorch np_transforms

Pytorch transforms based on numpy arrays. I decided to re-write some of the standard pytorch transforms using only numpy operations that allow for High Dynamic Range image manipulation. The file exr_data.py include some methods for loading HDR images in exr format into numpy arrays and writing numpy arrays into exr files.

A list of implemented transforms based on numpy arrays are:

  • Bilateral Filter
  • Median Filter
  • Image Rotation (either randomly sample an angle between two bounds or with a fixed angle)
  • Random Horizontal Flip
  • Random Vertical Flip
  • Random Crop
  • Center Crop
  • Five Crops (4 courners + center)
  • Normalize 0-1 (Normalize the image between 0-1)
  • Random Erasing (Random Erasing Data Augmentation by Zhong et al.)
  • To Tensor
  • rgb2xyz (Change in the color space)
  • xyz2rgb (The opposite change in color space)
  • Lambda (Based on torchvision.transforms.Lambda)
  • Compose (Based on torchvision.transforms.Compose)
  • Normalize (Based on torchvision.transforms.Normalize)

Dependencies

Usage examples

Create a dataset that loads hdr images in .exr format:

import exr_data

trf = np_transforms.Compose([
    np_transforms.Scale(size=(256, 256)),
    np_transforms.RandomCrop(size=(224, 224)),
    np_transforms.RandomVerticalFlip(prob=0.5),
    np_transforms.RandomHorizontalFlip(prob=0.5),
    np_transforms.RotateImage(angles=(-15, 15)),
    np_transforms.ToTensor(),
])

data_train = exr_data.exrData(root=os.path.join(ROOT_DIR, 'train'),
                                  loader=exr_data.exr_loader,
                                  transform=trf)

                                  

Write a numpy array into a .exr file:

import exr_data

# let's assume we have a numpy array called 'pic' with the image stored in the form [HxWxC]

(Rs, Gs, Bs) = [pic[:, :, channel].tostring() for channel in range(pic.shape[-1])]
exr_data.exr_writer(out_path, size=im_size, pixel_values={'R': Rs, 'G': Gs, 'B': Bs})

About

Pytorch transforms based on numpy arrays instead of PIL Image

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%