Module tch::vision::imagenet

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