Commit efdd548b authored by mashun1's avatar mashun1
Browse files

Update README.md

parent 1adf587a
...@@ -42,26 +42,26 @@ $F_{3d}$ ...@@ -42,26 +42,26 @@ $F_{3d}$
## 环境配置 ## 环境配置
### Docker(方法一) ### Docker(方法一)
docker build --no-cache -t MetaPortrait:latest . docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk-23.04-py37-latest
docker run --rm --shm-size 10g --network=host --name=metaportrait --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -it <Image ID> bash docker run --shm-size 10g --network=host --name=metaportrait --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v 项目地址<绝对路径>:/home/ -it <Image ID> bash
pip install -r requirements.txt
cd sr_model/Basicsr cd sr_model/Basicsr
pip uninstall basicsr pip uninstall basicsr
python setup.py develop python setup.py develop
pip install urllib3==1.26.15 pip install urllib3==1.26.15
# 若遇到Dockerfile启动的方式安装环境需要长时间等待,可注释掉里面的pip安装,启动容器后再安装python库:pip install -r requirements.txt
### Docker(方法二) ### Docker(方法二)
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk-23.04-py37-latest docker build --no-cache -t MetaPortrait:latest .
docker run --rm --shm-size 10g --network=host --name=metaportrait --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v 项目地址<绝对路径>:/home/ -it <Image ID> bash docker run --shm-size 10g --network=host --name=metaportrait --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -it <Image ID> bash
pip install -r requirements.txt
cd sr_model/Basicsr cd sr_model/Basicsr
pip uninstall basicsr pip uninstall basicsr
python setup.py develop python setup.py develop
pip install urllib3==1.26.15 pip install urllib3==1.26.15
# 若遇到Dockerfile启动的方式安装环境需要长时间等待,可注释掉里面的pip安装,启动容器后再安装python库:pip install -r requirements.txt
### Anaconda (方法三) ### Anaconda (方法三)
1、关于本项目DCU显卡所需的特殊深度学习库可从光合开发者社区下载安装: 1、关于本项目DCU显卡所需的特殊深度学习库可从光合开发者社区下载安装:
...@@ -147,7 +147,7 @@ sr_model ...@@ -147,7 +147,7 @@ sr_model
3.训练sr model 3.训练sr model
cd sr_model cd ../sr_model
CUDA_VISIBLE_DEVICES=0 python -m torch.distributed.launch --nproc_per_node=1 --master_port=4321 Experimental_root/train.py -opt options/train/train_sr_hdtf.yml --launcher pytorch CUDA_VISIBLE_DEVICES=0 python -m torch.distributed.launch --nproc_per_node=1 --master_port=4321 Experimental_root/train.py -opt options/train/train_sr_hdtf.yml --launcher pytorch
...@@ -160,7 +160,7 @@ sr_model ...@@ -160,7 +160,7 @@ sr_model
2.提升图片分辨率 2.提升图片分辨率
cd sr_model cd ../sr_model
CUDA_VISIBLE_DEVICES=0 python -m torch.distributed.launch --nproc_per_node=1 --master_port=4321 Experimental_root/test.py -opt options/test/same_id_demo.yml --launcher pytorch CUDA_VISIBLE_DEVICES=0 python -m torch.distributed.launch --nproc_per_node=1 --master_port=4321 Experimental_root/test.py -opt options/test/same_id_demo.yml --launcher pytorch
...@@ -183,17 +183,17 @@ sr_model ...@@ -183,17 +183,17 @@ sr_model
### 算法类别 ### 算法类别
`计算机视觉` `风格迁移`
### 热点应用行业 ### 热点应用行业
`人脸识别,反欺诈,美颜特效` `科研,教育,广媒`
## 源码仓库及问题反馈 ## 源码仓库及问题反馈
https://developer.hpccube.com/codes/modelzoo/metaportrait_pytorch https://developer.hpccube.com/codes/modelzoo/metaportrait_pytorch
## 参考 ## 参考资料
https://github.com/Meta-Portrait/MetaPortrait https://github.com/Meta-Portrait/MetaPortrait
......
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