From c496ffc1ba740f8df315e910e80b31aa565f0f95 Mon Sep 17 00:00:00 2001 From: luanhailiang Date: Sat, 23 Sep 2023 00:49:42 +0800 Subject: [PATCH] 4 --- app.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/app.py b/app.py index e085d4d..a7bba1c 100644 --- a/app.py +++ b/app.py @@ -14,7 +14,6 @@ from multiprocessing import Process, Event app = Flask(__name__) -device_count = torch.cuda.device_count() workers = {} @app.route('/start/') @@ -63,7 +62,6 @@ def worker(id,source,region=None,stream=False): region = 10 else: region = int(region) - print("GPU device count:", device_count) print("start loading model...",id,source,region) model = YOLO('./best.pt') print("start loaded model!!!") @@ -95,8 +93,7 @@ def worker(id,source,region=None,stream=False): print("Error opening video stream....") break continue - print("device_count",int(id)%device_count) - result = model.track(frame,show=False,stream=False,persist=True,device=int(id)%device_count) + result = model.track(frame,show=False,stream=False,persist=True,device=int(id)%4) use += " track:"+str((time.time()*1000) - bgn) del(ret) del(frame)