这是一个facenet-pytorch的库，可以用于训练自己的人脸识别模型。

Last update: Jan 06, 2023

Related tags

Overview

Facenet：人脸识别模型在Pytorch当中的实现

性能情况

训练数据集	权值文件名称	测试数据集	输入图片大小	accuracy
CASIA-WebFace	facenet_mobilenet.pth	LFW	160x160	98.23%
CASIA-WebFace	facenet_inception_resnetv1.pth	LFW	160x160	98.78%

所需环境

pytorch==1.2.0

文件下载

已经训练好的facenet_mobilenet.pth和facenet_inception_resnetv1.pth可以在百度网盘下载。
链接: https://pan.baidu.com/s/1slUYdpskFpUX62WpJeLByA 提取码: fe1w

训练用的CASIA-WebFaces数据集以及评估用的LFW数据集可以在百度网盘下载。
链接: https://pan.baidu.com/s/1fhiHlylAFVoR43yfDbi4Ag 提取码: gkch

预测步骤

a、使用预训练权重

下载完库后解压，在model_data文件夹里已经有了facenet_mobilenet.pth，可直接运行predict.py输入：

img\1_001.jpg
img\1_002.jpg

也可以在百度网盘下载facenet_inception_resnetv1.pth，放入model_data，修改facenet.py文件的model_path后，输入：

img\1_001.jpg
img\1_002.jpg

b、使用自己训练的权重

按照训练步骤训练。
在facenet.py文件里面，在如下部分修改model_path和backbone使其对应训练好的文件；model_path对应logs文件夹下面的权值文件，backbone对应主干特征提取网络。

_defaults = {
    "model_path"    : "model_data/facenet_mobilenet.pth",
    "input_shape"   : (160, 160, 3),
    "backbone"      : "mobilenet",
    "cuda"          : True,
}

运行predict.py，输入

img\1_001.jpg
img\1_002.jpg

训练步骤

本文使用如下格式进行训练。

|-datasets
    |-people0
        |-123.jpg
        |-234.jpg
    |-people1
        |-345.jpg
        |-456.jpg
    |-...

下载好数据集，将训练用的CASIA-WebFaces数据集以及评估用的LFW数据集，解压后放在根目录。
在训练前利用txt_annotation.py文件生成对应的cls_train.txt。
利用train.py训练facenet模型，训练前，根据自己的需要选择backbone，model_path和backbone一定要对应。
运行train.py即可开始训练。

评估步骤

下载好评估数据集，将评估用的LFW数据集，解压后放在根目录
在eval_LFW.py设置使用的主干特征提取网络和网络权值。
运行eval_LFW.py来进行模型准确率评估。

Reference

https://github.com/davidsandberg/facenet
https://github.com/timesler/facenet-pytorch

Comments

训练过程经常遇到BrokenPipeError: [Errno 32] Broken pipe

Epoch 1/100: 100%|██████████| 583/583 [07:36<00:00, 1.07it/s, accuracy=0.89, lr=0.01, total_CE_loss=9.02, total_triple_loss=0.101]Traceback (most recent call last): File "/opt/vitis_ai/conda/envs/vitis-ai-optimizer_pytorch/lib/python3.7/multiprocessing/queues.py", line 242, in _feed send_bytes(obj) File "/opt/vitis_ai/conda/envs/vitis-ai-optimizer_pytorch/lib/python3.7/multiprocessing/connection.py", line 200, in send_bytes self._send_bytes(m[offset:offset + size]) File "/opt/vitis_ai/conda/envs/vitis-ai-optimizer_pytorch/lib/python3.7/multiprocessing/connection.py", line 404, in _send_bytes self._send(header + buf) File "/opt/vitis_ai/conda/envs/vitis-ai-optimizer_pytorch/lib/python3.7/multiprocessing/connection.py", line 368, in _send n = write(self._handle, buf) BrokenPipeError: [Errno 32] Broken pipe

opened by jia0511 1
lfw数据集处理有什么区别？精度目前97%

使用facenet_mobilenet.pth 在 LFW 数据集上，调整图片大小为 160x160 ，得到了0.97的精度，没有到| 98.23%，而在百度网盘提供的slfw数据上，精度可以到98%，但是我看网页上提供的数据图片大小是96*112，请问下，LFW处理上应用什么其他方法吗？

Test Epoch: [5888/6000 (96%)]: : 24it [00:32, 1.34s/it] Accuracy: 0.97383+-0.00675 Best_thresholds: 1.16000 Validation rate: 0.82100+-0.03127 @ FAR=0.00100

opened by xiaomujiang 1

Releases(v2.0)

v2.0(Mar 24, 2022)
仓库特点

支持多backbone训练。

支持训练时评估，通过train.py的lfw_eval_flag调整、

大量的注释，多个可调整参数。

summary.py文件用于观看网络结构。

使用softmax辅助训练

Source code(tar.gz)
Source code(zip)
v1.0(May 16, 2021)

