From 4f27f21ed4a1ab346e3facb82f61d257b96ad615 Mon Sep 17 00:00:00 2001 From: joizhang Date: Tue, 13 Dec 2022 14:27:09 +0800 Subject: [PATCH] feat: sleep and yield cpu --- app/api/deepfake_detection.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/app/api/deepfake_detection.py b/app/api/deepfake_detection.py index 24cc75a..1e3252b 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) -- Gitee