Xiilab-model/xaiva-kiosk-models

📋 Overview

This repository contains optimized AI models for XAIVA Kiosk Analysis System - an intelligent workplace safety and compliance monitoring solution. The models are specifically designed for real-time analysis in industrial environments.

🚀 Model Collection

🎯 Detection Models

• —`best_0327.engine` (131MB)
• —Type: Object Detection (TensorRT Optimized)
• —Purpose: Safety equipment detection (caps, masks, gloves, boots, etc.)
• —Input: RGB images
• —Output: Bounding boxes with class predictions

👤 Human Analysis Models

• —`yolo11n-seg.engine` (10.2MB)
• —Type: Instance Segmentation (TensorRT Optimized)
• —Purpose: Person segmentation for workspace analysis
• —Architecture: YOLOv11 Nano
• —Output: Pixel-level person masks

• —`yolo11n-pose.engine` (10.1MB)
• —Type: Pose Estimation (TensorRT Optimized)
• —Purpose: Human pose detection for safety compliance
• —Architecture: YOLOv11 Nano
• —Output: 17-point keypoint coordinates

🔍 Face Recognition Models

• —`facedet_fp16.trt` (10.3MB)
• —Type: Face Detection (TensorRT FP16)
• —Purpose: Real-time face detection in workplace
• —Optimization: Half-precision for speed
• —Output: Face bounding boxes with landmarks

• —`adaface_ir101.trt` (132MB)
• —Type: Face Recognition (TensorRT Optimized)
• —Purpose: Employee identification and access control
• —Architecture: AdaFace with ResNet-101 backbone
• —Output: 512-dimensional face embeddings

🛠️ Usage

Prerequisites

pip install torch torchvision
pip install ultralytics
pip install opencv-python
pip install tensorrt

🏗️ System Requirements

• —GPU: NVIDIA GPU with Compute Capability 6.0+
• —VRAM: 8GB+ recommended
• —CUDA: 11.0+
• —TensorRT: 8.0+
• —Python: 3.8+

📁 Model Files

models/
├── best_0327.engine          # Main detection model
├── yolo11n-seg.engine        # Person segmentation
├── yolo11n-pose.engine       # Pose estimation
├── facedet_fp16.trt          # Face detection
└── adaface_ir101.trt         # Face recognition

🔧 Integration

These models are designed to work together as part of the XAIVA Kiosk system:

1. 1.Face Detection → Face Recognition for employee identification
2. 2.Detection → Safety equipment compliance checking
3. 3.Segmentation → Workspace occupancy analysis
4. 4.Pose Estimation → Safety posture monitoring

🏢 About XIILAB

Developed by XIILAB - Advanced AI solutions for industrial automation and safety.