В настоящее время я использую tenorboardX для визуализации входных изображений при обучении классификатору изображений Re sNet. Есть ли способ добавить заголовок изображения вместе с добавленным изображением? Я хотел бы, чтобы имя изображения (сохраненное в наборе данных) отображалось под изображением на экране тензорной доски.
До сих пор я пытался передать параметр comment в мою программу записи тензорной доски, которая не выглядит сделать работу. В настоящее время соответствующие строки моего кода:
pretrain_train_writer = SummaryWriter('log/pretrain_train')
img_grid = vutils.make_grid(inputs[tp_idx_0], normalize=True, scale_each=True, nrow=8)
pretrain_val_writer.add_image('true_positive_class_0', img_grid, global_step=epoch, comment = img_path)