Visualizations

Tensorboard visuals

add_animated_gif(writer, tag, image_tensor, max_out, scale_factor, global_step=None)[source]

Creates an animated gif out of an image tensor and writes it with SummaryWriter.

Parameters
  • writer – Tensorboard SummaryWriter to write to

  • tag – Data identifier

  • image_tensor – tensor for the image to add, expected to be in CDHW format

  • max_out – maximum number of slices to animate through

  • scale_factor – amount to multiply values by. If the image data is between 0 and 1, using 255 for this value will scale it to displayable range

  • global_step – Global step value to record

add_animated_gif_no_channels(writer, tag, image_tensor, max_out, scale_factor, global_step=None)[source]

Creates an animated gif out of an image tensor and writes it with SummaryWriter.

Parameters
  • writer – Tensorboard SummaryWriter to write to

  • tag – Data identifier

  • image_tensor – tensor for the image to add, expected to be in DHW format

  • max_out – maximum number of slices to animate through

  • scale_factor – amount to multiply values by. If the image data is between 0 and 1, using 255 for this value will scale it to displayable range

  • global_step – Global step value to record

make_animated_gif_summary(tag, tensor, max_out=3, animation_axes=(3, ), image_axes=(1, 2), other_indices=None, scale_factor=1)[source]

Creates an animated gif out of an image tensor and returns Summary.

Parameters
  • tag – Data identifier

  • tensor – tensor for the image, expected to be in CHWD format

  • max_out – maximum number of slices to animate through

  • animation_axes – axis to animate on (not currently used)

  • image_axes – axes of image (not currently used)

  • other_indices – (not currently used)

  • scale_factor – amount to multiply values by. if the image data is between 0 and 1, using 255 for this value will scale it to displayable range