Cara menggunakan Python untuk memesongkan imej

Cara menggunakan Python untuk memesongkan gambar

Pengenalan:
Penyelewengan gambar ialah teknik pemprosesan imej biasa yang boleh mengubah bentuk dan ubah bentuk imej. Python mempunyai perpustakaan pemprosesan imej yang berkuasa yang menjadikannya sangat mudah untuk mencapai kesan herotan. Artikel ini akan memperkenalkan cara menggunakan Python untuk merealisasikan kesan herotan imej, dan melampirkan contoh kod.

1. Pasang perpustakaan yang diperlukan
Sebelum kita mula, kita perlu memasang beberapa perpustakaan Python. Di sini kami menggunakan perpustakaan Bantal untuk memproses imej dan perpustakaan NumPy untuk memproses data imej. Mula-mula kita perlu memasang dua perpustakaan ini:

pip install pillow
pip install numpy

Selepas pemasangan selesai, kita boleh mula menulis kod.

2. Import perpustakaan yang diperlukan
Mula-mula, kita perlu mengimport perpustakaan yang diperlukan:

from PIL import Image
import numpy as np

3. Muat dan tukar imej
Sebelum memulakan pemprosesan imej, kita perlu memuatkan dan menukar imej kepada tatasusunan NumPy. Kod di bawah menunjukkan cara memuatkan imej dan menukarnya kepada tatasusunan NumPy:

# 加载图像
image = Image.open('input.jpg')

# 转换为NumPy数组
image_array = np.array(image)

4. Cipta kesan herotan
Terdapat banyak cara untuk mencapai kesan herotan imej, yang paling biasa ialah menggunakan kesan fisheye. Kod berikut menunjukkan cara untuk mencapai kesan fisheye:

# 创建鱼眼效果
def create_fisheye(image_array, strength):
    height, width, channels = image_array.shape
    cy, cx = height // 2, width // 2
    radius = min(cy, cx)

    result = np.zeros_like(image_array)

    for y in range(height):
        for x in range(width):
            dy = y - cy
            dx = x - cx
            distance = np.sqrt(dx * dx + dy * dy)
            if distance < radius:
                theta = np.arctan2(dy, dx)
                r = distance / radius
                r = r ** strength * radius
                nx = cx + r * np.cos(theta)
                ny = cy + r * np.sin(theta)
                result[y, x] = image_array[int(ny), int(nx)]

    return result

Dalam kod di atas, kita mula-mula mengira koordinat tengah (cx, cy) dan jejari imej, dan kemudian melintasi setiap piksel imej untuk mengira piksel semasa ke pusat Jarak jarak titik. Untuk titik yang jaraknya lebih kecil daripada jejari, kami mengira kedudukan (nx, ny) piksel baharu berdasarkan jarak dan nilai kekuatan, dan kemudian menetapkan piksel pada kedudukan sepadan imej asal kepada piksel baharu. Akhirnya kembalikan tatasusunan imej baharu.

5. Simpan keputusan dan paparkannya
Langkah terakhir ialah menyimpan keputusan dan memaparkannya. Kod di bawah menunjukkan cara untuk menyimpan hasil dan memaparkan imej menggunakan perpustakaan Matplotlib:

# 创建扭曲效果
distorted_image_array = create_fisheye(image_array, strength=0.5)

# 转换为PIL图像
distorted_image = Image.fromarray(distorted_image_array)

# 保存结果
distorted_image.save('distorted.jpg')

# 展示图像
import matplotlib.pyplot as plt

fig, axes = plt.subplots(1, 2)

axes[0].imshow(image)
axes[0].set_title('Original Image')
axes[0].axis('off')

axes[1].imshow(distorted_image)
axes[1].set_title('Distorted Image')
axes[1].axis('off')

plt.show()

Dalam kod di atas, kami mula-mula memanggil fungsi create_fisheye untuk mencipta kesan herotan. Kami kemudian menukar tatasusunan imej yang melencong menjadi imej PIL dan menyimpannya secara tempatan. Akhir sekali, kami menggunakan perpustakaan Matplotlib untuk memaparkan imej asal dan imej terherot.

6. Ringkasan
Artikel ini memperkenalkan cara menggunakan Python untuk memesongkan imej dan melampirkan contoh kod yang sepadan. Dengan mempelajari artikel ini, pembaca boleh menguasai cara menggunakan Python untuk pemprosesan imej dan mengembangkan lagi kemahiran mereka dalam pemprosesan imej.

Atas ialah kandungan terperinci Cara menggunakan Python untuk memesongkan imej. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!