Class wandb.Image
A class for logging images to W&B.Args
ImageDataOrPathType
Accepts NumPy array/pytorch tensor of image data,
a PIL image object, or a path to an image file. If a NumPy
array or pytorch tensor is provided,
the image data will be saved to the given file type.
If the values are not in the range [0, 255] or all values are in the range [0, 1],
the image pixel values will be normalized to the range [0, 255]
unless
normalize is set to False.- pytorch tensor should be in the format (channel, height, width)
- NumPy array should be in the format (height, width, channel)
str | None
The PIL mode for an image. Most common are “L”, “RGB”, “RGBA”. Full Pillow docs for more information https://pillow.readthedocs.io/en/stable/handbook/concepts.html#modes
str | None
Label for display of image.
int | None
The grouping number for the image.
Classes | Sequence[dict] | None
A list of class information for the image, used for labeling bounding boxes, and image masks.
dict[str, BoundingBoxes2D] | dict[str, dict] | None
A dictionary containing bounding box information for the image. see https://docs.wandb.ai/models/ref/python/data-types/boundingboxes2d
dict[str, ImageMask] | dict[str, dict] | None
A dictionary containing mask information for the image. see https://docs.wandb.ai/models/ref/python/data-types/imagemask
str | None
The file type to save the image as. This parameter has no effect if
data_or_path is a path to an image file.bool
If
True, normalize the image pixel values to fall within the range of [0, 255]. Normalize is only applied if data_or_path is a numpy array or pytorch tensor.