我想尝试一个项目,我必须检测参加马拉松的人穿的运动服的品牌+型号,这个项目非常复杂,所以我的第一个方法是使用一个已经预先训练好的模型将绘制人的边界框,然后使用输出的绑定框作为另一个模型的输入,该模型将为我绘制衬衫、鞋子、短裤等的边界框,并按照此过程直到我得到品牌。然后是运动服的型号。从理论上讲,它非常简单,问题是,我很难弄清楚如何通过代码实现它,将一个模型的输出作为输入传递给另一个模型,互联网上有没有好的例子可以用代码来说明这种方法,我的意思是?
任何建议将不胜感激,谢谢。
我想尝试一个项目,我必须检测参加马拉松的人穿的运动服的品牌+型号,这个项目非常复杂,所以我的第一个方法是使用一个已经预先训练好的模型将绘制人的边界框,然后使用输出的绑定框作为另一个模型的输入,该模型将为我绘制衬衫、鞋子、短裤等的边界框,并按照此过程直到我得到品牌。然后是运动服的型号。从理论上讲,它非常简单,问题是,我很难弄清楚如何通过代码实现它,将一个模型的输出作为输入传递给另一个模型,互联网上有没有好的例子可以用代码来说明这种方法,我的意思是?
任何建议将不胜感激,谢谢。
我觉得你的问题有点难以理解。如果我理解正确,您知道如何获得一个人的边界框。使用这些坐标,您可以使用 open-cv 或 matplotlib 或任何图像处理库选择此矩形。如果您的第二个模型可以处理多分辨率或多种尺寸,您可以使用此选择作为输入。另一方面,如果您的第二个模型需要固定大小,您可以调整选择的大小并在必要时使用填充。