天台上的鱼 2022-09-06 17:14 采纳率: 100%
浏览 154
已结题

yolov5检测结果输出

使用yolov5对视频流进行检测时,可以同时输出检测结果视频和没有检测的原始视频吗?就相当于一个输入,两个输出。

  • 写回答

2条回答 默认 最新

  • 万里鹏程转瞬至 领域专家: 人工智能技术领域 2022-09-07 15:07
    关注

    可以参考这样实现,给模型在套一层模型

    class WarpModel(nn.Layer):
        def __init__(self,model):
            super(WarpModel, self).__init__()
            self.model = model
        @paddle.jit.to_static
        def forward(self, x):
            result= self.model(x)
            return x,result
    model.eval()
    warp_model=WarpModel(model)
    
    
    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(1条)

报告相同问题?

问题事件

  • 系统已结题 9月20日
  • 已采纳回答 9月12日
  • 创建了问题 9月6日