目录

修改后可以加载预训练：

训练自己的数据代码：

默认分割后特征下采样8倍，最后用上采样恢复到原图；

修改后可以加载预训练：

import math
import torch
import numpy as np 
import torch.nn as nn
import torch.nn.functional as F
from torch.nn import init
from collections import OrderedDict

BatchNorm2d = nn.Bat