From 4ba81cb14790f7d815262eb31901edbbcbbeeb6d Mon Sep 17 00:00:00 2001
From: Emiliana <38952746+emilianavt@users.noreply.github.com>
Date: Sat, 2 Jan 2021 23:45:55 +0100
Subject: [PATCH] Fixed thread calculation.

---
 tracker.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tracker.py b/tracker.py
index 7c0f73e..861d684 100644
--- a/tracker.py
+++ b/tracker.py
@@ -497,7 +497,7 @@ class Tracker():
     def __init__(self, width, height, model_type=3, detection_threshold=0.6, threshold=None, max_faces=1, discard_after=5, scan_every=3, bbox_growth=0.0, max_threads=4, silent=False, model_dir=None, no_gaze=False, use_retinaface=False, max_feature_updates=0, static_model=False, feature_level=2, try_hard=False):
         options = onnxruntime.SessionOptions()
         options.inter_op_num_threads = 1
-        options.intra_op_num_threads = max(max_threads,4)
+        options.intra_op_num_threads = min(max_threads,4)
         options.execution_mode = onnxruntime.ExecutionMode.ORT_SEQUENTIAL
         options.graph_optimization_level = onnxruntime.GraphOptimizationLevel.ORT_ENABLE_ALL
         options.log_severity_level = 3
@@ -536,7 +536,7 @@ class Tracker():
         for i in range(self.max_workers):
             options = onnxruntime.SessionOptions()
             options.inter_op_num_threads = 1
-            options.intra_op_num_threads = max_threads // self.max_workers
+            options.intra_op_num_threads = min(max(max_threads // self.max_workers, 4), 1)
             if options.intra_op_num_threads < 1:
                 options.intra_op_num_threads = 1
             elif i < extra_threads: