def video_feed(): print('i am called') #source="rtmp://localhost:1935/live/film" camera=Camera() tmp = Tmp.query.filter_by(user_id=current_user.id).first() camera.set_video_source(tmp.rtmpaddr) return Response(gen(camera), mimetype='multipart/x-mixed-replace; boundary=frame')
while True: frame = camera.get_frame() image = segnet.segment(frame) yield (b'--frame\r\n'+b'Content-Type: image/jpeg\r\n\r\n' + image + b'\r\n') @app.route('/video_feed') def video_feed(): return Response(gen(Camera()), mimetype='multipart/x-mixed-replace; boundary=frame') parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter) parser.add_argument('--model', help='File path of .tflite file.', required=True) parser.add_argument('--labels', help='File path of labels file.', required=True) parser.add_argument('--overlay', help='Overlay original image.', default=True) parser.add_argument('--source', help='picamera or cv', default='cv') args = parser.parse_args() if args.source == "cv": from camera_opencv import Camera source = 0 elif args.source == "picamera": from camera_pi import Camera source = 0 Camera.set_video_source(source) segnet = Segnet(args.labels, args.model, args.overlay) if __name__ == "__main__" : app.run(host = '0.0.0.0', port = 5000, debug = True)