yerang commited on
Commit
749e6ac
·
verified ·
1 Parent(s): e6acc0a

Update stf/stf-api-alternative/src/stf_alternative/preprocess.py

Browse files
stf/stf-api-alternative/src/stf_alternative/preprocess.py CHANGED
@@ -2,7 +2,7 @@ import gc
2
  import os
3
  from pathlib import Path
4
 
5
- import cv2
6
  from PIL import Image
7
  from tqdm import tqdm
8
 
 
2
  import os
3
  from pathlib import Path
4
 
5
+ import cv2; cv2.setNumThreads(0); cv2.ocl.setUseOpenCL(False)
6
  from PIL import Image
7
  from tqdm import tqdm
8