大家都知道,yolov7目前支持图像分类、目标检测和实例分割,而实例分割一般都是用predict.py去预测图片这个如果要引入pyqt5或着嵌入其他python模块有点麻烦,因此需要封装一个类去预测图像,这样我们可以轻松完成各种实例分割任务,封装好以后我们预测图片可以这么调用:
from Yolov7SegManager import *
if __name__ == '__main__':
ysm = Yolov7SegManager()
img = cv2.imread('data/bus.jpg')
mask, result_list = ysm.inference_image2(img, True)
print(result_list)
cv2.imshow('result', mask)
cv2.waitKey(0)
关于封装细节可以参考视频介绍:
将yolov7封装成一个类几行代码完成语义分割任务_哔哩哔哩_bilibili这个是使用yolov7实例分割框架封装一个类,只用几句代码即可完成实例分割任务。由于本人封装的风格,可以看出和yolov5、yolov8实例分割的类差不多。总之言简意赅,方便二次开发。, 视频播放量 96、弹幕量 0、点赞数 1、投硬币枚数 0、收藏人数 1、转发人数 0, 视频作者 未来自主研究中心, 作者简介 未来自主研究中心,相关视频:将yolov8封装成一个类几行代码完成语义分割任务,将yolov5-6.2封装成一个类几行代码完成语义分割任务,[课程][原创]yolov8检测封装成类调用几句代码完成目标检测任务课程导论,使用C#部署yolov7的tensorrt模型,将yoloair框架封装成一个类几行代码实现目标检测任务,使用yolov7-segment进行实例分割视频演示,C#实现全网yolov7目前最快winform目标检测,基于yolov8+deepsort实现目标追踪视频演示,[课程][原创]yolov7检测封装成类调用几句代码完成目标检测任务课程导论,yolox+deepsort+pyqt5实现目标追踪结果演示https://www.bilibili.com/video/BV1Fz4y1H7py/?vd_source=989ae2b903ea1b5acebbe2c4c4a635ee