Module tch::vision::imagenet

source ·
Expand description

Helper functions for ImageNet like datasets.

Constants

CLASSES
CLASS_COUNT

Functions

load_from_dir
Loads a dataset from a directory.
load_image
Loads an image from a file and applies the ImageNet normalization.
load_image_and_resize
Loads an image from a file and resize it to the specified width and height. This applies the ImageNet normalization.
load_image_and_resize224
Loads an image from a file and resize it to 224x224. This applies the ImageNet normalization.
load_image_and_resize224_from_memory
Loads an image from memory and resize it to 224x224. This applies the ImageNet normalization.
load_image_and_resize_from_memory
Loads an image from memory and resize it to the specified width and height. This applies the ImageNet normalization.
load_image_from_memory
Loads an image from memory and applies the ImageNet normalization.
normalize
save_image
Saves an image to a path. This unapplies the ImageNet normalization.
top
Returns the top k classes as well as the associated scores.
unnormalize