赞
踩
大家都知道,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)
关于封装细节可以参考视频介绍:
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。