Selfie Segmentation crashes after running a while, on Linux CPU

[de-code]

System information (Please provide as much relevant information as possible)

• Have I written custom code (as opposed to using a stock example script provided in Mediapipe): I have written an example to better reproduce the issue
• OS Platform and Distribution (e.g., Linux Ubuntu 16.04, Android 11, iOS 14.4): Linux 20.04
• MediaPipe version: 0.8.6
• Bazel version: n/a
• Solution (e.g. FaceMesh, Pose, Holistic): Selfie Segmentation
• Programming Language and version ( e.g. C++, Python, Java): Python

Describe the expected behavior:

It shouldn't crash.

Standalone code you may have used to try to get what you need :

This more or less matches #2219 but for Linux and using Python.

I created a gist with Dockerfile to replicate the issue.

Copied the Python code here:

from time import sleep

import numpy as np
import mediapipe as mp
from tqdm import tqdm


mp_drawing = mp.solutions.drawing_utils
mp_selfie_segmentation = mp.solutions.selfie_segmentation

image_shape = (480, 640, 3)

np.random.seed(42)

print('Press Ctrl+C to exit')

try:
    with mp_selfie_segmentation.SelfieSegmentation(
        model_selection=1
    ) as selfie_segmentation:
        with tqdm() as pbar:
            while True:
                image = (np.random.random_sample(image_shape) * 255).astype('uint8')
                result = selfie_segmentation.process(image)
                result.segmentation_mask
                sleep(0.01)
                pbar.update(1)
except KeyboardInterrupt:
    print('exiting')

It might crash after a few hundred or only after more than 10k iterations.

I tried it with Pyhon 3.7, 3.8 and 3.9 as the base image.

Other info / Complete Logs :

Example output:

Press Ctrl+C to exit
INFO: Created TensorFlow Lite XNNPACK delegate for CPU.
10913it [04:23, 31.69it/s]

(It shouldn't exit until Ctrl+C was pressed)

In another app where I actually wanted to use it, I was getting a stacktrace (which I don't seem to see in the minimal example):

Fatal Python error: PyEval_SaveThread: NULL tstate

Thread 0x00007fd2e9ad0740 (most recent call first):
  File "/path/to/layered-vision/venv/lib/python3.8/site-packages/mediapipe/python/solution_base.py", line 307 in process
  File "/path/to/layered-vision/venv/lib/python3.8/site-packages/mediapipe/python/solutions/selfie_segmentation.py", line 76 in process
  File "/path/to/layered-vision/layered_vision/filters/mp_selfie_segmentation.py", line 94 in get_segmentation_mask
  ...
Aborted (core dumped)

[jiuqiant]

There is a GIL fix in the latest v0.8.7.1 python binaries, which may help resolve this issue. Try pip install mediapipe==0.8.7.1

[de-code]

At least with my minimal example it hasn't crashed anymore after more than an hour (using 0.8.7.1).

[cova-fe]

There is a GIL fix in the latest v0.8.7.1 python binaries, which may help resolve this issue. Try pip install mediapipe==0.8.7.1

I tried in the same conditions that made the previous version to crash, and it is not happening again. Maybe not a definitive proof, but looks promising.