tensorboard-可视化pytorch网络模型

1.Anaconda环境配置

①pytorch 1.4

②tensorflow（2.0版本及对应的tensorboard）

③tensorboardX（采用pytorch自带的，可能不需要再安装）

2.SummaryWriter写出需要的log信息

①采用torch.utils.tensorboard下的SummaryWriter

②示例代码如下：

③程序运行后在程序所在文件夹上生成runs文件夹

3.执行tensorboard命令

• 采用Anaconda Prompt执行

• 采用pycharm执行

 

4.查看，在浏览器中输入对应网址：http://localhost:6006/

5.源码

# Rodrigo Caye Daudt
# https://rcdaudt.github.io/
# Daudt, R. C., Le Saux, B., & Boulch, A. "Fully convolutional siamese networks for change detection". In 2018 25th IEEE International Conference on Image Processing (ICIP) (pp. 4063-4067). IEEE.

import torch
import torch.nn as nn
import torch.nn.functional as F
from torch.nn.modules.padding import ReplicationPad2d

class SiamUnet_conc(nn.Module):
    """SiamUnet_conc segmentation network."""

    def __init__(self, input_nb