Question

我正在尝试训练神经网络模型，这是使图像居中的代码的一部分，问题是当我运行这段代码时 -

def centering_image(img):
  size = [256,256]

  img_size = img.shape[:2]

  # centering
  row = (size[1] - img_size[0]) // 2
  col = (size[0] - img_size[1]) // 2
  resized = np.zeros(list(size) + [img.shape[2]], dtype=np.uint8)
  resized[row:(row + img.shape[0]), col:(col + img.shape[1])] = img

  return resized


x = []
for i, file_path in enumerate(file_paths):
  #read image
  img = cv2.imread(file_path)
  img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)

  #resize
  if(img.shape[0] > img.shape[1]):
    tile_size = (int(img.shape[1]*256/img.shape[0]),256)
  else:
    tile_size = (256, int(img.shape[0]*256/img.shape[1]))

  #centering
  img = centering_image(cv2.resize(img, dsize=tile_size))

  #out put 224*224px 
  img = img[16:240, 16:240]
  x.append(img)

x = np.array(x)

我收到此错误 -

error                                     Traceback (most recent call 
last)
<ipython-input-11-2a14d86a9a00> in <module>()
 17     #read image
 18     img = cv2.imread(file_path)
---> 19     img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
 20 
 21     #resize

error: /io/opencv/modules/imgproc/src/color.cpp:9716: error: (-215) 
scn == 3 || scn == 4 in function cvtColor`

任何想法如何解决这个问题？提前致谢

Answer 1

~~看起来你加载的图像中有一个有四个通道。因此，从BGR到RBG的颜色转换不起作用，因为它需要三个通道。~~

~~您可以改为使用cv2.COLOR_BGRA2BGR标志。~~

错误表明cvtColor COLOR_BGR2RGB期望有3或4个频道的图片，但却有所不同。因此，您应该使用print img.shape仔细检查频道数。

scn == 3 || scn == 4在函数cvtColor中

1 个答案: