diff --git a/app/api/deepfake_detection.py b/app/api/deepfake_detection.py index 24cc75a5fb2ab73e5ce13e366485ad425b54cdde..1e3252b9eb9ee8b441de851202b4808240ce462e 100644 --- a/app/api/deepfake_detection.py +++ b/app/api/deepfake_detection.py @@ -1,9 +1,11 @@ import logging +import time from concurrent.futures import ThreadPoolExecutor import cv2 import numpy as np import pymysql +import torch.cuda from PIL import Image from flask import jsonify from sqlalchemy import create_engine, update @@ -68,6 +70,8 @@ def run_detection(video_entity, face_detector, deepfake_detector): .values(progress=progress) ) session.commit() + if not torch.cuda.is_available(): + time.sleep(0.5) # Init mask frame mask_frame = np.zeros_like(frame)