Upload 3 files

Dockerfile

@@ -0,0 +1,11 @@
+FROM python:3.9
+
+WORKDIR /code
+
+COPY ./requirements.txt /code/requirements.txt
+
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+COPY . .
+
+CMD ["gunicorn", ".b","0.0.0.0.7860","main:app"]

main.py

@@ -0,0 +1,91 @@
+from flask import Flask, request, jsonify
+from flask_sqlalchemy import SQLAlchemy
+import cv2
+import face_detection
+import numpy as np
+import base64
+
+app = Flask(__name__)
+
+# Configuration for SQLite database
+app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///faces.db'
+app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
+db = SQLAlchemy(app)
+
+class Frame(db.Model):
+    id = db.Column(db.Integer, primary_key=True)
+    frame_data = db.Column(db.LargeBinary)  # Use LargeBinary to store binary data
+    count_of_people = db.Column(db.Integer)
+    
+def create_database():
+    db.create_all()
+
+
+def draw_faces(im, bboxes):
+    for bbox in bboxes:
+        x0, y0, x1, y1 = [int(_) for _ in bbox]
+        cv2.rectangle(im, (x0, y0), (x1, y1), (0, 0, 255), 2)
+
+def detect_faces_and_save(frame):
+    detector = face_detection.build_detector("DSFDDetector", confidence_threshold=0.5, nms_iou_threshold=0.3)
+    det_raw = detector.detect(frame[:, :, ::-1])
+    dets = det_raw[:, :4]
+    draw_faces(frame, dets)
+
+    count_of_people = len(dets)
+
+
+    frame_data_encoded = base64.b64encode(cv2.imencode('.jpg', frame)[1].tobytes())
+    new_frame = Frame(frame_data=frame_data_encoded, count_of_people=count_of_people)
+    db.session.add(new_frame)
+    db.session.commit()
+
+    return frame, count_of_people
+
+@app.route('/video_feed', methods=['POST'])
+def video_feed():
+    if 'video' not in request.files:
+        return jsonify({'error': 'No video file in the request'})
+
+    video_file = request.files['video']
+
+    if video_file.filename == '':
+        return jsonify({'error': 'Empty video file'})
+
+    video_bytes = video_file.read()
+    video_np = np.frombuffer(video_bytes, dtype=np.uint8)
+    vidcap = cv2.imdecode(video_np, cv2.IMREAD_UNCHANGED)
+
+    if vidcap is None:
+        return jsonify({'error': 'Error decoding video file'})
+
+    create_database()
+
+    success, image = vidcap.read()
+
+    while success:
+
+        detect_faces_and_save(image)
+
+        success, image = vidcap.read()
+
+    return jsonify({'result': 'Video parsed'})
+
+@app.route('/get_frames', methods=['GET'])
+def get_frames():
+    frames = Frame.query.all()
+    frames_data = []
+
+    for frame in frames:
+        frames_data.append({
+            'frame_data': frame.frame_data.decode('latin1'),
+            'count_of_people': frame.count_of_people
+        })
+    
+    db.session.query(Frame).delete()
+    db.session.commit()
+    
+    return jsonify(frames_data)
+
+if __name__ == '__main__':
+    app.run(debug=True)

requirements.txt

@@ -0,0 +1,75 @@
+asttokens==2.2.1
+backcall==0.2.0
+blinker==1.7.0
+certifi==2023.11.17
+charset-normalizer==3.3.2
+click==8.1.7
+colorama==0.4.6
+comm==0.1.4
+contourpy
+cvzone==1.6.1
+cycler==0.12.1
+debugpy
+decorator==5.1.1
+executing==1.2.0
+face-detection==0.2.2
+filelock==3.13.1
+Flask==3.0.0
+Flask-SQLAlchemy==3.1.1
+fonttools==4.46.0
+fsspec==2023.12.1
+greenlet==3.0.2
+gunicorn
+idna==3.6
+imutils==0.5.4
+ipykernel==6.25.0
+ipython==8.14.0
+itsdangerous==2.1.2
+jedi==0.19.0
+Jinja2==3.1.2
+jupyter_client==8.3.0
+jupyter_core==5.3.1
+kiwisolver==1.4.5
+MarkupSafe==2.1.3
+matplotlib==3.8.2
+matplotlib-inline==0.1.6
+mpmath==1.3.0
+nest-asyncio==1.5.7
+networkx==3.2.1
+numpy==1.26.2
+opencv-python==4.8.1.78
+packaging==23.1
+pandas==2.1.3
+parso==0.8.3
+pickleshare==0.7.5
+Pillow==10.1.0
+platformdirs==3.10.0
+prompt-toolkit==3.0.39
+psutil==5.9.5
+pure-eval==0.2.2
+py-cpuinfo==9.0.0
+Pygments==2.15.1
+pyparsing==3.1.1
+python-dateutil==2.8.2
+pytz==2023.3.post1
+PyYAML==6.0.1
+pyzmq==25.1.0
+requests==2.31.0
+scipy==1.11.4
+seaborn==0.13.0
+six==1.16.0
+SQLAlchemy==2.0.23
+stack-data==0.6.2
+sympy==1.12
+thop==0.1.1.post2209072238
+torch==2.1.1
+torchvision==0.16.1
+tornado==6.3.2
+tqdm==4.66.1
+traitlets==5.9.0
+typing_extensions==4.8.0
+tzdata==2023.3
+ultralytics==8.0.222
+urllib3==2.1.0
+wcwidth==0.2.6
+Werkzeug==3.0.1