【深度学习图片】图片清洗,只留下图像中只有一张人脸的,而且人脸是全的

环境:


conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia -ypip install onnx==1.15 onnxruntime-gpu==1.17pip install insightface==0.7.3pip install opencv-pythonpip install gradio

图片清洗,只留下图像中只有一张人脸的,而且人脸是全的。

import os
import shutilimport cv2
import numpy as np
from insightface.app import FaceAnalysisdef is_full_face(facedata, img_shape, threshold=0.5):"""判断是否为整张脸:param facedata: 人脸数据:param img_shape: 图片尺寸:param threshold: 阈值:return: 布尔值,True 表示整张脸,False 表示部分脸"""img_width, img_height = img_shape[1], img_shape[0]# 检查人脸关键点是否在图片内部kps = facedata['kps']if np.all(kps >= 10) and np.all(kps[:, 0] <= img_width - 10) and np.all(kps[:, 1] <= img_height - 10):keypoints_inside = Trueelse:keypoints_inside = False# 满足阈值并且关键点在图片内部return keypoints_insidedef listPathAllfiles(dirname):result = []for maindir, subdir, file_name_list in os.walk(dirname):for filename in file_name_list:apath = os.path.join(maindir, filename)result.append(apath)return result# 使用的检测模型名为 buffalo_sc
app = FaceAnalysis(name='buffalo_sc', providers=['CUDAExecutionProvider'])
app.prepare(ctx_id=0, det_size=(640, 640))  # ctx_id 小于0 表示用 CPU 预测,det_size 表示 resize 后的图片分辨率src = r"C:\Users\Administrator\Pictures\girl_no_train\mangguo_dst"
dst = r"C:\Users\Administrator\Pictures\girl_no_train\mangguo_dst2"
os.makedirs(dst, exist_ok=True)files = listPathAllfiles(src)
for file in files:img = cv2.imread(file)  # 读取图片faces = app.get(img)  # 得到人脸信息if len(faces) == 0:continueif len(faces) > 1:continuefor facedata in faces:if is_full_face(facedata, img.shape):# print("This is a full face.")shutil.copy(file, dst)

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/diannao/48222.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

java中传引用问题

在 Java 中&#xff0c;所有对象都是通过引用传递的&#xff0c;而基本数据类型是通过值传递的。 引用传递&#xff1a; 当一个对象作为参数传递给方法时&#xff0c;传递的是对象的引用。对这个对象引用进行的修改会影响到原始对象。例如&#xff1a; public class Test {p…

CentOS Mysql8 数据库安装

添加mysql yum仓库 这里安装的是8.0版本&#xff0c;如需其他版本在此查看mysql版本列表 wget https://dev.mysql.com/get/mysql80-community-release-el7-3.noarch.rpm sudo rpm -Uvh mysql80-community-release-el7-3.noarch.rpm安装mysql sudo yum install mysql-server …

新能源汽车缩写含义

汽车类型的缩写及其对应的英文全称和中文解释的表格&#xff1a; 缩写英文全称中文解释EVElectric Vehicle电动车HEVHybrid Electric Vehicle混合动力电动车BEVBattery Electric Vehicle纯电动车PHEVPlug-in Hybrid Electric Vehicle插电式混合动力电动车ICEVInternal Combust…

在eclipse中使用Junit单元测试的准备工作和基本介绍

JUnit准备工作和基本介绍 JUnit是一个Java语言的单元测试框架。。多数Java的开发环境都已经集成了JUnit作为单元测试的工具。 Junit 测试也是程序员测试&#xff0c;即所谓的白盒测试&#xff0c;它需要程序员知道被测试的代码如何完成功能&#xff0c;以及完成什么样的功能 …

sqlalchemy事件监听

sqlalchemy事件监听 SQLAlchemy 中的事件监听允许您在特定事件发生时执行自定义的 Python 代码。这些事件可以是与ORM(对象关系映射)或核心组件相关的操作,比如表、类、会话或事务的插入、更新、删除等操作。通过事件监听,您可以实现日志记录、审计或执行业务规则等功能。…

Ubuntu24.04 NFS 服务配置

1、NFS 介绍 NFS 是 Network FileSystem 的缩写&#xff0c;顾名思义就是网络文件存储系统&#xff0c;它允许网络中的计算机之间通过 TCP/IP 网络共享资源。通过 NFS&#xff0c;我们本地 NFS 的客户端应用可以透明地读写位于服务端 NFS 服务器上的文件&#xff0c;就像访问本…

宠物空气净化器哪款品牌好?口碑好的猫用空气净化器排名

猫咪每年掉毛两次&#xff0c;一次掉半年的现象真让人头疼。作为一位5年资深铲屎官&#xff0c;特别是在掉毛季节&#xff0c;猫毛无处不在&#xff0c;对此深有体会。宠物空气净化器已成为铲屎官们的救星&#xff0c;能迅速清理家中的宠物毛发和异味&#xff0c;是养猫家庭的必…

关于APPCRASH的问题处理

