MonumentDetection/ContinualLearningFastAPI
0
1a
2 -�e� � @ s� d dl mZ d dlmZmZ d dlmZ ddlmZ d dl m3Z4 d dl m Z d dl m
Z
mZmZmZ d d lZd d lZd d lZd d lmZ d d5lmZ d d lmZmZ d d l m Z d d l m Z d d
� Z!dd� Z"dd� Z#dd� Z$d S )� ��render)� HttpResponse� JsonResponse)�loader� )�ImageUploadForm)�settings)�datetime)�fasterrcnn_resnet50_fpn� filter_pred�classes�CLASSESN)�6functional)�Image� ImageDrawc C s7 t | d�S )Nz8index.htmlr ��request� r �2C:\Users\DELL\MonumentDetection\monuments\views.py� monuments s r c C s9 t | d�S )Nz welcome.htmlr r r r r �index s r c C sh | j dkrNt| j| j�}|�� rd|�� |j}t|jj � t10| d||d��S nt� }t11| dd|i�S d S )N�POSTz upload.html)�form�12img_objectzimage_form.htmlr ) �methodr r �FILES�is_valid�save�instance�print�image�urlr )r r r r r r �upload s 13 r# c
C s� | j dk�r�| j�d�}tj�|�}tj�dd|�}t|� | j dkr\| j�d�}|| j d< |d k�r�t14� |�� d�}t
�t
�� g�}||�}|�d�}td � d15}td d �}|�tj|t�tj�� r�d
nd�d�� |�tj�� r�d
nd� |�� t�� � ||�} W d � n1 �s 0 Y t| �}16|17d d � � � � } |18d d � � � � } |19d d � � � � }
t| | |
� 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-�.� �/d$�}d%|� d&�}tj�t0j1|�}tj�t0j2d'|�}t �3|� t|� td(� t|� t4| d)d|i�S t5d*d+d,�S )-Nr �20image_path�media�images�GET�model�RGBr zcode reached herez-C:/Users/DELL/Downloads/Base Model/model2.pth� )Z num_classes�cuda�cpu)Z map_location�boxes�labels�scores)�21 � )Zfigsize�offZOriginalr � � �r�none)Z linewidth� edgecolor� facecolorr1 r0 �whiteg�������?z
round,pad=0.2)r8 �alphar7 Zboxstyle)�colorZfontsizeZbboxz
%Y%m%d_%H%M%SZfile_z.png�staticz"successfully reached here yeah boyz predict.htmlz(reaching to this message is not possiblei� )�status)6r r �get�os�path�basename�joinr r' �sessionr �open�convert�22transformsZComposeZToTensorZ unsqueezer Zload_state_dict�torch�loadZdevicer+ Z is_available�to�evalZno_gradr r, �numpy�npZarray�pltZsubplotsZimshowZaxis� set_title� enumerate�patchesZ RectangleZ add_patch�textr �int�dictr23 �now�strftimer �24STATIC_URL�BASE_DIRZsavefigr r )r r$ �filename�
absolute_pathr( r! � transformZ25img_tensorZ26model_pathZ predictionsZoutputsr- r. r/ Z original_npZ image_bgrZfigZaxs�jZboxZrect� timestampZimage_filenamer r r �predict' sj 27 2829�3031(32( 333435360�37$�38 39r] )%Zdjango.shortcutsr � django.httpr r �django.templater �formsr � django.confr r40 Z"monuments.faster_models.fasterrcnnr r r
r rK rL r? rG Ztorchvision.transformsrF r �F�PILr r Zmatplotlib.pyplotZpyplotrM Zmatplotlib.patchesrP r r r# r] r r r r �<module> s$