鱼C论坛

 找回密码
 立即注册
查看: 3265|回复: 4

帮忙找出代码报错原因。并在原代码上给出修改代码。谢谢

[复制链接]
发表于 2023-8-7 16:16:35 | 显示全部楼层 |阅读模式

马上注册,结交更多好友,享用更多功能^_^

您需要 登录 才可以下载或查看,没有账号?立即注册

x
代码:
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, 100, 200)
    return edges

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

# 5. 将各个图幅合并为一个图幅
def merge_images(images, block_size):
    rows = int(np.sqrt(len(images)))
    cols = rows
    merged_image = np.zeros((rows * block_size, cols * block_size), dtype=np.uint8)
    for i, image in enumerate(images):
        x = (i % cols) * block_size
        y = (i // cols) * block_size
        merged_image[y:y+block_size, x:x+block_size] = image
    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 = 100

# 读取遥感影像
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 64, in <module>
    merged_image = merge_images(denoised_images, block_size)
  File "C:\Users\WINDOWS\Desktop\taiyuan\python\segement.py", line 37, in merge_images
    merged_image[y:y+block_size, x:x+block_size] = image
ValueError: could not broadcast input array from shape (100,32) into shape (100,100)

Process finished with exit code 1
小甲鱼最新课程 -> https://ilovefishc.com
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

小黑屋|手机版|Archiver|鱼C工作室 ( 粤ICP备18085999号-1 | 粤公网安备 44051102000585号)

GMT+8, 2025-9-29 02:34

Powered by Discuz! X3.4

© 2001-2023 Discuz! Team.

快速回复 返回顶部 返回列表