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)
