热门标签 | HotTags
当前位置:  开发笔记 > 编程语言 > 正文

用AI把好朋友的照片转换为铅笔素描——【模型识别2020之U2Net】

❤️【深度学习入门项目】❤️之【风格迁移】初识🚀本文墨理学AI带你了解如何用AI把好朋友的照片转换为铅笔素描❤️【带你了解】❤️❤️人脸》素描【humanportr

❤️【深度学习入门项目】❤️ 之 【风格迁移】 初识


🚀 本文 墨理学AI 带你了解如何 用AI把好朋友的照片转换为铅笔素描


❤️ 【带你了解】❤️


  • ❤️ 人脸 ===》素描【 human portrait drawing】
  • 💙 图像 ===》突出物体检测 【Salient Object Detection】


文章目录

    • 🚀我的好朋友大概就是这样呐
    • 🔔基础信息
    • 🎄环境搭建
    • 🎄项目结构如下
    • 🎄项目测试步骤(官方教程)
    • 🎄项目测试(个人记录)
    • 💬总结
    • 🎉可运行代码 + 数据
    • 📢 2021-1-11 素描数据集补充
    • 🚀🚀文末专栏推荐部分🚀🚀




🚀我的好朋友大概就是这样呐




❤️ 爱了 爱了💙

1-3
1-4

1-5

2-1

2-2
2-3



🔔基础信息




  • U2-Net: Going Deeper with Nested U-Structure for Salient Object Detection
  • https://arxiv.org/abs/2005.09007
  • https://github.com/NathanUA/U-2-Net

本博文,测试该代码的人像 - 素描 转换(风格迁移),记录环境搭建和效果测试;




🎄环境搭建




官方的库版本需求要求如下:

Python 3.6
numpy 1.15.2
scikit-image 0.14.0
python-opencv PIL 5.2.0
PyTorch 0.4.0
torchvision 0.2.1
glob

我自己的 Conda 环境搭建过程如下(安装torch==1.1.0 可用):


  • 服务器:ubuntu1~18.04
  • Quadro RTX 5000 16G
  • CUDA版本 V10.0.130

conda create -n nice python=3.6.9conda activate nicepip install torch==1.1.0pip install torchvision==0.3.0
或者(建议使用 conda 来安装)
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=10.0 -c pytorchpip install pillow==5.2.0pip install opencv-pythonpip install scipy
pip install thop



🎄项目结构如下



0



🎄项目测试步骤(官方教程)



1-1



🎄项目测试(个人记录)




一: 人像素描

python u2net_portrait_test.py

  • 该命令是瑞整个图像进行,素描
  • 生成效果如下:

3-1

二: 人脸素描

python u2net_portrait_demo.py

  • 会检测 图像中人脸位置,只对人脸进行素描

3-2

三: 测试命令

python u2net_portrait_composite.py -s 20 -a 0.5

  • -s indicates the sigma of gaussian function for blurring the orignal image and -a denotes the alpha weights of the orignal image when fusing them【理解起来就是变的更好看】

  • ❤️ 藏着掖着,不存在的, 好吧 ❤️

3-3

  • ❤️ 雨露均沾,大叔跑不了 ❤️

  • ❤️ 原图越高清, 转换效果越惊艳 ❤️

3-4

GPU内存占用:


  • 测试运行:GPU单卡最高占用 5073MiB



💬总结




  • 该论文本身是一篇关于 salient object detection(突出物检测) 的文章;
  • 不曾想有大佬 把它 用于 human portrait drawing(人物肖像素描)
  • 取得不错响应,成为一个如此有趣的项目,不明觉厉…

【突出物检测】的运行效果如下


不同于图像分割,【突出物检测】只检测图像中 最可爱的 你

4-0

4-1



🎉可运行代码 + 数据




考虑到一些小伙伴 官方代码 和 模型下载有困难,博主这里把此次博文对应代码上传到网盘,里面包含本次测试使用的代码和下载的预训练模型;


有需要的小伙伴,搜索关注博主同名公号:墨理学AI ,公号后台,回复 20200101 即可获取 云盘下载链接:

20200101



📢 2021-1-11 素描数据集补充




来自 https://github.com/yiranran/APDrawingGAN 的 人脸照片和对应艺术肖像画数据集 下载链接:


  • https://cg.cs.tsinghua.edu.cn/people/~Yongjin/APDrawingDB.zip



🚀🚀文末专栏推荐部分🚀🚀




  • 🎄如果感觉文章看完了不过瘾,还想更进一步,那么可以来我的其他 专栏 看一下哦~

  • ❤️ 图像风格转换 —— 代码环境搭建 实战教程【关注即可阅】!
  • 💜 图像修复-代码环境搭建-知识总结 实战教程 【据说还行】
  • 💙 超分重建-代码环境搭建-知识总结 解秘如何让图像变得清晰【老少皆宜】
  • 💛 YOLO专栏,只有实战,不讲道理 图像分类【建议收藏】!

  • 🎄 个人整理的Cuda系列 Linux安装教程【适合小白进阶】


  • 💜 ubuntu18给当前用户安装cuda11.2 图文教程 | 配置cuDNN8.1 |
  • 💜 Linux服务器下给当前用户安装自己的CUDA10.0
  • 💜 Linux下cuda10.0安装Pytorch和Torchvision
  • 💜 Linux 可以安装多个版本的Cuda 吗 | 给我一台新的服务器,我会怎么安排 Cuda
  • 💜 查看CUDA和cuDNN的版本号

