ddrnet 分割学习笔记

目录

修改后可以加载预训练：

训练自己的数据代码：

---

修改后可以加载预训练：

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.BatchNorm2d
bn_mom = 0.1


def conv3x3