Layout of the output array img is incompatible with cv::Mat

最新推荐文章于 2024-08-23 10:35:38 发布

原创最新推荐文章于 2024-08-23 10:35:38 发布 · 509 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#opencv #计算机视觉 #python

在使用OpenCV读取并处理含有透明通道的彩色图像时，代码在进行点标记操作时遇到错误。通过检查源代码发现，问题在于没有对图像的透明通道进行深拷贝。修正方法是将`img[:,:,3]`改为`img[:,:,3].copy()`，确保在操作时不直接修改原图像。应用这个修改后，代码能成功地对图像进行点标记并保存结果。

读入的图像是包含透明通道的彩色图像，在对图像进行点标记的时候出现如下错误：
在这里插入图片描述
源代码：

 img = cv2.imread(os.path.join(args.cloth_dir, name), cv2.IMREAD_UNCHANGED)
 if img is None:
     print('read image failed!')
 elif img.shape[2] != 4:
     print('image has no alpha channel!')
 else:
     img_mask = img[:, :, 3]#源代码
     is_get, left_pt, right_pt = get_points(img_mask)
     if is_get:
         cv2.circle(img_mask, (left_pt[0], left_pt[1]), 5, (255, 255, 255), -1)
         cv2.circle(img_mask, (right_pt[0], right_pt[1]), 5, (255, 255, 255), -1)

     else:
         print(f'{name} get neckline points failed')
     cv2.imwrite(os.path.join(args.output_dir, name.split('.')[0] + '_mask.jpg'), img_mask)

对图像进行深拷贝后可以解决该问题：
修改后代码：

 img = cv2.imread(os.path.join(args.cloth_dir, name), cv2.IMREAD_UNCHANGED)
 if img is None:
     print('read image failed!')
 elif img.shape[2] != 4:
     print('image has no alpha channel!')
 else:
     img_mask = img[:, :, 3].copy()#修改代码
     is_get, left_pt, right_pt = get_points(img_mask)
     if is_get:
         cv2.circle(img_mask, (left_pt[0], left_pt[1]), 5, (255, 255, 255), -1)
         cv2.circle(img_mask, (right_pt[0], right_pt[1]), 5, (255, 255, 255), -1)

     else:
         print(f'{name} get neckline points failed')
     cv2.imwrite(os.path.join(args.output_dir, name.split('.')[0] + '_mask.jpg'), img_mask)