此代码改正import cv2 import numpy as np import matplotlib.pyplot as plt img = cv2.imread(r'E:\\postgraduate\\three\\DIP3E_Original_Images_CH03\\3.8.tif', 0) def gamma_transformation(img,gamma): height, width = img.shape dst = np.zeros((height, width), np.uint8) cv2.normalize(img, dst=dst, alpha=0, beta=1.0) dst = pow(img/255,gamma) * 255 dst =np.round(dst).astype(np.uint8) return dst image1 = gamma_transformation(img,0.6) image2 = gamma_transformation(img,0.4) image3 = gamma_transformation(img,0.3) fig = plt.figure(figsize=(8, 6)) vs1 = np.concatenate([img, image1,image2,image3]) vs2 = np.hstack((image2, image3)) result = np.vstack((vs1, vs2)) plt.show('3.8', result) cv2.waitKey(0) cv2.destroyAllWindows()
时间: 2023-12-05 22:06:41 浏览: 89
这段代码有几个问题需要修改:
1. `plt.show()` 函数没有传入参数,应该将图像数组作为参数传入,即 `plt.imshow(result)`。
2. `gamma_transformation()` 函数中的 `img` 参数与 `cv2.imread()` 中读入的图像变量名相同,会导致函数中的 `img` 变量与外部变量重名,建议将函数中的参数名改为 `src` 或其他不与外部变量名重复的名称。
3. 在 `gamma_transformation()` 函数中,计算 `dst` 的过程可以简化为 `dst = (img/255)**gamma * 255`,无需调用 `np.round(dst).astype(np.uint8)`。
4. 在使用 `cv2.imshow()` 显示图像时,应该传入窗口名称参数,如 `cv2.imshow('image', img)`。
修改后的代码如下:
```python
import cv2
import numpy as np
import matplotlib.pyplot as plt
img = cv2.imread(r'E:\\postgraduate\\three\\DIP3E_Original_Images_CH03\\3.8.tif', 0)
def gamma_transformation(src, gamma):
height, width = src.shape
dst = np.zeros((height, width), np.uint8)
cv2.normalize(src, dst=dst, alpha=0, beta=1.0)
dst = (src/255)**gamma * 255
dst = dst.astype(np.uint8)
return dst
image1 = gamma_transformation(img, 0.6)
image2 = gamma_transformation(img, 0.4)
image3 = gamma_transformation(img, 0.3)
fig = plt.figure(figsize=(8, 6))
vs1 = np.concatenate([img, image1, image2, image3], axis=1)
vs2 = np.concatenate([image2, image3], axis=1)
result = np.concatenate([vs1, vs2], axis=0)
plt.imshow(result)
plt.show()
cv2.imshow('image', img)
cv2.waitKey(0)
cv2.destroyAllWindows()
```
其中,`np.concatenate()` 函数第二个参数应该为 `axis=1` 或者 `axis=0`,表示沿着列或者行方向进行合并操作。
阅读全文