当前位置:网站首页>图片上色项目 —— Deoldify
图片上色项目 —— Deoldify
2022-07-06 09:13:00 【zkkkkkkkkkkkkk】
目录
一、拉取Deoldify项目代码
git地址:https://github.com/jantic/DeOldify
二、下载预训练模型
2.1、步骤
1)在项目根目录创建models文件夹
2)下载预训练模型文件,将三个文件上传至models目录
三、图片上色
from deoldify import device
from deoldify.device_id import DeviceId
from deoldify.visualize import *
from pathlib import Path
import os
os.environ['KMP_DUPLICATE_LIB_OK'] = "TRUE"
import warnings
warnings.filterwarnings("ignore", category=UserWarning, message=".*?Your .*? set is empty.*?")
#choices: CPU, GPU0...GPU7
device.set(device=DeviceId.GPU0)
plt.style.use('dark_background')
torch.backends.cudnn.benchmark=True
render_factor=35
result_path = None
p = Path('D:\\Gitlab\\DeOldify\\test_images')
all_file_path = []
count = 0
for file_name in p.rglob('*.jpg'):
all_file_path.append(file_name)
count+=1
for file_name in p.rglob('*.png'):
all_file_path.append(file_name)
count+=1
for file_name in p.rglob('*.jppeg'):
all_file_path.append(file_name)
count+=1
print(f"遍历待上色文件夹结束,共有{count}个图片文件\n")
# 遍历图片,上色
for img_file_path in all_file_path:
# arctan
colorizer = get_image_colorizer(artistic=True)
img_file_name = str(img_file_path).split('\\')[-1]
print(f"开始处理{img_file_name}")
colorizer.plot_transformed_image(path=img_file_path, render_factor=render_factor, compare=True)
print(f"{img_file_name}处理结束")
print("\n")
# 结束
print("图片上色程序结束")
四、视频上色
from deoldify import device
from deoldify.device_id import DeviceId
from deoldify.visualize import *
plt.style.use('dark_background')
import warnings
warnings.filterwarnings("ignore", category=UserWarning, message=".*?Your .*? set is empty.*?")
#choices: CPU, GPU0...GPU7
device.set(device=DeviceId.GPU0)
colorizer = get_video_colorizer()
#NOTE: Max is 44 with 11GB video cards. 21 is a good default
render_factor=21
result_path = None
file_name_ext = 'D:\\Gitlab\\DeOldify\\test_video\\test1.mp4'
colorizer.colorize_from_file_name(file_name_ext, render_factor=render_factor)
边栏推荐
- Global and Chinese market of transfer switches 2022-2028: Research Report on technology, participants, trends, market size and share
- C language string function summary
- @controller,@service,@repository,@component区别
- Valentine's Day is coming, are you still worried about eating dog food? Teach you to make a confession wall hand in hand. Express your love to the person you want
- CSDN markdown editor
- Idea import / export settings file
- Mysql26 use of performance analysis tools
- API learning of OpenGL (2004) gl_ TEXTURE_ MIN_ FILTER GL_ TEXTURE_ MAG_ FILTER
- Global and Chinese market of wafer processing robots 2022-2028: Research Report on technology, participants, trends, market size and share
- Mysql25 index creation and design principles
猜你喜欢
MySQL18-MySQL8其它新特性
[recommended by bloggers] asp Net WebService background data API JSON (with source code)
Moteur de stockage mysql23
CSDN question and answer module Title Recommendation task (II) -- effect optimization
La table d'exportation Navicat génère un fichier PDM
Postman Interface Association
Mysql21 user and permission management
Mysql25 index creation and design principles
C language advanced pointer Full Version (array pointer, pointer array discrimination, function pointer)
1. Mx6u learning notes (VII): bare metal development (4) -- master frequency and clock configuration
随机推荐
FRP intranet penetration
SSM整合笔记通俗易懂版
How to change php INI file supports PDO abstraction layer
API learning of OpenGL (2001) gltexgen
Ubuntu 20.04 安装 MySQL
JDBC原理
Django运行报错:Error loading MySQLdb module解决方法
NPM an error NPM err code enoent NPM err syscall open
Have you mastered the correct posture of golden three silver four job hopping?
CSDN问答标签技能树(五) —— 云原生技能树
Mysql26 use of performance analysis tools
Idea import / export settings file
A trip to Macao - > see the world from a non line city to Macao
API learning of OpenGL (2004) gl_ TEXTURE_ MIN_ FILTER GL_ TEXTURE_ MAG_ FILTER
IDEA 导入导出 settings 设置文件
[recommended by bloggers] C # generate a good-looking QR code (with source code)
Remember a company interview question: merge ordered arrays
CSDN question and answer tag skill tree (I) -- Construction of basic framework
Navicat 导出表生成PDM文件
csdn-Markdown编辑器