可用的目标
C4S 训练的目标检测模型,onnx 模型共有10类目标:
- 人
- face
- 机动车
- license plate
- 自行车
- 摩托车
- 枪
- 刀
- 火
- 烟
MobileDet 模型共有4类目标:
- 人
- face
- 机动车
- license plate
face 和 license plate 用于获取人/车正面质量尽量更好的抓拍,其中自行车、摩托车的数据集偏小,可能有场景无法正确识别;刀、枪、火、烟的数据集同样偏小,对它们的检测仅供体验,不可直接用于生产性部署。
模型适配
用 C4S 源码构建的镜像在已测试适配 Google EdgeTpu 的 MobileDet 模型、nVIDIA TensorRT 的 Yolov7 模型、Yolo-Nas ONNX 模型和人车ReID模型。您也可以用卷挂载使用自己的模型:
如果模型配置与默认值不同,还需要更新模型配置。