【语义分割系列】基于改进Unet的火灾现场分割算法（完全创新可直接发论文）（二）

基于改进Unet的火灾现场分割算法（二）

前言

在（一）中因为引入HyCTAS自注意力机制，导致参数量增加很多。因次本节将对参数量这一块进行优化。首先设计了一个轻量的LightASPP结构，先将特征图处理后，再送入HyCTAS结构中。然后在Decoder层中，使用CSPHet替换原有的卷积层，在减少参数量的同时，也能提升模型精度。

1.代码部分

1.1LightASPP结构

LightASPP，在LRASPP结构的基础上实现，但是多引入了一个dilation为3的卷积结构。在nets/tricks/necks.py中添加如下代码

import torch
import torch.nn as nn
import torch.