目录
默认分割后特征下采样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
微信扫一扫
目录
默认分割后特征下采样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
相关推荐