Bagaimana Anda Memaparkan Susunan Imej NumPy dalam FastAPI?

Rendering NumPy Array dalam FastAPI

Manakala artikel "Bagaimana untuk mengembalikan tatasusunan numpy sebagai imej menggunakan FastAPI?" memberikan maklumat yang berguna, ia tidak langsung menangani isu paparan imej. Untuk membetulkannya, mari kita mendalami teknik asas:

Pilihan 1: Kembalikan Imej sebagai Bait

Kaedah ini memerlukan penukaran data imej kepada bait menggunakan perpustakaan seperti PIL atau OpenCV. Bait yang terhasil kemudiannya boleh disampaikan sebagai respons tersuai dengan jenis kandungan dan pengepala yang sesuai.

Menggunakan PIL:

<code class="python">from PIL import Image
import io

@app.get('/image', response_class=Response)
def get_image():
    im = Image.open('test.png')
    with io.BytesIO() as buf:
        im.save(buf, format='PNG')
        im_bytes = buf.getvalue()
    headers = {'Content-Disposition': 'inline; filename=&quot;test.png&quot;'}
    return Response(im_bytes, headers=headers, media_type='image/png')</code>

Menggunakan OpenCV:

<code class="python">import cv2

@app.get('/image', response_class=Response)
def get_image():
    arr = cv2.imread('test.png', cv2.IMREAD_UNCHANGED)
    success, im = cv2.imencode('.png', arr)
    headers = {'Content-Disposition': 'inline; filename=&quot;test.png&quot;'}
    return Response(im.tobytes(), headers=headers, media_type='image/png')</code>

Pilihan 2: Kembalikan Imej sebagai Array NumPy Dikodkan JSON

Walaupun pendekatan ini tidak disyorkan untuk memaparkan imej, ia boleh digunakan untuk menukar imej kepada dikodkan JSON tatasusunan numpy, yang kemudiannya boleh ditukar kembali kepada imej pada bahagian klien.

Menggunakan PIL:

<code class="python">from PIL import Image
import numpy as np

@app.get('/image')
def get_image():
    im = Image.open('test.png')
    arr = np.asarray(im)
    return json.dumps(arr.tolist())</code>

Menggunakan OpenCV:

<code class="python">import cv2

@app.get('/image')
def get_image():
    arr = cv2.imread('test.png', cv2.IMREAD_UNCHANGED)
    return json.dumps(arr.tolist())</code>

Untuk memaparkan imej menggunakan kaedah ini, anda perlu menukar bait yang diterima atau data yang dikodkan JSON kembali kepada format imej pada bahagian klien.

Atas ialah kandungan terperinci Bagaimana Anda Memaparkan Susunan Imej NumPy dalam FastAPI?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!