Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
U-KAN-optimize_pytorch
Commits
e983a983
Commit
e983a983
authored
Jul 08, 2024
by
chenzk
Browse files
v1.0.1
parent
7339f0b0
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
6 additions
and
6 deletions
+6
-6
Seg_UKAN/README.md
Seg_UKAN/README.md
+5
-5
Seg_UKAN/model.properties
Seg_UKAN/model.properties
+1
-1
No files found.
Seg_UKAN/README.md
View file @
e983a983
...
@@ -7,13 +7,13 @@ U-KAN精度高于Mamba、Transformer构建的Unet模型,本算法基于U-KAN
...
@@ -7,13 +7,13 @@ U-KAN精度高于Mamba、Transformer构建的Unet模型,本算法基于U-KAN
## 模型结构
## 模型结构
KAN已成为MLP的一个有前景的替代方案,U-KAN将新兴算法KAN的优势融入到成熟的U-Net Pipeline 中提升精度,同时增加可解释性,本算法进一步引入部分QKV注意力、KAN ffn等来增强模型的特征提取能力,弥补KAN的不足。
KAN已成为MLP的一个有前景的替代方案,U-KAN将新兴算法KAN的优势融入到成熟的U-Net Pipeline 中提升精度,同时增加可解释性,本算法进一步引入部分QKV注意力、KAN ffn等来增强模型的特征提取能力,弥补KAN的不足。
<div
align=
center
>
<div
align=
center
>
<img
src=
"./doc/structure.png"
/>
<img
src=
"./
Seg_UKAN/
doc/structure.png"
/>
</div>
</div>
## 算法原理
## 算法原理
U-KAN与其它图像分割算法一样,将图片数据送入模型后,依次通过预处理、特征提取,最后经过卷积预测出像素类别实现分割。
U-KAN与其它图像分割算法一样,将图片数据送入模型后,依次通过预处理、特征提取,最后经过卷积预测出像素类别实现分割。
<div
align=
center
>
<div
align=
center
>
<img
src=
"./doc/algorithm.png"
/>
<img
src=
"./
Seg_UKAN/
doc/algorithm.png"
/>
</div>
</div>
## 环境配置
## 环境配置
...
@@ -81,11 +81,11 @@ Seg_UKAN
...
@@ -81,11 +81,11 @@ Seg_UKAN
cd Seg_UKAN
cd Seg_UKAN
python train.py --arch UKAN --dataset busi --input_w 256 --input_h 256 --name busi_UKAN --data_dir ./inputs
python train.py --arch UKAN --dataset busi --input_w 256 --input_h 256 --name busi_UKAN --data_dir ./inputs
```
```
更多资料可参考源项目的
[
`README_origin`
](
.
.
/README_origin.md
)
更多资料可参考源项目的
[
`README_origin`
](
./README_origin.md
)
## result
## result
<div
align=
center
>
<div
align=
center
>
<img
src=
"./doc/seg.png"
/>
<img
src=
"./
Seg_UKAN/
doc/seg.png"
/>
</div>
</div>
### 精度
### 精度
...
@@ -102,7 +102,7 @@ python train.py --arch UKAN --dataset busi --input_w 256 --input_h 256 --name bu
...
@@ -102,7 +102,7 @@ python train.py --arch UKAN --dataset busi --input_w 256 --input_h 256 --name bu
### 热点应用行业
### 热点应用行业
`医疗,电商,制造,能源`
`医疗,电商,制造,能源`
## 源码仓库及问题反馈
## 源码仓库及问题反馈
-
http://developer.hpccube.com/codes/modelzoo/
repvit
-optimize_pytorch.git
-
http://developer.hpccube.com/codes/modelzoo/
u-kan
-optimize_pytorch.git
## 参考资料
## 参考资料
-
https://github.com/CUHK-AIM-Group/U-KAN.git
-
https://github.com/CUHK-AIM-Group/U-KAN.git
-
https://github.com/KindXiaoming/pykan.git
-
https://github.com/KindXiaoming/pykan.git
...
...
Seg_UKAN/model.properties
View file @
e983a983
# 模型编码
# 模型编码
modelCode
=
69
7
modelCode
=
74
7
# 模型名称
# 模型名称
modelName
=
u-kan-optimize_pytorch
modelName
=
u-kan-optimize_pytorch
# 模型描述
# 模型描述
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment