首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >要解压的值太多(读取图像形状时)

要解压的值太多(读取图像形状时)
EN

Stack Overflow用户
提问于 2021-04-10 22:06:07
回答 2查看 230关注 0票数 0

我正在尝试运行以下代码片段:

代码语言:javascript
运行
复制
from scipy.stats import wasserstein_distance
from imageio import imread
import numpy as np

def get_histogram(img):
  h, w = img.shape
  hist = [0.0] * 256
  for i in range(h):
    for j in range(w):
      hist[img[i, j]] += 1
  return np.array(hist) / (h * w)

a = imread("./IMG_4835.jpg")
b = imread("./IMG_4836.jpg")
a_hist = get_histogram(a)
b_hist = get_histogram(b)
dist = wasserstein_distance(a_hist, b_hist)
print(dist)

但我在以下位置得到一个错误:

代码语言:javascript
运行
复制
h, w = img.shape
b = imread('b.jpg', mode='L')
代码语言:javascript
运行
复制
ValueError: too many values to unpack (expected 2)

使用的原始代码:

代码语言:javascript
运行
复制
from scipy.ndimage import imread

来读取图像文件,但是因为我无法导入它,所以我使用了来自另一个库的imread。这可能与错误有关吗?

EN

回答 2

Stack Overflow用户

回答已采纳

发布于 2021-04-10 22:15:28

h,w = img.shape[:2]应该可以解决这个问题。

票数 2
EN

Stack Overflow用户

发布于 2021-07-27 10:16:22

RGB图像有3个通道,您的代码只能使用2个通道。您可以将您的图像转换为灰色:

代码语言:javascript
运行
复制
from skimage.color import rgb2gray
from skimage import img_as_ubyte
img = img_as_ubyte(rgb2gray(img))

或者,如果您不关心正确的RGB2灰色:(参见https://e2eml.school/convert_rgb_to_grayscale.html)

代码语言:javascript
运行
复制
img = np.mean(img, axis=2).astype(np.uint8)

但看起来你是在重新发明一个轮子。要使用直方图,请执行以下操作:

代码语言:javascript
运行
复制
a_hist, _ = np.histogram(a, bins=256)
b_hist, _ = np.histogram(b, bins=256)
票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/67035263

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档