File size: 654 Bytes
a8eb386 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
import sys
import time
import io
import os
import random
import string
import datetime
import base64
from PIL import Image, ImageFile
from flask import request, jsonify, Flask, url_for
import numpy as np
import tensorflow as tf
# import tensorflow.compat.v1 as tf
# tf.compat.v1.disable_eager_execution()
from detectweb import label_image
model_file = 'static/tf_files/retrained_graph.pb'
label_file = 'static/tf_files/retrained_labels.txt'
input_height = 224
input_width = 224
input_mean = 128
input_std = 128
input_layer = "input"
output_layer = "final_result"
g = tf.Graph()
with g.as_default():
graph = label_image.load_graph(model_file)
|