最近小编遇到了APPCRASH的问题,打开任意一个exe可执行文件运行时候,windows记录是APPCRASH的问题. 因为是第一次遇到,当时就懵逼了,通过查找资料,现在把这类问题总结一下: 1.软件不兼容引起的crash,通常我们启用兼容性运行就可以解决,前提是以前在其他版本的系统上运行,现在换…

普中51单片机:LED点阵屏组成结构及实现方法详解(九)

文章目录 引言什么是LED点阵屏&#xff1f;工作原理74HC595移位寄存器基本引脚作用级联工作原理 电路图代码演示——16*16LED点阵屏轮播点亮每行LED代码演示——显示数字0代码演示——16*16游动字幕显示 引言 LED点阵屏作为一种广泛应用于现代显示技术的设备&#xff0c;因其能…

docker自建rustdesk-server远程桌面

rustdesk简介 RustDesk 是一款可以平替 TeamViewer 的开源软件&#xff0c;旨在提供安全便捷的自建方案。 RustDesk 是一款功能齐全的远程桌面应用&#xff0c;具有以下特性&#xff1a; 支持 Windows、macOS、Linux、iOS、Android、Web 等多个平台。支持 VP8 / VP9 / AV1 …

Qt实现仿微信在线聊天工具(服务器、客户端)V1_ 04

上一篇实现了客户端与服务器的通信,这一篇继续实现相关功能 本章内容 服务器与数据库的连接通信格式的规范登录信息的验证 1.数据库的建立 这里连接的是Mysql8.0数据库,如果想要简单点可以直接用sqlite3数据库,调用逻辑基本差不多,数据库语法也基本一致。 在服务器工程里…

python需要掌握那些语法

1-list数据类型 内置方法查看长度len&#xff08;list&#xff09; 2.array数据类型 查看形状 3.tuple 取出元组 t (1, 2, 3, 4, 5) # 取出第一个元素 2first_element t[0] 3print(first_element) # 输出&#xff1a;1 4 5# 取出第三个元素 6third_element t[2] 7pr…

数据流通环节如何规避安全风险

由于参与数据流通与交易的数据要素资源通常是经过组织加工的高质量数据集&#xff0c;甚至可能涉及国家核心战略利益&#xff0c;一旦发生针对数据流通环节的恶意事件&#xff0c;将造成较大负面影响&#xff0c;对数据要素市场的价值激活造成潜在威胁。具体来说&#xff0c;数…

【面试题】Golang之互斥锁与读写锁(第七篇)

在Go语言&#xff08;Golang&#xff09;中&#xff0c;锁是用来实现并发控制的一种机制&#xff0c;它可以帮助多个goroutine安全地访问共享资源&#xff0c;防止数据竞争和条件竞争的发生。Go标准库提供了多种锁的实现&#xff0c;其中最常见和最基本的包括互斥锁&#xff08…

好用的接口文档swagger

本篇文章记录怎么给我们的后端项目整一个好用的接口文档 这个东西好像叫什么swagger吧 1. 依赖引入&#xff1a; <dependency><groupId>com.github.xiaoymin</groupId><artifactId>knife4j-spring-boot-starter</artifactId></dependency>…

IP转接服务的重要性及其应用

在现今互联网高度发达的时代&#xff0c;IP转接服务的重要性日益凸显。对于家庭和企业而言&#xff0c;它不仅是连接互联网的桥梁&#xff0c;更是确保网络稳定、高效运行的关键。本文将深入探讨IP转接服务的核心意义及其在互联网世界中的应用。 IP转接服务&#xff0c;简而言之…

记录|C# winform布局学习

目录 前言一、自适应布局Step1. 添加AutoAdaptWindowsSize类Step2. Form中引用Step3. 创建SizeChanged事件函数Step4. 在Fram.Disiger中添加 更新时间 前言 参考视频&#xff1a; C#5分钟winform快速自适应布局 参考文章&#xff1a; 其他参考&#xff1a; 写这篇文章&#xff…

在 Java 中,怎样设计一个可扩展且易于维护的微服务架构?

在Java中设计一个可扩展且易于维护的微服务架构&#xff0c;可以考虑以下几个方面&#xff1a; 模块化设计&#xff1a;将应用拆分为多个小的、独立的模块&#xff0c;每个模块负责处理特定的业务逻辑。每个模块可以独立开发、测试和部署&#xff0c;增加或替换模块时不会影响其…

AI学习指南机器学习篇-t-分布领域嵌入(t-Distributed Stochastic Neighbor Embedding,t-SNE)简介

AI学习指南机器学习篇-t-分布领域嵌入&#xff08;t-Distributed Stochastic Neighbor Embedding&#xff0c;t-SNE&#xff09;简介 导言 在机器学习中&#xff0c;高维数据的降维和可视化一直是一个重要的问题。高维数据往往难以理解和分析&#xff0c;而且直接在高维空间中…

C#知识|账号管理系统-修改账号按钮功能的实现

哈喽,你好啊,我是雷工! 前边学习了通过选择条件查询账号的功能: 《提交查询按钮事件的编写》 本节继续学习练习C#,今天练习修改账号的功能实现。 以下为学习笔记。 01 实现功能 ①:从查询到的账号中,选择某一账号,然后点击【修改账号】按钮,将选中的信息获取显示到…