Auto-format by https://ultralytics.com/actions
UltralyticsAssistant
parent
744f7c172b
commit
a27d403461
88
app.py
88
app.py
|
|
@ -15,19 +15,20 @@ import subprocess
|
|||
# from wtforms.validators import InputRequired
|
||||
|
||||
|
||||
|
||||
app = Flask(__name__)
|
||||
|
||||
streaming_active = False
|
||||
output_folder = 'videos'
|
||||
output_folder = "videos"
|
||||
video_writer = None
|
||||
# class UploadFileForm(FlaskForm):
|
||||
# file = FileField("File", validators=[InputRequired()])
|
||||
# submit = SubmitField("Upload File")
|
||||
|
||||
@app.route('/')
|
||||
|
||||
@app.route("/")
|
||||
def index():
|
||||
return render_template('index.html')
|
||||
return render_template("index.html")
|
||||
|
||||
|
||||
# @app.route('/home', methods=['GET',"POST"])
|
||||
# def home():
|
||||
|
|
@ -38,8 +39,8 @@ def index():
|
|||
# return "File has been uploaded."
|
||||
# return render_template('index.html', form=form)
|
||||
|
||||
@app.route('/start_stream')
|
||||
|
||||
@app.route("/start_stream")
|
||||
def start_stream():
|
||||
global streaming_active
|
||||
global out
|
||||
|
|
@ -47,58 +48,63 @@ def start_stream():
|
|||
streaming_active = True
|
||||
start_recording()
|
||||
|
||||
return jsonify({'status': 'success', 'message': 'Streaming started and recording initiated'})
|
||||
return jsonify({"status": "success", "message": "Streaming started and recording initiated"})
|
||||
else:
|
||||
return jsonify({'status': 'error', 'message': 'Streaming is already active'})
|
||||
return jsonify({"status": "error", "message": "Streaming is already active"})
|
||||
|
||||
@app.route('/stop_stream')
|
||||
|
||||
@app.route("/stop_stream")
|
||||
def stop_stream():
|
||||
global streaming_active
|
||||
if streaming_active:
|
||||
streaming_active = False
|
||||
stop_recording()
|
||||
|
||||
return jsonify({'status': 'success', 'message': 'Streaming stopped and recording saved'})
|
||||
|
||||
return jsonify({"status": "success", "message": "Streaming stopped and recording saved"})
|
||||
else:
|
||||
return jsonify({'status': 'error', 'message': 'Streaming is not active'})
|
||||
return jsonify({"status": "error", "message": "Streaming is not active"})
|
||||
|
||||
|
||||
def start_recording():
|
||||
global video_writer
|
||||
|
||||
|
||||
filename = f"recording_{datetime.now().strftime('%Y%m%d_%H%M%S')}.mp4"
|
||||
fourcc = cv2.VideoWriter_fourcc(*'mp4v') # You can change the codec as needed
|
||||
fourcc = cv2.VideoWriter_fourcc(*"mp4v") # You can change the codec as needed
|
||||
frame_size = (640, 480) # Adjust the frame size as needed
|
||||
# video_writer = cv2.VideoWriter(filename, fourcc, 10.0, frame_size)
|
||||
video_writer = cv2.VideoWriter(os.path.join(output_folder, filename), fourcc, 40.0, frame_size)
|
||||
|
||||
|
||||
|
||||
def stop_recording():
|
||||
global video_writer
|
||||
|
||||
|
||||
if video_writer is not None:
|
||||
video_writer.release()
|
||||
video_writer = None
|
||||
|
||||
@app.route('/video_feed')
|
||||
|
||||
|
||||
@app.route("/video_feed")
|
||||
def video_feed():
|
||||
return Response(generate_frames(), mimetype='multipart/x-mixed-replace; boundary=frame')
|
||||
return Response(generate_frames(), mimetype="multipart/x-mixed-replace; boundary=frame")
|
||||
|
||||
@app.route('/static/<path:filename>')
|
||||
|
||||
@app.route("/static/<path:filename>")
|
||||
def static_files(filename):
|
||||
return send_from_directory('static', filename)
|
||||
return send_from_directory("static", filename)
|
||||
|
||||
@app.route('/videos')
|
||||
|
||||
@app.route("/videos")
|
||||
def list_videos():
|
||||
videos = [video for video in os.listdir('videos') if video.endswith('.mp4')]
|
||||
videos = [video for video in os.listdir("videos") if video.endswith(".mp4")]
|
||||
return jsonify(videos)
|
||||
|
||||
@app.route('/video/<filename>')
|
||||
def stream_video(filename):
|
||||
return send_from_directory('videos', filename)
|
||||
|
||||
@app.route('/detection/<filename>')
|
||||
@app.route("/video/<filename>")
|
||||
def stream_video(filename):
|
||||
return send_from_directory("videos", filename)
|
||||
|
||||
|
||||
@app.route("/detection/<filename>")
|
||||
# def detection(filename):
|
||||
# # Placeholder for detection logic
|
||||
# print(f"Detection started for {filename}")
|
||||
|
|
@ -107,7 +113,7 @@ def detection(filename):
|
|||
try:
|
||||
print(f"filename################={filename}")
|
||||
# Construct the command string
|
||||
command = f'python3 detect.py --source ./videos/{filename}'
|
||||
command = f"python3 detect.py --source ./videos/{filename}"
|
||||
|
||||
# Execute the command
|
||||
process = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
||||
|
|
@ -115,19 +121,20 @@ def detection(filename):
|
|||
|
||||
if process.returncode != 0:
|
||||
# Handle errors if the command failed
|
||||
return jsonify({'status': 'Error', 'message': stderr.decode()}), 500
|
||||
return jsonify({"status": "Error", "message": stderr.decode()}), 500
|
||||
|
||||
# Return success response
|
||||
return jsonify({'status': 'Detection Done for ' + filename})
|
||||
return jsonify({"status": "Detection Done for " + filename})
|
||||
except Exception as e:
|
||||
# Handle any exceptions
|
||||
return jsonify({'status': 'Error', 'message': str(e)}), 500
|
||||
return jsonify({"status": "Error", "message": str(e)}), 500
|
||||
|
||||
|
||||
def generate_frames():
|
||||
global video_writer
|
||||
folder_path = '/tmp/camera_save_tutorial'
|
||||
folder_path = "/tmp/camera_save_tutorial"
|
||||
while streaming_active:
|
||||
image_files = [f for f in os.listdir(folder_path) if f.endswith(('.jpg', '.png'))]
|
||||
image_files = [f for f in os.listdir(folder_path) if f.endswith((".jpg", ".png"))]
|
||||
if image_files:
|
||||
try:
|
||||
latest_image = max(image_files, key=lambda x: os.path.getctime(os.path.join(folder_path, x)))
|
||||
|
|
@ -137,15 +144,14 @@ def generate_frames():
|
|||
if frame is None:
|
||||
raise FileNotFoundError("Empty image file or format not supported")
|
||||
|
||||
_, buffer = cv2.imencode('.jpg', frame)
|
||||
_, buffer = cv2.imencode(".jpg", frame)
|
||||
frame = buffer.tobytes()
|
||||
|
||||
# Write the frame to the video file if recording is active
|
||||
if video_writer is not None:
|
||||
video_writer.write(cv2.imdecode(np.frombuffer(frame, dtype=np.uint8), 1))
|
||||
|
||||
yield (b'--frame\r\n'
|
||||
b'Content-Type: image/jpeg\r\n\r\n' + frame + b'\r\n')
|
||||
yield (b"--frame\r\n" b"Content-Type: image/jpeg\r\n\r\n" + frame + b"\r\n")
|
||||
# time.sleep(0.05)
|
||||
|
||||
except Exception as e:
|
||||
|
|
@ -153,14 +159,8 @@ def generate_frames():
|
|||
continue
|
||||
|
||||
else:
|
||||
yield (b'--frame\r\n'
|
||||
b'Content-Type: image/jpeg\r\n\r\n' + b'\r\n')
|
||||
yield (b"--frame\r\n" b"Content-Type: image/jpeg\r\n\r\n" + b"\r\n")
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
||||
|
||||
app.run(debug=True)
|
||||
if __name__ == "__main__":
|
||||
app.run(debug=True)
|
||||
|
|
|
|||
|
|
@ -109,7 +109,7 @@ def run(
|
|||
# Directories
|
||||
# save_dir = increment_path(Path(project) / name, exist_ok=exist_ok) # increment run
|
||||
# (save_dir / "labels" if save_txt else save_dir).mkdir(parents=True, exist_ok=True) # make dir
|
||||
save_dir=Path(project)
|
||||
save_dir = Path(project)
|
||||
# Load model
|
||||
device = select_device(device)
|
||||
model = DetectMultiBackend(weights, device=device, dnn=dnn, data=data, fp16=half)
|
||||
|
|
|
|||
Loading…
Reference in New Issue