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