This function plots images on tensotboard.
(writer, step, img, title, j=0)
| 35 | |
| 36 | |
| 37 | def plot_tb(writer, step, img, title, j=0): |
| 38 | """ |
| 39 | This function plots images on tensotboard. |
| 40 | """ |
| 41 | img_resized = resize_for_tb(img) |
| 42 | writer.add_image(title, img_resized[j].data, step) |
| 43 | |
| 44 | |
| 45 | def plot_norm_tb(writer, step, img, title, j=0): |