test_hardware_detection.sh

#!/bin/bash

# RV1106 AiCam 硬件检测测试脚本
# 检测x86_64环境下的真实硬件支持

echo "🔍 RV1106 AiCam 硬件检测测试"
echo "=================================="

# 检测系统架构
echo "📋 系统信息:"
echo "架构: $(uname -m)"
echo "内核: $(uname -r)"
echo "发行版: $(lsb_release -d 2>/dev/null | cut -f2 || echo "未知")"

echo ""
echo "📹 摄像头设备检测:"

# 检测V4L2设备
if ls /dev/video* >/dev/null 2>&1; then
    echo "✅ 发现V4L2设备:"
    ls -la /dev/video*
    echo ""
    
    # 检测摄像头详细信息
    if command -v v4l2-ctl >/dev/null 2>&1; then
        echo "📷 摄像头详细信息:"
        v4l2-ctl --list-devices
        echo ""
        
        # 检测第一个摄像头的支持格式
        echo "🎥 /dev/video0 支持的格式:"
        v4l2-ctl --device=/dev/video0 --list-formats-ext 2>/dev/null | head -20
    else
        echo "⚠️  v4l2-ctl未安装，无法获取详细信息"
        echo "   安装命令: sudo apt install v4l-utils"
    fi
else
    echo "❌ 未发现V4L2摄像头设备"
fi

echo ""
echo "🎮 GPU和CUDA检测:"

# 检测NVIDIA GPU
if command -v nvidia-smi >/dev/null 2>&1; then
    echo "✅ 发现NVIDIA GPU:"
    nvidia-smi --query-gpu=name,driver_version,memory.total --format=csv,noheader,nounits
    echo ""
    
    # 检测CUDA版本
    if command -v nvcc >/dev/null 2>&1; then
        echo "✅ CUDA编译器版本:"
        nvcc --version | grep "release"
    else
        echo "⚠️  CUDA编译器未安装"
    fi
    
    # 检测CUDA运行时
    if [ -f "/usr/local/cuda/version.txt" ]; then
        echo "✅ CUDA运行时版本:"
        cat /usr/local/cuda/version.txt
    elif [ -f "/usr/local/cuda/version.json" ]; then
        echo "✅ CUDA运行时版本:"
        cat /usr/local/cuda/version.json | grep -o '"version":"[^"]*"'
    fi
    
    # 检测TensorRT
    echo ""
    echo "🧠 TensorRT检测:"
    if find /usr -name "libnvinfer.so*" 2>/dev/null | head -1; then
        echo "✅ 发现TensorRT库:"
        find /usr -name "libnvinfer.so*" 2>/dev/null
        
        # 尝试获取TensorRT版本
        if command -v dpkg >/dev/null 2>&1; then
            echo "📦 TensorRT包信息:"
            dpkg -l | grep tensorrt | head -5
        fi
    else
        echo "❌ 未发现TensorRT库"
        echo "   安装指南: https://developer.nvidia.com/tensorrt"
    fi
else
    echo "❌ 未发现NVIDIA GPU或驱动"
    echo "   请安装NVIDIA驱动和CUDA工具包"
fi

echo ""
echo "📚 OpenCV检测:"

# 检测OpenCV
if command -v pkg-config >/dev/null 2>&1; then
    if pkg-config --exists opencv4; then
        echo "✅ 发现OpenCV 4:"
        echo "   版本: $(pkg-config --modversion opencv4)"
        echo "   库路径: $(pkg-config --libs opencv4 | cut -d' ' -f1)"
    elif pkg-config --exists opencv; then
        echo "✅ 发现OpenCV:"
        echo "   版本: $(pkg-config --modversion opencv)"
        echo "   库路径: $(pkg-config --libs opencv | cut -d' ' -f1)"
    else
        echo "❌ 未发现OpenCV"
        echo "   安装命令: sudo apt install libopencv-dev"
    fi
else
    echo "⚠️  pkg-config未安装，无法检测OpenCV"
fi

echo ""
echo "🔧 编译环境检测:"

# 检测编译工具
echo "编译器:"
if command -v gcc >/dev/null 2>&1; then
    echo "✅ GCC: $(gcc --version | head -1)"
else
    echo "❌ GCC未安装"
fi

if command -v g++ >/dev/null 2>&1; then
    echo "✅ G++: $(g++ --version | head -1)"
else
    echo "❌ G++未安装"
fi

if command -v cmake >/dev/null 2>&1; then
    echo "✅ CMake: $(cmake --version | head -1)"
else
    echo "❌ CMake未安装"
fi

echo ""
echo "📊 推荐配置:"

# 生成推荐的硬件配置
cat > config/hardware_detected.conf << EOF
# 自动检测的硬件配置
# 生成时间: $(date)

PLATFORM=X86_64

[VIDEO_CAPTURE]
TYPE=V4L2
DEVICE=/dev/video0
WIDTH=1920
HEIGHT=1080
FPS=30
FORMAT=MJPG

[AI_INFERENCE]
EOF

if command -v nvidia-smi >/dev/null 2>&1 && find /usr -name "libnvinfer.so*" >/dev/null 2>&1; then
    cat >> config/hardware_detected.conf << EOF
TYPE=TENSORRT
MODEL_PATH=models/yolov5s.onnx
EOF
    echo "✅ 推荐使用TensorRT进行AI推理"
else
    cat >> config/hardware_detected.conf << EOF
TYPE=OPENCV_DNN
MODEL_PATH=models/yolov5s.onnx
EOF
    echo "⚠️  推荐使用OpenCV DNN进行AI推理（CPU）"
fi

cat >> config/hardware_detected.conf << EOF
INPUT_WIDTH=640
INPUT_HEIGHT=640
CONF_THRESHOLD=0.5
NMS_THRESHOLD=0.4
GPU_DEVICE_ID=0

[VIDEO_ENCODING]
EOF

if command -v nvidia-smi >/dev/null 2>&1; then
    cat >> config/hardware_detected.conf << EOF
TYPE=NVENC
EOF
    echo "✅ 推荐使用NVENC进行硬件编码"
else
    cat >> config/hardware_detected.conf << EOF
TYPE=FFMPEG
EOF
    echo "⚠️  推荐使用FFmpeg进行软件编码"
fi

cat >> config/hardware_detected.conf << EOF
CODEC=H264
BITRATE=4000
GOP_SIZE=30

[PERFORMANCE]
AI_THREADS=2
SKIP_FRAMES=2
MAX_QUEUE_SIZE=10
ENABLE_GPU_ACCELERATION=true

[DEBUG]
ENABLE_PERFORMANCE_MONITOR=true
SAVE_DETECTION_IMAGES=false
LOG_LEVEL=INFO
EOF

echo ""
echo "📁 配置文件已生成: config/hardware_detected.conf"

echo ""
echo "🎯 下一步操作:"
echo "1. 确保所有依赖已安装"
echo "2. 下载YOLOv5模型文件到 models/ 目录"
echo "3. 重新编译项目: cd build && make -j4"
echo "4. 使用检测到的配置运行系统"

echo ""
echo "✨ 硬件检测完成！"