Eye Detection | Luxonis

Our new model ZOO works with DepthAI V3. Find out more in our documentation.

Model Details

Model Description

This model is based on predefined (light variant) from LuxonisTrain and has been custom-trained to detect eyes. Specifically, the model is targeted to work best on crops of people faces.

Developed by: Luxonis
Shared by: Luxonis
Model type: Computer Vision
License: Apache 2.0
Resources for more information:

Training Details

Training Data

The dataset was used as the primary source of training images. In particular, we worked with the Menpo2D, 300W, and COFW subsets, restricted to semi-frontal images. Since this dataset provides keypoint annotations, we converted them into eye bounding boxes by selecting the relevant keypoints and generating axis-aligned bounding boxes around them. To improve generalization, we also incorporated the dataset from Roboflow Universe. In total, the final dataset comprised 19283 images.

Testing Details

Metrics

The model was trained on 512x512 resolution using letterbox resizing. On the testing set, which comprised of ~6k images, the model achieved a mean Average Precision (mAP) of 59.3%.

Technical Specifications

Input/Output Details

Input:
- Name: image
  - Info: NCHW RGB format with images normalized to a range of 0-1.
Output:
- Name: multiple (see NN archive)
  - Info: Unprocessed outputs of a multitude of detections

Model Architecture

Backbone: EfficientRep backbone
Neck: Rep-PAN neck
Head: Efficient decoupled head that is anchor-free

For more details, see the official documentation (light variat).

Throughput

Model variant: eye-detection:512x512

• Input shape: [1, 3, 512, 512] • Output shape: [[1, 6, 64, 64], [1, 6, 32, 32], [1, 6, 16, 16]]

• Params (M): 4.299 • GFLOPs: 3.579

Platform	Precision	Throughput (infs/sec)	Power Consumption (W)
RVC2	FP16	44.09	N/A
RVC4	FP16	419.75	3.73
RVC4	INT8	770.25	6.14

Benchmarked with , using 2 threads (and the DSP runtime in balanced mode for RVC4).
Parameters and FLOPs are obtained from the package.

Quantization

RVC4 version of the model was quantized using a custom dataset based on mixture of validation and test images. The dataset has 128 images.

Utilization

Models converted for RVC Platforms can be used for inference on OAK devices. DepthAI pipelines are used to define the information flow linking the device, inference model, and the output parser (as defined in model head(s)). Below, we present the most crucial utilization steps for the particular model. Please consult the docs for more information.

Install DAIv3 library:

pip install depthai

Define model:

model_description = dai.NNModelDescription(
    "luxonis/ppe-detection:640x640"
)

nn = pipeline.create(dai.node.DetectionNetwork).build(
    <CameraNode>, model_description
)

The model is automatically parsed by DAI and it outputs the message (bounding boxes, labels, and scores of the detected objects).

Get model output(s):

while pipeline.isRuning():
    nn_output: dai.ImgDetections = parser_output_queue.get()

Example

You can quickly run the model using our script. It automatically downloads the model, creates a DepthAI pipeline, runs the inference, and displays the results using our DepthAI visualizer tool. To try it out, run:

python3 main.py \
    --model luxonis/eye-detection:512x512

Eye detection model
License	Apache 2.0 Commercial use
Downloads	1572
Tasks	Object Detection
Model Types	ONNX

Model Variants

Name	Version	Available For	Created At	Deploy
		RVC2, RVC4	8 months ago