def main(args):
inference = Inference(args.model)
inference.load_model()
input = args.input
if input == 0:
input_feeder = InputFeeder('cam', input)
elif input.endswith('.jpg') or input.endswith('.jpeg') or input.endswith(
'.bmp'):
input_feeder = InputFeeder('image', input)
is_image = True
else:
input_feeder = InputFeeder('video', input)
input_feeder.load_data()
if is_image:
outputs = inference.predict(input_feeder.cap)
inference.preprocess_output(outputs)
return 0
frames = 0
for ret, frame in input_feeder.next_batch():
if not ret:
break
frames += 1
key = cv2.waitKey(60)
if key == 27:
break
outputs = inference.predict(frame)
inference.preprocess_output(outputs)
input_feeder.close()
file_handler = logging.FileHandler('app.log')
file_handler.setLevel(logging.ERROR)
file_handler.setFormatter(formatter)
logger.addHandler(file_handler)
# Instantiate our flask app
app = Flask(__name__)
port = int(os.environ.get("PORT", 5000))
Markdown(app)
APP_ROOT = os.path.dirname(os.path.abspath(__file__))
# Instantiate model
inf = Inference()
inf.load_model()
@app.route('/', methods=['GET'])
def index():
"""Renders the index '/' endpoint of the webpage
Returns:
html template -- Rendered index html endpoint
"""
return render_template('index.html')
@app.route('/info', methods=['GET'])
def short_description():
"""Renders the '/info' endpoint of the webpage
