3
f\ @ s d dl mZmZmZmZmZmZ d dlZd dl Z
d dl m Z d dl
Z
d dlmZmZmZ d dlZd dlZd dlZd dlmZ d dlmZ eeZde_ee d dl Z d dl Z dej d < e j a!d
d Z"e j#j$j%e" ed a&ed
Z'ej(d dd Z)ej(ddd Z*ej(ddd Z+ej(dddddd gd d d Z,ej(d ddddd gd d!d" Zej(d#d$d% Z-ed&krej.ddd' dS )( )Flaskredirecturl_forrequestrender_templatesend_from_directoryN)Image)
predict_image
load_model_h5get_class_names)Cache)CORST2ZTF_CPP_MIN_LOG_LEVELc C s$ t j } d| j_d| j_t j| dS )Ng?T)config)tfZ ConfigProtoZ gpu_optionsZ per_process_gpu_memory_fractionZ allow_growthSession)r r -/var/www/html/image_classification/api/api.py get_session s r zD/var/www/html/models/model1/Concat_VGG16_64_100_0_224_adam_0.0001.h5z+/var/www/html/models/model1/class_names.csv/c C s dS )NZStartedr r r r r start# s r z /healthcheckc C s dS )Npingr r r r r r & s r z/uploadc C s dS )Nzr r r r r upload* s r z/echoGETPOSTPATCHPUTDELETE)methodsc C sJ t jdkrdS t jdkr dS t jdkr*dS t jdkr8dS t jd krFd
S d S ) Nr z
ECHO: GET
r z ECHO: POST
r z ECHO: PACTH
r z
ECHO: PUT
r z ECHO: DELETE)r methodr r r r api_echo. s
r z/apic C s2 t jtjd j} | jd} ttt| }t j
|S )NZimageRGB) r openr filesstreamconvertr model class_namesjsondumps)img
predictionr r r predict@ s
r, z/inputc C s t j tj } dtjkrztjjd}d}d|i}tjj|d |}tjj |}t
j |}|j d}t
tt|}tj|S d}t
j |}|j d}t
tt|}tj|S | j W d Q R X d S )NfilenamezZMozilla/5.0 (Windows; U; Windows NT 5.1; en-US; rv:1.9.0.7) Gecko/2009021910 Firefox/3.0.7z
User-Agentr! z3/var/www/html/image_classification/images/49173.jpg)graphZ
as_defaultr r r argsgeturllibZRequestZurlopenr r" r% r r& r' r( r) close) ZsessZ image_urlZ
user_agentZheadersZ
request_imageZ
response_datar* r+ Z
myfilenamer r r renderG s$
r3 __main__)debugZ use_reloader)/Zflaskr r r r r r Zurllib.requestr1 ZnumpyZnpZPILr Zrequestsr, r r
r sysosr( Z
flask_cachingr Z
flask_corsr
__name__apir5 ZkerasZ
tensorflowr environZget_default_graphr. r ZbackendZtensorflow_backendZ set_sessionr& r' Zrouter r r r r3 runr r r r s: