帧预处理从 26.5ms 降至 1.7ms(15.8×),端到端 52.4 → 16.0ms。惰性导入 cv2、 无 cv2 自动回退 float 路径;非 uint8/非 3 通道/尺寸不匹配输入走 PIL 转换 与 float 路径语义一致(7 类输入实测 max_diff=0)。OpenCV 5.0 仅支持单通道 LUT,逐通道应用后 merge。依赖加入 requirements.txt。
13 lines
411 B
Plaintext
13 lines
411 B
Plaintext
# Core
|
|
torch>=2.0,<3.0
|
|
torchvision>=0.15,<1.0
|
|
numpy>=1.24,<3.0
|
|
Pillow>=9.0,<14.0 # torchvision 依赖,上限放宽避免与 torch 源安装冲突
|
|
|
|
# 推理(Hailo 部署):帧归一化+量化合并查表(hailo_infer.py cv2.LUT 快路径,
|
|
# 无 cv2 自动回退 float 路径——本依赖缺失仅损失 3.3× 端到端加速)
|
|
opencv-python-headless>=4.9,<6.0
|
|
|
|
# Training utilities
|
|
tqdm>=4.60,<5.0
|