update client

1996scarlet · Jan 22, 2021 · dc396ca · dc396ca
1 parent 7ee3d55
commit dc396ca
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 7 deletions.
diff --git a/PythonClient/TFLiteFaceDetector.py b/PythonClient/TFLiteFaceDetector.py
@@ -10,8 +10,7 @@ def __init__(self, filepath, input_size=(320, 240), conf_threshold=0.6,
                  nms_max_output_size=200, nms_iou_threshold=0.3) -> None:
 
         self._feature_maps = np.array([[40, 30], [20, 15], [10, 8], [5, 4]])
-        self._min_boxes = np.array([[10, 16, 24], [32, 48],
-                                    [64, 96], [128, 192, 256]])
+        self._min_boxes = [[10, 16, 24], [32, 48], [64, 96], [128, 192, 256]]
 
         self._resize = partial(cv2.resize, dsize=input_size)
         self._input_size = np.array(input_size)[:, None]

diff --git a/PythonClient/vtuber_link_start.py b/PythonClient/vtuber_link_start.py
@@ -45,9 +45,6 @@ def face_detection():
     while True:
         ret, frame = cap.read()
 
-        # frame = frame[150:800, 800:1600, :]
-        # frame = cv2.resize(frame, (960, 720))
-
         if not ret:
             break
 

diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,6 @@
 tensorflow >= 2.3
 opencv-python
-python-socketio == 4.6.0
-python-engineio == 3.13.0
+bidict == 0.21.2
+websocket-client==0.57.0
+python-socketio == 5.0.4
+python-engineio == 4.0.0