Python+wxPython实现将图片转换为草图

pip install opencv-python

import cv2
def convert_to_sketch(image_path):
    # 加载图像
    image = cv2.imread(image_path)
    # 将图像转换为灰度图像
    gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
    # 将灰度图像转换为边缘图像
    edges = cv2.Canny(gray_image, 30, 100)
    # 将边缘图像转换为彩色图像
    sketch = cv2.cvtColor(edges, cv2.COLOR_GRAY2BGR)
    return sketch
 

import wx
import cv2
def convert_to_sketch(image_path):
    # 加载图像
    image = cv2.imread(image_path)
    # 将图像转换为灰度图像
    gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
    # 将灰度图像转换为边缘图像
    edges = cv2.Canny(gray_image, 30, 100)
    # 将边缘图像转换为彩色图像
    sketch = cv2.cvtColor(edges, cv2.COLOR_GRAY2BGR)
    return sketch
def save_sketch(image_path, sketch):
    # 获取原始文件名和扩展名
    file_name, extension = image_path.rsplit('.', 1)
    # 构建保存的文件名
    save_path = f"{file_name}(sketch).{extension}"
    # 保存素描图像
    cv2.imwrite(save_path, sketch)
def main():
    # 选择图像文件
    app = wx.App()
    dialog = wx.FileDialog(None, "选择图像文件", wildcard="图像文件 (*.jpg;*.png)|*.jpg;*.png", style=wx.FD_OPEN | wx.FD_FILE_MUST_EXIST)
    if dialog.ShowModal() == wx.ID_OK:
        image_path = dialog.GetPath()
        # 将图像转换为素描
        sketch = convert_to_sketch(image_path)
        # 保存素描图像
        save_sketch(image_path, sketch)
        print("转换完成！")
    dialog.Destroy()
    app.MainLoop()
if __name__ == "__main__":
    main()