9-6



推荐阅读
  • 本文详细介绍了Java中org.neo4j.helpers.collection.Iterators.single()方法的功能、使用场景及代码示例,帮助开发者更好地理解和应用该方法。 ... [详细]
  • Explore a common issue encountered when implementing an OAuth 1.0a API, specifically the inability to encode null objects and how to resolve it. ... [详细]
  • 本文详细介绍如何使用Python进行配置文件的读写操作,涵盖常见的配置文件格式(如INI、JSON、TOML和YAML),并提供具体的代码示例。 ... [详细]
  • 本文详细介绍了Java编程语言中的核心概念和常见面试问题,包括集合类、数据结构、线程处理、Java虚拟机(JVM)、HTTP协议以及Git操作等方面的内容。通过深入分析每个主题,帮助读者更好地理解Java的关键特性和最佳实践。 ... [详细]
  • DNN Community 和 Professional 版本的主要差异
    本文详细解析了 DotNetNuke (DNN) 的两种主要版本:Community 和 Professional。通过对比两者的功能和附加组件,帮助用户选择最适合其需求的版本。 ... [详细]
  • 本文深入探讨了Linux系统中网卡绑定(bonding)的七种工作模式。网卡绑定技术通过将多个物理网卡组合成一个逻辑网卡,实现网络冗余、带宽聚合和负载均衡,在生产环境中广泛应用。文章详细介绍了每种模式的特点、适用场景及配置方法。 ... [详细]
  • 本文详细介绍了 Apache Jena 库中的 Txn.executeWrite 方法,通过多个实际代码示例展示了其在不同场景下的应用,帮助开发者更好地理解和使用该方法。 ... [详细]
  • Windows服务与数据库交互问题解析
    本文探讨了在Windows 10(64位)环境下开发的Windows服务,旨在定期向本地MS SQL Server (v.11)插入记录。尽管服务已成功安装并运行,但记录并未正确插入。我们将详细分析可能的原因及解决方案。 ... [详细]
  • 本文详细介绍了 GWT 中 PopupPanel 类的 onKeyDownPreview 方法,提供了多个代码示例及应用场景,帮助开发者更好地理解和使用该方法。 ... [详细]
  • 本文详细介绍了Java中org.eclipse.ui.forms.widgets.ExpandableComposite类的addExpansionListener()方法,并提供了多个实际代码示例,帮助开发者更好地理解和使用该方法。这些示例来源于多个知名开源项目,具有很高的参考价值。 ... [详细]
  • 本文总结了在使用Ionic 5进行Android平台APK打包时遇到的问题,特别是针对QRScanner插件的改造。通过详细分析和提供具体的解决方法,帮助开发者顺利打包并优化应用性能。 ... [详细]
  • XNA 3.0 游戏编程:从 XML 文件加载数据
    本文介绍如何在 XNA 3.0 游戏项目中从 XML 文件加载数据。我们将探讨如何将 XML 数据序列化为二进制文件,并通过内容管道加载到游戏中。此外,还会涉及自定义类型读取器和写入器的实现。 ... [详细]
  • RecyclerView初步学习(一)
    RecyclerView初步学习(一)ReCyclerView提供了一种插件式的编程模式,除了提供ViewHolder缓存模式,还可以自定义动画,分割符,布局样式,相比于传统的ListVi ... [详细]
  • 掌握远程执行Linux脚本和命令的技巧
    本文将详细介绍如何利用Python的Paramiko库实现远程执行Linux脚本和命令,帮助读者快速掌握这一实用技能。通过具体的示例和详尽的解释,让初学者也能轻松上手。 ... [详细]
  • 从 .NET 转 Java 的自学之路:IO 流基础篇
    本文详细介绍了 Java 中的 IO 流,包括字节流和字符流的基本概念及其操作方式。探讨了如何处理不同类型的文件数据,并结合编码机制确保字符数据的正确读写。同时,文中还涵盖了装饰设计模式的应用,以及多种常见的 IO 操作实例。 ... [详细]
author-avatar
Ax珊
这个家伙很懒,什么也没留下!
PHP1.CN | 中国最专业的PHP中文社区 | DevBox开发工具箱 | json解析格式化 |PHP资讯 | PHP教程 | 数据库技术 | 服务器技术 | 前端开发技术 | PHP框架 | 开发工具 | 在线工具
Copyright © 1998 - 2020 PHP1.CN. All Rights Reserved | 京公网安备 11010802041100号 | 京ICP备19059560号-4 | PHP1.CN 第一PHP社区 版权所有