Commit 448c252c authored by dcuai's avatar dcuai
Browse files

Update README.md——dtk24.04.1

parent e7071db6
...@@ -22,11 +22,12 @@ Vary-toy 利用 Vary-tiny+ 管道来生成新的愿景 Vary-toy的词汇。这 ...@@ -22,11 +22,12 @@ Vary-toy 利用 Vary-tiny+ 管道来生成新的愿景 Vary-toy的词汇。这
## 环境配置 ## 环境配置
`注:在部署环境前需将Vary-toy/vary/demo/run_qwen_vary.py和Vary-toy/vary/model/vary_qwen_vary.py和Vary-toy/vary/model/vary_toy_qwen1_8.py中的模型路径改为本地模型路径,同时将Vary-toy/cache/models--HaoranWei--Vary-toy模型中的config.json文件中的模型路径改为本地路径,完成以上操作后再执行pip install e .指令。` `**注:在部署环境前需将[vary/demo/run_qwen_vary.py](https://developer.sourcefind.cn/codes/modelzoo/vary-toy_pytorch/-/blob/main/vary/demo/run_qwen_vary.py)、[vary/model/vary_qwen_vary.py](https://developer.sourcefind.cn/codes/modelzoo/vary-toy_pytorch/-/blob/main/vary/model/vary_qwen_vary.py)、[vary/model/vary_toy_qwen1_8.py](https://developer.sourcefind.cn/codes/modelzoo/vary-toy_pytorch/-/blob/main/vary/model/vary_toy_qwen1_8.py)中的模型路径改为本地模型路径,同时将Vary-toy模型中的config.json文件中的模型路径改为本地路径,完成以上操作后再执行pip install e .指令。**`
```
### Docker(方法一) ### Docker(方法一)
[光源](https://www.sourcefind.cn/#/service-details)拉取docker镜像的地址与使用步骤 [光源](https://www.sourcefind.cn/#/service-details)拉取docker镜像的地址与使用步骤
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu22.04-dtk23.10.1-py310 ```
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10
docker run -it -v /path/your_code_data/:/path/your_code_data/ -v /opt/hyhal/:/opt/hyhal/:ro --shm-size=64G --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name vary-toy <your imageID> bash docker run -it -v /path/your_code_data/:/path/your_code_data/ -v /opt/hyhal/:/opt/hyhal/:ro --shm-size=64G --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name vary-toy <your imageID> bash
...@@ -34,7 +35,6 @@ cd /path/your_code_data/ ...@@ -34,7 +35,6 @@ cd /path/your_code_data/
pip install e . pip install e .
pip install ninja
``` ```
### Dockerfile(方法二) ### Dockerfile(方法二)
...@@ -49,14 +49,12 @@ cd /path/your_code_data/ ...@@ -49,14 +49,12 @@ cd /path/your_code_data/
pip intall e . pip intall e .
pip install ninja
``` ```
### Anaconda(方法三) ### Anaconda(方法三)
关于本项目DCU显卡所需的特殊深度学习库可从[光合](https://developer.hpccube.com/tool/)开发者社区下载安装。 关于本项目DCU显卡所需的特殊深度学习库可从[光合](https://developer.hpccube.com/tool/)开发者社区下载安装。
``` ```
DTK驱动:dtk23.10 DTK驱动:dtk24.04.1
python:python3.10 python:python3.10
torch:2.1 torch:2.1
torchvision: 0.16.0 torchvision: 0.16.0
...@@ -73,7 +71,6 @@ cd /path/your_code_data/ ...@@ -73,7 +71,6 @@ cd /path/your_code_data/
pip install e . pip install e .
pip install ninja
``` ```
## 数据集 ## 数据集
...@@ -115,22 +112,12 @@ bash run.sh ...@@ -115,22 +112,12 @@ bash run.sh
<img src="./image/ocr_cn.png"/> <img src="./image/ocr_cn.png"/>
</div> </div>
### 车牌识别结果
<div align=center>
<img src="./image/car.png"/>
</div>
<div align=center>
<img src="./image/car_number.png"/>
</div>
## 应用场景 ## 应用场景
### 算法类别 ### 算法类别
`OCR` `OCR`
### 热点应用行业 ### 热点应用行业
`金融,教育,政府,科研,交通,广媒` `金融,教育,政府,科研`
## 预训练权重 ## 预训练权重
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment