Bagaimana untuk Membuat Array NumPy sebagai Imej dalam FastAPI: Menyelesaikan Masalah Isu White Square?

Cara Memaparkan NumPy Array sebagai Imej dalam FastAPI

Masalah:

Walaupun mempunyai Tatasusunan NumPy yang mewakili imej, ia kelihatan sebagai segi empat sama putih apabila cuba memaparkannya.

Penyelesaian:

Pilihan 1: Kembalikan Imej sebagai Bait

Menggunakan kelas Respons tersuai, ikuti langkah berikut:

1. Muat imej daripada cakera atau tukar tatasusunan NumPy dalam memori kepada imej menggunakan PIL atau OpenCV.
2. Simpan imej ke dalam penimbal BytesIO menggunakan kaedah save() atau imencode() masing-masing.
3. Tetapkan pengepala Pelupusan Kandungan untuk menentukan paparan atau muat turun sebaris.
4. Kembalikan bait imej dalam Respons dengan media_type ditetapkan kepada format imej yang sesuai (cth., 'imej/png').

Pilihan 2: Kembalikan Imej sebagai Array NumPy yang dikodkan JSON

Pilihan ini tidak sesuai untuk memaparkan imej dalam penyemak imbas tetapi membenarkan anda mengekod tatasusunan NumPy sebagai JSON dan menyahkodnya semula menjadi imej pada bahagian klien.

1. Tukar Tatasusunan NumPy ke senarai menggunakan kaedah tolist().
2. Ekod senarai sebagai JSON dan kembalikannya dalam Respons.
3. Di sisi klien, nyahkod rentetan JSON kembali ke dalam tatasusunan NumPy dan gunakan PIL atau OpenCV untuk mencipta imej daripadanya.

Nota:

• Untuk kedua-dua pilihan, menetapkan pengepala Pelupusan Kandungan adalah penting untuk memaparkan imej dalam penyemak imbas.
• StreamingResponse tidak diperlukan jika tatasusunan NumPy dimuatkan ke dalam memori dan boleh dikembalikan dengan cekap dalam satu respons.

Atas ialah kandungan terperinci Bagaimana untuk Membuat Array NumPy sebagai Imej dalam FastAPI: Menyelesaikan Masalah Isu White Square?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!