Skip to content

Latest commit

 

History

History
37 lines (23 loc) · 902 Bytes

vision.rst

File metadata and controls

37 lines (23 loc) · 902 Bytes

Vision Callbacks

Useful callbacks for vision models.

Note

We rely on the community to keep these updated and working. If something doesn't work, we'd really appreciate a contribution to fix!


Confused Logit

Shows how the input would have to change to move the prediction from one logit to the other

Example outputs:

Example of prediction confused between 5 and 8

pl_bolts.callbacks.vision.confused_logit.ConfusedLogitCallback


Tensorboard Image Generator

Generates images from a generative model and plots to tensorboard

pl_bolts.callbacks.vision.image_generation.TensorboardGenerativeModelImageSampler