老照片修复AI模型介绍及使用训练方法（四）

四、使用Anaconda3搭建基础环境、AI模型下载安装

1.安装Anaconda3，官网：https://www.anaconda.com/ 

2.使用cona命令创建虚拟环境python3.8.18或者使用Anaconda3可视化界面创建虚拟环境

3.安装深度学习框架

#安装pytorch
pip install torch torchvision torchaudio
#安装Tensorflow
pip install --upgrade tensorflow==2.13.0 # 仅支持 CPU 的版本

4.安装ModelScope库、安装cv相关库

#安装基础库
pip install modelscope
#安装cv2库
pip install opencv-python
#安装cv相关依赖
pip install "modelscope[cv]" -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html

测试范例代码

import cv2
from modelscope.pipelines import pipeline
from modelscope.utils.constant import Tasks
from modelscope.outputs import OutputKeys

portrait_enhancement = pipeline(Tasks.image_portrait_enhancement, model='damo/cv_gpen_image-portrait-enhancement')
result = portrait_enhancement('https://modelscope.oss-cn-beijing.aliyuncs.com/test/images/marilyn_monroe_4.jpg')
cv2.imwrite('result.png', result[OutputKeys.OUTPUT_IMG])

完美执行无报错即可

未完待续...

作者：admin

TTF的家园-www.ttfde.top 个人博客以便写写东西，欢迎喜欢互联网的朋友一起交流!

TTF的家园 站点 QQ交谈