代码拉取完成,页面将自动刷新
基于YOLOv8的人脸图片过滤与管理工具
曾逸夫,从事人工智能研究与开发;主研领域:计算机视觉;YOLOv8官方开源项目代码贡献人;YOLOv5官方开源项目代码贡献人
❤️ Github:https://github.com/Zengyf-CVer
2023-10-09
⚡ Face Img Filter v0.1正式上线
.
├── Face-Img_Filter # 项目名称
│ ├── util # 工具包
│ │ ├── time_format.py # 时间格式化
│ │ ├── classify_parse.py # 分类解析
│ │ └── img_fix.py # 图片修复
│ ├── sample # 样本图片集
│ ├── face_img_filter.py # 主程序文件
│ ├── setup.cfg # pre-commit CI检查源配置文件
│ ├── .pre-commit-config.yaml # pre-commit配置文件
│ ├── LICENSE # 项目许可
│ ├── .gitignore # git忽略文件
│ ├── README.md # 项目说明
│ ├── yolov8_widerface.md # 基于yolov8的widerface人脸检测教程
│ ├── yolov8_classify.md # 基于yolov8的人脸属性分类教程
│ └── requirements.txt # 脚本依赖包
conda create -n yolo python==3.9
conda activate yolo # 进入环境
git clone https://gitee.com/CV_Lab/face-img-filter.git
cd face-img-filter
pip install -r ./requirements.txt -U
❤️点击上方的链接,查看yolov8_widerface.md文件,关于基于WIDER FACE的YOLOv8人脸检测模型的构建过程
# 人脸过滤
python face_img_filter.py -ir ./origin_img_path -it ./target_img_path
# 调节置信度阈值
python face_img_filter.py -ir ./origin_img_path -it ./target_img_path -cf 0.8
# 调节IoU阈值
python face_img_filter.py -ir ./origin_img_path -it ./target_img_path -iu 0.6
# 调节推理尺寸
python face_img_filter.py -ir ./origin_img_path -it ./target_img_path -is 320
# 更换人脸检测模型
python face_img_filter.py -ir ./origin_img_path -it ./target_img_path -fm ./custom_face_model.pt
❤️点击上方的链接,查看yolov8_classify.md文件,关于基于YOLOv8的人脸属性分类模型的构建过程
# 性别过滤(默认为man)
python face_img_filter.py -ir ./origin_img_path -it ./target_img_path -gc -gn woman
# 更换人脸性别分类模型
python face_img_filter.py -ir ./origin_img_path -it ./target_img_path -gc -gn woman -gm ./face_gender_model.pt
{0: 'man', 1: 'woman'}
# 年龄过滤(默认为Youth)
python face_img_filter.py -ir ./origin_img_path -it ./target_img_path -ac -an Teenagers
# 更换人脸年龄分类模型
python face_img_filter.py -ir ./origin_img_path -it ./target_img_path -ac -an Teenagers -am ./face_age_model.pt
{0: 'Children', 1: 'Elderly', 2: 'Infants', 3: 'Middle', 4: 'Teenagers', 5: 'Youth'}
sample
目录提供图片样本,可以进行测试实验此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。