Unet改进37：添加KACNConvNDLayer(2024最新改进方法)

AICurator

于 2024-09-17 19:06:54 发布

阅读量130

点赞数 4

分类专栏： Unet改进专栏文章标签：深度学习神经网络机器学习语义分割 unet

本文链接：https://blog.csdn.net/m0_51530640/article/details/142317487

版权

Unet改进专栏专栏收录该内容

38 篇文章 1 订阅 ¥99.90 ¥299.90

订阅专栏

超级会员免费看

本文内容：在不同位置添加KACNConvNDLayer

论文简介

1.步骤一

新建block/kacn_conv.py文件，添加如下代码：

import torch
import torch.nn as nn

##源码地址：https://github.com/SynodicMonth/ChebyKAN
class KACNConvNDLayer(nn.Module):
    def __init__(self, conv_class, norm_class, input_dim, output_dim, degree, kernel_size,
                 groups=1, padding=0, stride=1, dilation=1,
                 ndim: int = 2, dropout=0.0):
        super(KACNConvNDLayer, self).__init__()
        self.inputdim = input_dim
        self.outdim = output

了解本专栏