pytorch 实时物体检测
前言使用了pytorch自己提供的maskrcnn_resnet50_fpn,也就是说,不过是调用人家训练好的代码而已,当个玩具了,供大家参考。github项目其实主要的代码量在结果的显示上,并不是模型的调用上。这个是github项目链接.效果:实现思路模型调用,这个就不必细说了。使用opencv读图片然后转化成torch能吃进去的数据,input到模型中,得到返回值。根据模型的返回值,一一把label标注到原图片中然后显示图片下面的是主要代码:不能直接运行,只是思路,完整的