近年、畳み込みニューラルネットワークを使用した顔検出で大きな進歩が見られました。多くの顔検出器は、顔の検出用に指定された設計を使用していますが、顔検出は一般的なオブジェクト検出タスクとして扱います。 YOLOv5オブジェクト検出器に基づく顔検出器を実装し、それをYOLO5Faceと呼びます。 5ポイントのランドマーク回帰ヘッドを追加し、翼損失関数を使用します。最高のパフォーマンスを実現するための大きなモデルから、組み込みデバイスまたはモバイルデバイスでのリアルタイム検出のための超小さなモデルまで、さまざまなモデルサイズの検出器を設計します。 WiderFaceデータセットの実験結果は、当社の顔検出器が、より複雑な指定顔検出器を超えて、ほぼすべてのEasy、Medium、およびHardサブセットで最先端のパフォーマンスを達成できることを示しています。コードはhttps://www.github.com/deepcam-cn/yolov5-faceで入手できます。
Tremendous progress has been made on face detection in recent years using convolutional neural networks. While many face detectors use designs designated for the detection of face, we treat face detection as a general object detection task. We implement a face detector based on YOLOv5 object detector and call it YOLO5Face. We add a five-point landmark regression head into it and use the Wing loss function. We design detectors with different model sizes, from a large model to achieve the best performance, to a super small model for real-time detection on an embedded or mobile device. Experiment results on the WiderFace dataset show that our face detectors can achieve state-of-the-art performance in almost all the Easy, Medium, and Hard subsets, exceeding the more complex designated face detectors. The code is available at https://www.github.com/deepcam-cn/yolov5-face.