研究Detectron2中的在线数据增强的源码,并显示数据增强后的图像时,遇到
TypeError: Expected cv::UMat for argument 'img’的问题
显然,问题要求是输入一个uint8的Mat数据,而博主因为数据增强后,明显有部分值不是uint8了
博主解决方案:
先将图像 clip到[0, 255]之间,然后再将数据格式转换为 uint8格式
image = np.clip(image, 0, 255).astype(np.uint8)
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)