Commit c5b7fb44 authored by dcuai's avatar dcuai
Browse files

Update README.md ——dtk24041

parent 927e75e1
...@@ -30,7 +30,7 @@ Vary-base:将新的视觉词汇表与原始词汇表(CLIP)合并,扩展 ...@@ -30,7 +30,7 @@ Vary-base:将新的视觉词汇表与原始词汇表(CLIP)合并,扩展
[光源](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 <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 <your imageID> bash
...@@ -38,7 +38,7 @@ cd /path/your_code_data/ ...@@ -38,7 +38,7 @@ cd /path/your_code_data/
pip install e . pip install e .
pip install ninja pip install numpy==1.24.3
``` ```
### Dockerfile(方法二) ### Dockerfile(方法二)
...@@ -53,13 +53,13 @@ cd /path/your_code_data/ ...@@ -53,13 +53,13 @@ cd /path/your_code_data/
pip intall e . pip intall e .
pip install ninja pip install numpy==1.24.3
``` ```
### 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
...@@ -76,7 +76,7 @@ cd /path/your_code_data/ ...@@ -76,7 +76,7 @@ cd /path/your_code_data/
pip install e . pip install e .
pip install ninja pip install numpy==1.24.3
``` ```
## 数据集 ## 数据集
...@@ -87,8 +87,6 @@ pip install ninja ...@@ -87,8 +87,6 @@ pip install ninja
## 训练 ## 训练
## 推理 ## 推理
**需严格按照本仓库代码目录进行排列** **需严格按照本仓库代码目录进行排列**
备注:在run.sh修改 --image-file 替换ocr文件 备注:在run.sh修改 --image-file 替换ocr文件
...@@ -130,16 +128,7 @@ bash run.sh ...@@ -130,16 +128,7 @@ bash run.sh
<img src="./assets/ocr_cn.png"/> <img src="./assets/ocr_cn.png"/>
</div> </div>
### 车牌识别结果 ### 内容描述结果
<div align=center>
<img src="./image/car.png"/>
</div>
<div align=center>
<img src="./assets/car_number.png"/>
</div>
### 内容识别结果
<div align=center> <div align=center>
<img src="./image/pic.jpg"/> <img src="./image/pic.jpg"/>
</div> </div>
...@@ -154,7 +143,7 @@ bash run.sh ...@@ -154,7 +143,7 @@ bash run.sh
`OCR` `OCR`
### 热点应用行业 ### 热点应用行业
`金融,教育,政府,科研,交通,广媒` `金融,教育,政府,科研`
## 预训练权重 ## 预训练权重
...@@ -165,6 +154,7 @@ bash run.sh ...@@ -165,6 +154,7 @@ bash run.sh
验证码可在本仓库留言索取。 验证码可在本仓库留言索取。
- [Download the CLIP-VIT-L]<https://huggingface.co/openai/clip-vit-large-patch14/> - [Download the CLIP-VIT-L]<https://huggingface.co/openai/clip-vit-large-patch14/>
- SCNet模型快速下载中心地址 [CLIP-VIT-L](http://113.200.138.88:18080/aimodels/clip-vit-large-patch14)
## 源码仓库及问题反馈 ## 源码仓库及问题反馈
......
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