主页 / ubuntu / 问题 / 1091493
Accepted
Mona Jalal
Asked: 2018-11-10 10:06:45 +0800 CST

将 1 通道图像转换为 3 通道图像

  • 772

我有一个只有 1 个通道的灰度图像。我正在使用一个期望所有图像都具有三个通道的代码。如何使用一些 Linux 命令将我的图像转换为 3 通道图像？

这是我得到的错误，我不想更改代码，而是想更改图像：

Traceback (most recent call last):
  File "get_img_vec.py", line 22, in <module>
    nd_arr = img2vec.get_vec(img)
  File "../img_to_vec.py", line 43, in get_vec
    h_x = self.model(image)
  File "/scratch/sjn-p3/anaconda/anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in __call__
    result = self.forward(*input, **kwargs)
  File "/scratch/sjn-p3/anaconda/anaconda3/lib/python3.6/site-packages/torchvision-0.2.1-py3.6.egg/torchvision/models/resnet.py", line 139, in forward
  File "/scratch/sjn-p3/anaconda/anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in __call__
    result = self.forward(*input, **kwargs)
  File "/scratch/sjn-p3/anaconda/anaconda3/lib/python3.6/site-packages/torch/nn/modules/conv.py", line 301, in forward
    self.padding, self.dilation, self.groups)
RuntimeError: Given groups=1, weight of size [64, 3, 7, 7], expected input[1, 1, 224, 224] to have 3 channels, but got 1 channels instead

图片：

$ identify 10524.jpg
0524.jpg JPEG 1050x550 1050x550+0+0 8-bit PseudoClass 256c 80.7KB 0.000u 0:00.000

这是图像：

一张脸的黑白照片

image-processing convert

2 个回答

  1. Best Answer

    以下 Python 代码有效：

    import cv2
import numpy as np
img = cv2.imread('10524.jpg')
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
img2 = np.zeros_like(img)
img2[:,:,0] = gray
img2[:,:,1] = gray
img2[:,:,2] = gray
cv2.imwrite('10524.jpg', img2)
    • 3

    1. 如果需要，安装 ImageMagick：

      sudo apt install imagemagick

    （您还可以按照本指南在 Ubuntu 18.04 上从源代码安装最新版本）

    1. 要分离图像通道，请运行：

        convert rose: -channel R -separate separate_red.gif
  convert rose: -channel G -separate separate_green.gif
  convert rose: -channel B -separate separate_blue.gif

      （按照本指南）

      或组合 RGB 图像通道运行：

      convert  separate_red.gif separate_green.gif separate_blue.gif \
         -combine -set colorspace sRGB rose_combined.gif

      （按照本指南）

    这也适用于 JPG。您也可以尝试将相同灰度图像的副本与 3 个不同的通道组合在一起。

    • 1

相关问题