v1.0

bb3841ea · chenzk · bb3841ea · bb3841ea · bb3841ea · bb3841ea
Commit bb3841ea authored Jul 10, 2024 by chenzk
20 changed files
--- a/Seg_UKAN/doc/algorithm.png
+++ b/Seg_UKAN/doc/algorithm.png
--- a/Seg_UKAN/doc/seg.png
+++ b/Seg_UKAN/doc/seg.png
--- a/Seg_UKAN/doc/structure.png
+++ b/Seg_UKAN/doc/structure.png
--- a/Seg_UKAN/environment.yml
+++ b/Seg_UKAN/environment.yml
--- a/Seg_UKAN/inputs/busi.zip
+++ b/Seg_UKAN/inputs/busi.zip
--- a/Seg_UKAN/kan.py
+++ b/Seg_UKAN/kan.py
--- a/Seg_UKAN/losses.py
+++ b/Seg_UKAN/losses.py
--- a/Seg_UKAN/metrics.py
+++ b/Seg_UKAN/metrics.py
--- a/Seg_UKAN/requirements.txt
+++ b/Seg_UKAN/requirements.txt
--- a/Seg_UKAN/scripts.sh
+++ b/Seg_UKAN/scripts.sh
--- a/Seg_UKAN/train.py
+++ b/Seg_UKAN/train.py
--- a/Seg_UKAN/train.sh
+++ b/Seg_UKAN/train.sh
+python train.py --arch UKAN --dataset busi --input_w 256 --input_h 256 --name busi_UKAN  --data_dir ./inputs
--- a/Seg_UKAN/utils.py
+++ b/Seg_UKAN/utils.py
--- a/Seg_UKAN/val.py
+++ b/Seg_UKAN/val.py
--- a/Seg_UKAN/val.sh
+++ b/Seg_UKAN/val.sh
+python val.py --name busi_UKAN --output_dir ckpt
--- a/assets/framework-1.jpg
+++ b/assets/framework-1.jpg
--- a/assets/logo_1.png
+++ b/assets/logo_1.png
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
+FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10
+ENV DEBIAN_FRONTEND=noninteractive
+# RUN yum update && yum install -y git cmake wget build-essential
+# RUN source /opt/dtk-24.04.1/env.sh
+# # 安装pip相关依赖
+COPY requirements.txt requirements.txt
+RUN pip3 install -r requirements.txt -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com
+
--- a/docker/requirements.txt
+++ b/docker/requirements.txt
--- a/icon.png
+++ b/icon.png