Models, Datasets and Transformations for Images
Base loader
Batched Non-maximum Suppression (NMS)
Box Area
Box Convert
box_cxcywh_to_xyxy
Box IoU
box_xywh_to_xyxy
box_xyxy_to_cxcywh
box_xyxy_to_xywh
Caltech Datasets
CIFAR datasets
Clip Boxes to Image
COCO Caption Dataset
COCO Detection Dataset
Convert COCO polygon to mask tensor (Robust Version)
Draws bounding boxes on image.
Draws Keypoints
Draw segmentation masks
EuroSAT datasets
FER-2013 Facial Expression Dataset
FGVC Aircraft Dataset
Flickr Caption Datasets
Oxford Flowers 102 Dataset
Generalized Box IoU
Create an image folder dataset
Load an Image using ImageMagick
MNIST and Derived Datasets
AlexNet Model Architecture
Inception v3 model
Constructs a MobileNetV2 architecture from [MobileNetV2: Inverted Resi...
ResNet implementation
VGG implementation
Non-maximum Suppression (NMS)
Oxford-IIIT Pet Classification Datasets
Oxford-IIIT Pet Segmentation Dataset
Remove Small Boxes
Display image tensor
Display image tensor
Tiny ImageNet dataset
Adjust the brightness of an image
Adjust the contrast of an image
Adjust the gamma of an RGB image
Adjust the hue of an image
Adjust the color saturation of an image
Apply affine transformation on an image keeping image center invariant
Crops the given image at the center
Randomly change the brightness, contrast and saturation of an image
Convert a tensor image to the given dtype and scale the values accor...
Crop the given image at specified location and output size
Crop image into four corners and a central crop
Convert image to grayscale
Horizontally flip a PIL Image or Tensor
Transform a tensor image with a square transformation matrix and a mea...
Normalize a tensor image with mean and standard deviation
Pad the given image on all sides with the given "pad" value
Perspective transformation of an image
Random affine transformation of the image keeping center invariant
Apply a list of transformations randomly with a given probability
Apply single transformation randomly picked from a list
Crop the given image at a random location
Randomly selects a rectangular region in an image and erases its pixel...
Randomly convert image to grayscale with a given probability
Horizontally flip an image randomly with a given probability
Apply a list of transformations in a random order
Random perspective transformation of an image with a given probability
Crop image to random size and aspect ratio
Rotate the image by angle
Vertically flip an image randomly with a given probability
Resize the input image to the given size
Crop an image and resize it to a desired size
Convert RGB Image Tensor to Grayscale
Angular rotation of an image
Crop an image and the flipped image each into four corners and a centr...
Convert an image to a tensor
Vertically flip a PIL Image or Tensor
A simplified version of torchvision.utils.make_grid
Provides access to datasets, models and preprocessing facilities for deep learning with images. Integrates seamlessly with the 'torch' package and it's 'API' borrows heavily from 'PyTorch' vision package.
Useful links