帮忙改正代码错误

Justheartyoung

代码：
import cv2
import numpy as np
# 1. 分割遥感影像为多块图幅
def split_image(image, block_size):
    height, width = image.shape[:2]
    block_images = []
    for i in range(0, height, block_size):
        for j in range(0, width, block_size):
            block = image[i:i+block_size, j:j+block_size]
            block_images.append(block)
    return block_images


# 2. 分别对每块图幅转为灰度图像
def convert_to_gray(image):
    gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
    return gray_image

# 3. 对各个图像进行边缘检测
# def edge_detection(image):
#     edges = cv2.Canny(image, 0.5, 2)
#     return edges

def edge_detection(image):
    sobel_x = cv2.Sobel(image, cv2.CV_64F, 1, 0, ksize=3)
    sobel_y = cv2.Sobel(image, cv2.CV_64F, 0, 1, ksize=3)
    edges = cv2.magnitude(sobel_x, sobel_y)
    edges = cv2.convertScaleAbs(edges)
    return edges



# 4. 消除噪声干扰
def remove_noise(image):
    denoised_image = cv2.medianBlur(image, 5)
    return denoised_image

# 5. 将各个图幅合并为一个图幅
#


def merge_images(images, original_image_shape):
    merged_image = np.zeros(original_image_shape, dtype=np.uint8)
    block_size = images[0].shape[0]  # assume all blocks have the same size
    rows = original_image_shape[0] // block_size
    cols = original_image_shape[1] // block_size
    for i, image in enumerate(images):
        x = (i % cols) * block_size
        y = (i // cols) * block_size
        merged_image[y:y+image.shape[0], x:x+image.shape[1], :] = cv2.cvtColor(image, cv2.COLOR_GRAY2BGR)  # 转换为彩色图像
    return merged_image
# 6. 导出结果
def export_result(image, file_path):
    cv2.imwrite(file_path, image)

# 示例代码
image_path = r"C:\Users\WINDOWS\Desktop\taiyuan\python\dune\dune.tif"
block_size = 800

# 读取遥感影像
image = cv2.imread(image_path)

# 1. 分割遥感影像为多块图幅
block_images = split_image(image, block_size)

# 2. 分别对每块图幅转为灰度图像
gray_images = [convert_to_gray(block) for block in block_images]

# 3. 对各个图像进行边缘检测
edges_images = [edge_detection(gray) for gray in gray_images]

# 4. 消除噪声干扰
denoised_images = [remove_noise(edges) for edges in edges_images]

# 5. 将各个图幅合并为一个图幅
merged_image = merge_images(denoised_images, block_size)


# 6. 导出结果
export_result(merged_image, "output.jpg")

错误：
Traceback (most recent call last):
  File "C:\Users\WINDOWS\Desktop\taiyuan\python\segement.py", line 85, in <module>
    merged_image = merge_images(denoised_images, block_size)
  File "C:\Users\WINDOWS\Desktop\taiyuan\python\segement.py", line 54, in merge_images
    rows = original_image_shape[0] // block_size
TypeError: 'int' object is not subscriptable

isdkz

错误的原因是在merge_images函数中，original_image_shape被当作了一个整数进行处理，而不是一个元组。这是由于在调用merge_images函数时，block_size被传递为了第二个参数，而不是original_image_shape。

要解决这个问题，我们需要修改merge_images函数的定义和调用。下面是修改后的代码：

1. def merge_images(images, original_image_shape):
   
2.     merged_image = np.zeros(original_image_shape, dtype=np.uint8)
   
3.     block_size = images[0].shape[0]  # assume all blocks have the same size
   
4.     rows = original_image_shape[0] // block_size
   
5.     cols = original_image_shape[1] // block_size
   
6.     for i, image in enumerate(images):
   
7.         x = (i % cols) * block_size
   
8.         y = (i // cols) * block_size
   
9.         merged_image[y:y+image.shape[0], x:x+image.shape[1], :] = cv2.cvtColor(image, cv2.COLOR_GRAY2BGR)  # 转换为彩色图像
   
10.     return merged_image
    
11. 
    
12. # 示例代码
    
13. image_path = r"C:\Users\WINDOWS\Desktop\taiyuan\python\dune\dune.tif"
    
14. block_size = 800
    
15. 
    
16. # 读取遥感影像
    
17. image = cv2.imread(image_path)
    
18. 
    
19. # 1. 分割遥感影像为多块图幅
    
20. block_images = split_image(image, block_size)
    
21. 
    
22. # 2. 分别对每块图幅转为灰度图像
    
23. gray_images = [convert_to_gray(block) for block in block_images]
    
24. 
    
25. # 3. 对各个图像进行边缘检测
    
26. edges_images = [edge_detection(gray) for gray in gray_images]
    
27. 
    
28. # 4. 消除噪声干扰
    
29. denoised_images = [remove_noise(edges) for edges in edges_images]
    
30. 
    
31. # 5. 将各个图幅合并为一个图幅
    
32. merged_image = merge_images(denoised_images, image.shape)
    
33. 
    
34. # 6. 导出结果
    
35. export_result(merged_image, "output.jpg")
    

复制代码

通过将image.shape作为merge_images函数的第二个参数传递，我们可以解决这个错误。这样，original_image_shape将被正确地解析为一个元组，从而避免了TypeError异常。

陶远航

错误的原因是在函数merge_images中，original_image_shape被错误地用作一个整数进行了索引操作。修改代码如下：

1. import cv2
   
2. import numpy as np
   
3. 
   
4. # 1. 分割遥感影像为多块图幅
   
5. def split_image(image, block_size):
   
6.     height, width = image.shape[:2]
   
7.     block_images = []
   
8.     for i in range(0, height, block_size):
   
9.         for j in range(0, width, block_size):
   
10.             block = image[i:i+block_size, j:j+block_size]
    
11.             block_images.append(block)
    
12.     return block_images
    
13. 
    
14. # 2. 分别对每块图幅转为灰度图像
    
15. def convert_to_gray(image):
    
16.     gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
    
17.     return gray_image
    
18. 
    
19. # 3. 对各个图像进行边缘检测
    
20. def edge_detection(image):
    
21.     sobel_x = cv2.Sobel(image, cv2.CV_64F, 1, 0, ksize=3)
    
22.     sobel_y = cv2.Sobel(image, cv2.CV_64F, 0, 1, ksize=3)
    
23.     edges = cv2.magnitude(sobel_x, sobel_y)
    
24.     edges = cv2.convertScaleAbs(edges)
    
25.     return edges
    
26. 
    
27. # 4. 消除噪声干扰
    
28. def remove_noise(image):
    
29.     denoised_image = cv2.medianBlur(image, 5)
    
30.     return denoised_image
    
31. 
    
32. # 5. 将各个图幅合并为一个图幅
    
33. def merge_images(images, block_size):
    
34.     rows = len(images) * block_size
    
35.     cols = block_size
    
36.     merged_image = np.zeros((rows, cols), dtype=np.uint8)
    
37.     for i, image in enumerate(images):
    
38.         x = 0
    
39.         y = i * block_size
    
40.         merged_image[y:y+block_size, x:x+block_size] = image
    
41.     return merged_image
    
42. 
    
43. # 6. 导出结果
    
44. def export_result(image, file_path):
    
45.     cv2.imwrite(file_path, image)
    
46. 
    
47. # 示例代码
    
48. image_path = r"C:\Users\WINDOWS\Desktop\taiyuan\python\dune\dune.tif"
    
49. block_size = 800
    
50. 
    
51. # 读取遥感影像
    
52. image = cv2.imread(image_path)
    
53. 
    
54. # 1. 分割遥感影像为多块图幅
    
55. block_images = split_image(image, block_size)
    
56. 
    
57. # 2. 分别对每块图幅转为灰度图像
    
58. gray_images = [convert_to_gray(block) for block in block_images]
    
59. 
    
60. # 3. 对各个图像进行边缘检测
    
61. edges_images = [edge_detection(gray) for gray in gray_images]
    
62. 
    
63. # 4. 消除噪声干扰
    
64. denoised_images = [remove_noise(edges) for edges in edges_images]
    
65. 
    
66. # 5. 将各个图幅合并为一个图幅
    
67. merged_image = merge_images(denoised_images, block_size)
    
68. 
    
69. # 6. 导出结果
    
70. export_result(merged_image, "output.jpg")
    

复制代码

这样修改后，你应该能够正确地合并图像并导出结果。
如果问题已经解决，请设置最佳答案

学习编程中的Ben

你好，根据错误信息和代码分析，问题出在获取原始图像形状时，使用了错误的索引方式。为了修复此问题，请修改 merge_images 函数中的以下行：

1. merged_image = np.zeros(original_image_shape, dtype=np.uint8)

复制代码

改为：

1. merged_image = np.zeros((original_image_shape[0], original_image_shape[1], 3), dtype=np.uint8)

复制代码

同时，请将 export_result 函数中的以下行：

1. cv2.imwrite(file_path, image)

复制代码

改为：

1. cv2.imwrite(file_path, cv2.cvtColor(image, cv2.COLOR_BGR2RGB))

复制代码

这样可以确保导出的图像的颜色通道顺序正确。修改后的完整代码如下：

1. import cv2
2. import numpy as np
4. # 1. 分割遥感影像为多块图幅
5. def split_image(image, block_size):
6.     height, width = image.shape[:2]
7.     block_images = []
8.     for i in range(0, height, block_size):
9.         for j in range(0, width, block_size):
10.             block = image[i:i+block_size, j:j+block_size]
11.             block_images.append(block)
12.     return block_images
14. # 2. 分别对每块图幅转为灰度图像
15. def convert_to_gray(image):
16.     gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
17.     return gray_image
19. # 3. 对各个图像进行边缘检测
20. def edge_detection(image):
21.     sobel_x = cv2.Sobel(image, cv2.CV_64F, 1, 0, ksize=3)
22.     sobel_y = cv2.Sobel(image, cv2.CV_64F, 0, 1, ksize=3)
23.     edges = cv2.magnitude(sobel_x, sobel_y)
24.     edges = cv2.convertScaleAbs(edges)
25.     return edges
27. # 4. 消除噪声干扰
28. def remove_noise(image):
29.     denoised_image = cv2.medianBlur(image, 5)
30.     return denoised_image
32. # 5. 将各个图幅合并为一个图幅
33. def merge_images(images, original_image_shape):
34.     merged_image = np.zeros((original_image_shape[0], original_image_shape[1], 3), dtype=np.uint8)
35.     block_size = images[0].shape[0]  # assume all blocks have the same size
36.     rows = original_image_shape[0] // block_size
37.     cols = original_image_shape[1] // block_size
38.     for i, image in enumerate(images):
39.         x = (i % cols) * block_size
40.         y = (i // cols) * block_size
41.         merged_image[y:y+image.shape[0], x:x+image.shape[1], :] = cv2.cvtColor(image, cv2.COLOR_GRAY2BGR)  # 转换为彩色图像
42.     return merged_image
44. # 6. 导出结果
45. def export_result(image, file_path):
46.     cv2.imwrite(file_path, cv2.cvtColor(image, cv2.COLOR_BGR2RGB))
48. # 示例代码
49. image_path = r"C:\Users\WINDOWS\Desktop\taiyuan\python\dune\dune.tif"
50. block_size = 800
52. # 读取遥感影像
53. image = cv2.imread(image_path)
55. # 1. 分割遥感影像为多块图幅
56. block_images = split_image(image, block_size)
58. # 2. 分别对每块图幅转为灰度图像
59. gray_images = [convert_to_gray(block) for block in block_images]
61. # 3. 对各个图像进行边缘检测
62. edges_images = [edge_detection(gray) for gray in gray_images]
64. # 4. 消除噪声干扰
65. denoised_images = [remove_noise(edges) for edges in edges_images]
67. # 5. 将各个图幅合并为一个图幅
68. merged_image = merge_images(denoised_images, image.shape)
70. # 6. 导出结果
71. export_result(merged_image, "output.jpg")

复制代码

希望对你有帮助！如果还有其他问题，请随时提问。

如果回答对你有帮助，请给我一个最佳答案！