Source code(tar.gz)
Source code(zip)
backbone_weights_of_inception_resnetv1.pth(86.31 MB)
backbone_weights_of_mobilenetv1.pth(16.25 MB)
facenet_inception_resnetv1.pth(92.42 MB)
facenet_mobilenet.pth(18.05 MB)

Owner

Bubbliiiing

GitHub Repository

Towards Understanding Quality Challenges of the Federated Learning: A First Look from the Lens of Robustness

FL Analysis This repository contains the code and results for the paper "Towards Understanding Quality Challenges of the Federated Learning: A First L

3 Oct 17, 2022

STYLER: Style Factor Modeling with Rapidity and Robustness via Speech Decomposition for Expressive and Controllable Neural Text to Speech

STYLER: Style Factor Modeling with Rapidity and Robustness via Speech Decomposition for Expressive and Controllable Neural Text to Speech Keon Lee, Ky

114 Dec 12, 2022

Interactive dimensionality reduction for large datasets

BlosSOM 🌼 BlosSOM is a graphical environment for running semi-supervised dimensionality reduction with EmbedSOM. You can use it to explore multidimen

19 Dec 14, 2022

An open-source, low-cost, image-based weed detection device for fallow scenarios.

Welcome to the OpenWeedLocator (OWL) project, an opensource hardware and software green-on-brown weed detector that uses entirely off-the-shelf compon

145 Jan 05, 2023

Official implementation of "An Image is Worth 16x16 Words, What is a Video Worth?" (2021 paper)

An Image is Worth 16x16 Words, What is a Video Worth? paper Official PyTorch Implementation Gilad Sharir, Asaf Noy, Lihi Zelnik-Manor DAMO Academy, Al

213 Nov 12, 2022

Facebook AI Research Sequence-to-Sequence Toolkit written in Python.

Fairseq(-py) is a sequence modeling toolkit that allows researchers and developers to train custom models for translation, summarization, language mod

20.5k Jan 08, 2023

Deep Anomaly Detection with Outlier Exposure (ICLR 2019)

Outlier Exposure This repository contains the essential code for the paper Deep Anomaly Detection with Outlier Exposure (ICLR 2019). Requires Python 3

464 Dec 27, 2022

Large-scale Hyperspectral Image Clustering Using Contrastive Learning, CIKM 21 Workshop

Spectral-spatial contrastive clustering (SSCC) Yaoming Cai, Yan Liu, Zijia Zhang, Zhihua Cai, and Xiaobo Liu, Large-scale Hyperspectral Image Clusteri

4 Nov 02, 2022

Demonstrates how to divide a DL model into multiple IR model files (division) and introduce a simplest way to implement a custom layer works with OpenVINO IR models.

Demonstration of OpenVINO techniques - Model-division and a simplest-way to support custom layers Description: Model Optimizer in Intel(r) OpenVINO(tm

12 Nov 09, 2022

这是一个facenet-pytorch的库，可以用于训练自己的人脸识别模型。

Related tags

Overview

Facenet：人脸识别模型在Pytorch当中的实现

目录

性能情况

所需环境

文件下载

预测步骤

a、使用预训练权重

b、使用自己训练的权重

训练步骤

评估步骤

Reference

You might also like...

Comments

训练过程经常遇到BrokenPipeError: [Errno 32] Broken pipe

lfw数据集处理有什么区别？ 精度目前97%

Releases(v2.0)

v2.0(Mar 24, 2022)

仓库特点

v1.0(May 16, 2021)

Owner

Bubbliiiing

Towards Understanding Quality Challenges of the Federated Learning: A First Look from the Lens of Robustness

STYLER: Style Factor Modeling with Rapidity and Robustness via Speech Decomposition for Expressive and Controllable Neural Text to Speech

Interactive dimensionality reduction for large datasets

An open-source, low-cost, image-based weed detection device for fallow scenarios.

Official implementation of "An Image is Worth 16x16 Words, What is a Video Worth?" (2021 paper)

Facebook AI Research Sequence-to-Sequence Toolkit written in Python.

Deep Anomaly Detection with Outlier Exposure (ICLR 2019)

Large-scale Hyperspectral Image Clustering Using Contrastive Learning, CIKM 21 Workshop

Demonstrates how to divide a DL model into multiple IR model files (division) and introduce a simplest way to implement a custom layer works with OpenVINO IR models.

The Instructed Glacier Model (IGM)

Poplar implementation of "Bundle Adjustment on a Graph Processor" (CVPR 2020)

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

Easy genetic ancestry predictions in Python

Implementation of light baking system for ray tracing based on Activision's UberBake

Code for the paper "Reinforced Active Learning for Image Segmentation"

ilpyt: imitation learning library with modular, baseline implementations in Pytorch

A collection of papers about Transformer in the field of medical image analysis.

Working demo of the Multi-class and Anomaly classification model using the CLIP feature space

The 3rd place solution for competition

NeRViS: Neural Re-rendering for Full-frame Video Stabilization

lfw数据集处理有什么区别？精度目前97%