1461521844lijin / trt_yolo_video_pipeline

TensorRT+YOLO系列的 多路 多卡 多实例 并行视频分析处理案例
MIT License
222 stars 43 forks source link

检测后加多个分类模型 #16

Closed Hongyuan-Liu closed 2 months ago

Hongyuan-Liu commented 4 months ago

你好,对于检测算法后面跟着多个分类算法,该如何优雅、高效的实现,比如某一帧的检测结果,有人、手机、安全帽三种类别的目标,其中人目标不进行分类,手机、安全帽目标需要分别用两个分类算法进行二分类,以再次确认,减少误报。也就是某一帧首先要经过检测模型,然后再进行抠图,分别送入两个分类模型,最终该帧的结果是分类结果与检测结果综合判断的结果,这样的流程该如何优雅、高效、并且通用的实现呢。

1461521844lijin commented 4 months ago

你好,对于检测算法后面跟着多个分类算法,该如何优雅、高效的实现,比如某一帧的检测结果,有人、手机、安全帽三种类别的目标,其中人目标不进行分类,手机、安全帽目标需要分别用两个分类算法进行二分类,以再次确认,减少误报。也就是某一帧首先要经过检测模型,然后再进行抠图,分别送入两个分类模型,最终该帧的结果是分类结果与检测结果综合判断的结果,这样的流程该如何优雅、高效、并且通用的实现呢。

我能想到的就是检测模型和分类模型串行构建流水线,检测模型后面跟着两个并行的分类模型,将手机安全帽数据分别送入另外两个分类模型中进行并行处理,结果汇总时需要注意数据同步问题。 Snipaste_2024-05-19_10-39-18

Hongyuan-Liu commented 3 months ago

期待大佬后续多级模型实现