pychram中调用transforms.Resize,但是tensorboard图像显示没有变化
时间: 2024-10-21 09:12:05 浏览: 16
pytorch-transforms.Resize()用法.pdf
在PyCharm中使用`torchvision.transforms.Resize`处理图像数据后,TensorBoard可能会无法实时反映出这些转换后的图像,因为TensorBoard通常加载的是模型训练过程中的中间数据,而不是直接展示图片。当你将图片传递给`Resize`函数并改变尺寸后,这个操作是在内存中的,并不会直接影响TensorBoard上看到的数据。
如果你想要在TensorBoard中查看转换后的图像,你需要:
1. **保存结果**:先将变换后的张量保存成图片文件,比如`.png`或`.jpg`格式。
```python
transformed_image = resize(image)
torchvision.utils.save_image(transformed_image, 'transformed_image.png')
```
2. **添加到日志**:在TensorBoard的writer对象中记录这些图片数据。
```python
import torch.utils.tensorboard as tensorboard
with writer.as_default():
writer.add_image('resized image', torchvision.utils.make_grid(transformed_image), global_step=step)
```
3. **刷新TensorBoard**:每次有新的图像数据时,记得更新TensorBoard视图。打开TensorBoard,在相应的目录下刷新查看。
如果还是看不到变化,确认你在TensorBoard中选择的tag是否正确,并且确保保存和读取的路径设置无误。
阅读全文