Update dtk24.04.1镜像

README.md

@@ -14,7 +14,7 @@ U-Net模型通过编码器部分进行特征提取，然后通过对称的解码
 ## 环境配置
 ### Docker（方法一）
 ```
-docker pull image.sourcefind.cn:5000/dcu/admin/base/paddlepaddle:2.3.2-centos7.6-dtk-22.10-py38-latest
+docker pull image.sourcefind.cn:5000/dcu/admin/base/paddlepaddle:2.5.2-ubuntu20.04-dtk24.04.1-py3.8
 
 docker run -it -v /path/your_code_data/:/path/your_code_data/ --shm-size=32G --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name docker_name imageID bash
 #进入容器后
@@ -43,9 +43,9 @@ pip install visualdl==2.5.0
 
 关于本项目DCU显卡所需的特殊深度学习库可从[光合](https://developer.hpccube.com/tool/)开发者社区下载安装。
 ```
-DTK驱动：dtk22.10
+DTK驱动：dtk24.04.1
 python：python3.8
-paddle:2.3.2
+paddle:2.5.2
 ```
 `Tips：以上dtk驱动、python、paddle等DCU相关工具版本需要严格一一对应`
 
@@ -64,6 +64,7 @@ pip install visualdl==2.5.0
 - http://113.200.138.88:18080/aidatasets/project-dependency/cityscapes_preprocess
 
 数据集结构如下：
+数据集存放位置为/data路径下
 ```
 ├── cityscapesScripts
 │   ├── cityscapesscripts
@@ -146,7 +147,7 @@ pip install visualdl==2.5.0
         └── munster
 ```
 
-### 数据预处理
+<!--### 数据预处理
 在主目录下执行
 
 ```
@@ -162,10 +163,16 @@ pip install -e .
 ```
 pip install cityscapesscripts
 python tools/data/convert_cityscapes.py --cityscapes_path data/cityscapes --num_workers 8
-```
+```-->
 
 ## 训练
 ### 单机多卡
+
+下载model.pdparams并存放在/data/unet/train路径下
+- https://bj.bcebos.com/paddleseg/dygraph/cityscapes/unet_cityscapes_1024x512_160k/model.pdparams
+scnet快速下载链接：
+- http://113.200.138.88:18080/aimodels/findsource-dependency/unet_cityscapes_1024x512_160k/-/raw/main/model.pdparams
+
 ```
 export HIP_VISIBLE_DEVICES=0,1,2,3
 export HSA_FORCE_FINE_GRAIN_PCIE=1