【深度学习】人物图片标签生成

最新推荐文章于 2024-08-07 11:40:50 发布

原创最新推荐文章于 2024-08-07 11:40:50 发布 · 3k 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#深度学习 #图像标签 #人物标签

机器学习专栏收录该内容

12 篇文章

订阅专栏

介绍Illustration2Vec，一款用于生成人物特征标签的开源工具。通过深度学习框架Chainer，可为图片添加丰富标签，如发型、眼睛颜色等，并提供代码示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

简介
图片处理的时候，经常会给图片标签处理，Illustration2Vec是网上一个比较好的任务标签开源库，用于生成人物特征的标签。

Illustration2Vec
下载地址：
https://github.com/rezoo/illustration2vec

并且在https://github.com/rezoo/illustration2vec/releases下载模板文件和标签文件。

Illustration2Vec用到chainer这个深度学习框架，以及一些其他库，如果没有则安装

pip install chainer Pillow scikit-image

Illustration2Vec有丰富的标签库，可以设定一个阈值，比如0.5，提取高于阈值的标签，或者给出指定的标签，获取对应的概率。

比如，百度图片下个萌妹子：
在这里插入图片描述

代码：

# -*- coding: utf-8 -*-
 
import i2v
from imageio import imread

illust2vec = i2v.make_i2v_with_chainer('illust2vec_tag_ver200.caffemodel', 'tag_list.json')
tags = ['blonde hair', 'brown hair', 'black hair', 'blue hair', 'pink hair', 'purple hair', 'green hair', 
        'red hair', 'silver hair', 'white hair', 'orange hair', 'aqua hair', 'grey hair',
        'long hair', 'short hair', 'twintails', 'drill hair', 'ponytail',
        'blue eyes', 'red eyes', 'brown eyes', 'green eyes', 'purple eyes', 'yellow eyes', 'pink eyes', 
        'aqua eyes', 'black eyes', 'orange eyes',
        'blush', 'smile', 'open mouth', 'hat', 'ribbon', 'glasses','photo']

def checkimage(filename, threshold=[]):    
    img = imread(filename)
    if (threshold==[]):
        tags = illust2vec.estimate_plausible_tags([img], threshold=0.5)
    else:
        tags = illust2vec.estimate_specific_tags([img], threshold)      
    return tags
 
if __name__ == '__main__':
    result = checkimage("E:/data/Jupyter/project21Code/ACGan/faces1/852.jpg")
    print(result)

获取标签：

[{‘general’: [(‘1girl’, 0.9891161918640137), (‘food’, 0.8902480602264404), (‘brown hair’, 0.8745222091674805), (‘solo’, 0.7861981391906738), (‘long hair’, 0.7356202602386475), (‘blush’, 0.7042754888534546), (‘fruit’, 0.6705906391143799), (‘eating’, 0.5464189052581787)], ‘copyright’: [], ‘character’: [], ‘rating’: <zip object at 0x0000000018A90B08>}]
标签反馈图片1个女孩，食物，棕色头发，长头发，吃饭等信息

也可以指定标签获取概率：
[{‘purple eyes’: 0.4817981719970703, ‘smile’: 0.06011262536048889, ‘orange hair’: 0.04947340488433838, ‘short hair’: 0.0349823534488678, ‘green hair’: 0.00037857890129089355, ‘brown eyes’: 0.02036881446838379, ‘aqua eyes’: 3.936886787414551e-05}]