yerang commited on
Commit
0128bf7
·
verified ·
1 Parent(s): bcc2b59

import cv2; cv2.setNumThreads(0); cv2.ocl.setUseOpenCL(False)

Browse files
stf/stf-api-alternative/src/stf_alternative/s2f_dir/src/datagen_aug.py CHANGED
@@ -6,7 +6,7 @@ import time
6
  from glob import escape, glob
7
  from pathlib import Path
8
 
9
- import cv2
10
  import numpy as np
11
  import pandas as pd
12
  import torch
 
6
  from glob import escape, glob
7
  from pathlib import Path
8
 
9
+ import cv2; cv2.setNumThreads(0); cv2.ocl.setUseOpenCL(False)
10
  import numpy as np
11
  import pandas as pd
12
  import torch