Skip to content

CescMessi/photo_filter

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

自拍筛选工具

基于深度学习颜值评分进行自拍照片筛选,灵感来源:看到@小众软件 转发的 @睡到人间吃饭时 的微博

原理

使用ResNet18模型对文件夹中的每张照片进行评分,排序后将分数最高的n张复制到目标文件夹中。

模型来源:CharlesPikachu/isBeauty

部署工具:使用pnnx将PyTorch模型转为ncnn

若对模型效果不满意,可以自行训练替换models目录下的模型,或对配置文件config.inimodel项的路径进行修改。

使用方法

  1. 从源代码运行 首先安装好需要的包
$ pip install -r requirements.txt

随后使用python运行photo_filter.py

$ python photo_filter.py
  1. 使用编译好的程序(Windows) 下载photo_filter_windows_x64.zip,解压后双击photo_filter.exe运行。

使用效果

example

The emoji graphics are from the open source project Twemoji. The graphics are copyright 2020 Twitter, Inc and other contributors. The graphics are licensed under CC-BY 4.0. You should review the license before usage in your project.

About

使用深度学习模型评分的自拍筛选工具

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages