ir_detection/normalize.py at master · evil-panda-team/ir_detection · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
import cv2
import os
import glob

# In[]:
dataset_path = '/home/datasets/ir/'
mipt_dataset_path = dataset_path + 'mipt/images/'
mipt_normilized_dataset_path = dataset_path + 'mipt/clahe/'
os.makedirs(mipt_normilized_dataset_path, exist_ok=True)

files = [f for f in glob.glob(mipt_dataset_path + "*.png", recursive=True)]

# In[]:
clahe = cv2.createCLAHE(clipLimit=2.0, tileGridSize=(8,8))

for f in files:
    img = cv2.imread(f,0)

#    img = cv2.equalizeHist(img)

#    cv2.normalize(img, img, 0, 255, cv2.NORM_MINMAX)

    img = clahe.apply(img)

    cv2.imwrite('{}{}'.format(mipt_normilized_dataset_path,f.split('/')[-1]), img)

# In[]:


# In[]:


# In[]:


# In[]:


# In[]:


# In[]: