diff --git a/backend/main.py b/backend/main.py index cddefa099390a78b7e1a48f031ac0ffd29759944..7ab27136d6675db99bbef0f8c48e9a2ab2880bdc 100644 --- a/backend/main.py +++ b/backend/main.py @@ -47,7 +47,7 @@ async def get_records(place: str, db: Session = Depends(get_db)): @app.post('/api/records', response_model=schemas.Record) -async def stats(record: schemas.RecordBase, db: Session = Depends(get_db)): +async def post_records(record: schemas.RecordBase, db: Session = Depends(get_db)): db_record = models.Records(**record.dict()) db.add(db_record) db.commit() @@ -56,7 +56,7 @@ async def stats(record: schemas.RecordBase, db: Session = Depends(get_db)): @app.delete('/api/records', response_model=None) -async def stats(id: int, db: Session = Depends(get_db)): +async def del_records(id: int, db: Session = Depends(get_db)): if id == 0: db.query(models.Records).delete() else: diff --git a/backend/video-capture.py b/backend/video-capture.py index 9437c1bf18a081f5eeab6ad58ff675324e358cdb..530f4e63382db7f550d58764c0458f069ac0a6f3 100644 --- a/backend/video-capture.py +++ b/backend/video-capture.py @@ -3,7 +3,7 @@ from datetime import datetime, timedelta import numpy as np import keras from utils.preprocessing import fix_singular_shape, norm_by_imagenet -from db import models, database, schemas +from db import models from db.database import SessionLocal @@ -24,13 +24,13 @@ while(cap.isOpened()): count_prediction = np.sum(pred_map) waiting_time = timedelta(seconds=120 + count_prediction * 30) record = {"place": "local", - "date": current_time, - "density": count_prediction, - "waiting_time": waiting_time} + "date": current_time, + "density": count_prediction, + "waiting_time": waiting_time} db_record = models.Records(**record) db.add(db_record) db.commit() db.refresh(db_record) count += 1 cap.release() -cv2.destroyAllWindows() \ No newline at end of file +cv2.destroyAllWindows()