文章/答案/技术大牛

发布

社区首页 >问答首页 >我想用TensorFlow识别一个图像。我得到了一个形状错误，但我该怎么办？

问我想用TensorFlow识别一个图像。我得到了一个形状错误，但我该怎么办？
EN

Stack Overflow用户

提问于 2019-11-03 06:44:30

回答 1查看 59关注 0票数 0

GmdMiss_Folder = os.path.join(os.getcwd(), '..', 'Photo', 'GMD Miss')
GmdMiss_List = os.listdir(GmdMiss_Folder)

for i in range(0, len(GmdMiss_List)):
    Img = os.path.join(os.getcwd(), GmdMiss_Folder, GmdMiss_List[i])
    Img = cv2.imread(Img, cv2.IMREAD_GRAYSCALE)
    Img = np.array(Img)
    Img = cv2.resize(Img, dsize=(1920, 1080), interpolation=cv2.INTER_AREA)
    Img_Miss_List.append(Img)
i=0

while True:
    Img = Img_Miss_List[i]
    with tf.Session() as sess:
        graph = tf.Graph()
        with graph.as_default():
            with tf.name_scope("Convolution"):
                Img = Convolution(Img)
i += 1

...
The codes below are omitted
...

def Convolution(img):
    kernel = tf.Variable(tf.truncated_normal(shape=[180, 180, 3, 3], stddev=0.1))
    img = img.astype('float32')
    img = tf.nn.conv2d(np.expand_dims(img, 0), kernel, strides=[ 1, 15, 15, 1], padding='VALID')
    return img

错误是.。

ValueError:形状必须为4级，但对于输入形状为1,1080,1920,180,180，3，3的“卷积/卷积2D”(op：“Conv2D”)，则为3级。

arrays

python-3.x

tensorflow

python-3.6

reshape

回答 1

Stack Overflow用户

回答已采纳

发布于 2019-11-03 08:59:20

您的conv2D内核[filter_height, filter_width, in_channels, out_channels]期望输入图像有3个通道，它是一个具有[batch, in_height, in_width, in_channels]形状的4D张量，而您则以灰度格式加载输入图像。因此，您需要加载具有三个通道的彩色图像：

Img = cv2.imread(Img)  # By default cv2 load image in BGR format
Img = cv2.cvtColor(Img, cv2.COLOR_BGR2RGB)  # convert BGR to RGB format

希望能帮上忙。

票数 0

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/58677984

复制

相似问题

问我想用TensorFlow识别一个图像。我得到了一个形状错误，但我该怎么办？
EN

回答 1

Stack Overflow用户

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐

问我想用TensorFlow识别一个图像。我得到了一个形状错误，但我该怎么办？EN

回答 1

Stack Overflow用户

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐

问我想用TensorFlow识别一个图像。我得到了一个形状错误，但我该怎么办？
EN