MonumentDetection/ContinualLearningFastAPI
0
1a
2 -f � @ s� d dl mZmZmZmZ d dlmZ d dlmZ d dl m3Z4 d dl m Z d dl
Z
d dlZd dlmZ d dlmZ d d lmZmZmZmZ d dlZd dlZd d5lmZ d dlZd dl m Z d d l m Z d dl Z!d dl"m#Z$ d dl%m&Z& e� Z'e'j(d e6d
d�d
d� e'j(de7dd�dd� e'j(de8dd�dd� edd�Z)da*dZ+G dd� de �Z,e'�-d�ed�dd��Z.e'j-ded �ed�d d ��Z/e'�0d �eed �d!d"��Z1e'�0d#�ed$�ed$�fee2e2d%�d&d'��Z3dS )(� )�FastAPI�Request�Form�9UploadFile)�Jinja2Templates)� HTMLResponse)� StaticFiles)� BaseModelN)�FigureCanvasAgg)�Path)�fasterrcnn_resnet50_fpn� filter_pred�classes�CLASSES)�datetime)�Imagez/staticZstatic)� directory)�namez/mediaZmediaz/output�output� templatesc @ s e Zd ZU eed< dS )� ImageForm�imageN)�__name__�10__module__� __qualname__r �__annotations__� r r �$C:\Users\DELL\fastapiProject\main.pyr s 11r �/��requestc � s t �dd| i�S )Nz welcome.htmlr �r �TemplateResponser r r r � monuments" s r# z/index/)Zresponse_classc � s t �dd| i�S )Nz12index.htmlr r! r r r r �index' s r$ z/upload/)r �filec � s| |� � I d H at�t�t��}|jdkr2|�d�}t�� }|j|dd� t �13|� � �� d�}|�
d�d }t�d| |d��S ) N�RGB�JPEG��format�utf-8�14r z upload.html)r �
image_content)�read�15imageBytesr �open�io�BytesIO�mode�convert�save�base64� b64encode�getvalue�decode�splitr r" )r r% r Z image_bytes�contentsr r r �upload8 s 1617r; z /predict/.)r �model�subsetc
� s z(t �t�t��}|jdkr&|�d�}W n2 tyZ } ztd|� �� W Y d }~n18d }~0 0 |�rt �19t � � g�}|�d�}||�}|� d�}t
j�t�}t|� d�� |dv r�|dv r�t|� t
j�|d||� d��}td d20�}|�tj|t�tj�� r�d nd �d
�� |�tj�� �r d nd � |�� t�� � ||�} W d � n1 �sT0 Y t | �}21|22d d � � � � } |23d d � � � � } |24d d � � � � }
t � |�}t!j"dd�\}}|�#|� |�$d� |�%d� t&| �D ]�\}}t'j(|d |d f|d |d |d |d dddd�}|�)|� |j*|d |d d t+t,| | � � dd t-d d dd d �d!� �q�t�� }t.t!�/� �}|�0|� |�1� }t �t�|��}|�d�}t�� }|j2|d"d#� t3�4|�1� ��5d$�}|�6d%�d }t7�8d&| |t9|
d d�d'��S d(d)iS )*Nr& zError: r z hey there)Zsubset1Zsubset2Zsubset3Zsubsset4)Z joint_modelZ25meta_modelZ26MNAD_modelz27Base Modelz.pth� )Z num_classes�cuda�cpu)Z map_location�boxes�labels�scores)�28 � )Zfigsize�offZOriginal� � � �r�none)Z linewidth� edgecolor� facecolorrE rD �whiteg�������?z
round,pad=0.2)rM �alpharL Zboxstyle)�colorZfontsizeZbboxr' r( r* r+ z predict.html)r r, Zconfidence_scoreZdetailzInvalid model specified.):r r/ r0 r1 r. r2 r3 � Exception�print�29transformsZComposeZToTensorZ unsqueeze�os�path�dirname�__file__�joinr Zload_state_dict�torch�loadZdevicer? Z is_available�to�evalZno_gradr
r@ �numpy�npZarray�pltZsubplotsZimshowZaxisZ set_title� enumerate�patchesZ RectangleZ add_patch�textr �int�dictr30 ZgcfZ print_pngr7 r4 r5 r6 r8 r9 r r" �round)r r< r= r �eZ transformZ31img_tensor�base_dirZ32model_pathZ predictionsZoutputsrA rB rC Z original_npZfigZaxs�jZboxZrect�bufferZcanvas�33bytes_dataZ image_rgbr r: r r r �predictG sn 34$�3536 37(38(394041420�43$�44 4546 rk )4Zfastapir r r r Zfastapi.templatingr Zfastapi.responsesr Zfastapi.staticfilesr Zpydanticr r0 r5 Z matplotlib.backends.backend_aggr47 �pathlibr Z"monuments.faster_models.fasterrcnnr r
r r rT �shutilr rY Ztorchvision.transformsrS ZPILr r] r^ Zmatplotlib.pyplotZpyplotr_ Zmatplotlib.patchesra �appZmountr r. r< r �getr# r$ Zpostr; �strrk r r r r �<module> sD 48