关于Yolo的seg模式

Viewed 39

问题描述


我们在用yolo的seg模式,提取result的时候发现,结果和detect的输出格式是一样的,x y w h,我们修改了yolo,输出了mask,发现mask是一个没有分割的一整张图片,有什么办法,能获取到seg该有的掩码呢

1 Answers

YOLO模块中有一个变量self.masks就是分割的mask图像,你可以拿这个使用