import os
import numpy as np
import cv2
old_ng='./train_B/'
new_wg_dir='./clear/train_A/'
new_ng_dir='./clear/train_B/'
if not os.path.exists(new_wg_dir):
new_wg_dir=os.mkdir(new_wg_dir)
if not os.path.exists(new_ng_dir):
new_ng_dir=os.mkdir(new_ng_dir)
HH=os.listdir(old_ng)
for i in range(0,len(HH)):
fi=HH[i]
clear_img = cv2.imread(old_ng+fi)
blur_img2 = cv2.blur(clear_img,(1,15))
blur_img2[:,:512,:]=clear_img[:,:512,:]
blur_img3 = cv2.blur(clear_img,(15,1))
blur_img3[:,:512,:]=clear_img[:,:512,:]
blur_img4 = cv2.blur(blur_img2,(15,1))
blur_img4[:,:512,:]=clear_img[:,:512,:]
blur_img5 = cv2.medianBlur(clear_img,11)
blur_img5[:,:512,:]=clear_img[:,:512,:]
cv2.imwrite(new_wg_dir + '0_blur21_' +str(i)+'.jpg',blur_img2)
cv2.imwrite(new_ng_dir + '1_blur21_' +str(i)+'.jpg',clear_img)
cv2.imwrite(new_wg_dir + '0_blur31_' +str(i)+'.jpg',blur_img3)
cv2.imwrite(new_ng_dir + '1_blur31_' +str(i)+'.jpg',clear_img)
cv2.imwrite(new_wg_dir + '0_blur41_' +str(i)+'.jpg',blur_img4)
cv2.imwrite(new_ng_dir + '1_blur41_' +str(i)+'.jpg',clear_img)
cv2.imwrite(new_wg_dir + '0_blur51_' +str(i)+'.jpg',blur_img5)
cv2.imwrite(new_ng_dir + '1_blur51_' +str(i)+'.jpg',clear_img)
blur_img2 = cv2.blur(clear_img,(1,15))
blur_img2[:,512:,:]=clear_img[:,512:,:]
blur_img3 = cv2.blur(clear_img,(15,1))
blur_img3[:,512:,:]=clear_img[:,512:,:]
blur_img4 = cv2.blur(blur_img2,(15,1))
blur_img4[:,512:,:]=clear_img[:,512:,:]
blur_img5 = cv2.medianBlur(clear_img,11)
blur_img5[:,512:,:]=clear_img[:,512:,:]
cv2.imwrite(new_wg_dir + '0_blur22_' +str(i)+'.jpg',blur_img2)
cv2.imwrite(new_ng_dir + '1_blur22_' +str(i)+'.jpg',clear_img)
cv2.imwrite(new_wg_dir + '0_blur32_' +str(i)+'.jpg',blur_img3)
cv2.imwrite(new_ng_dir + '1_blur32_' +str(i)+'.jpg',clear_img)
cv2.imwrite(new_wg_dir + '0_blur42_' +str(i)+'.jpg',blur_img4)
cv2.imwrite(new_ng_dir + '1_blur42_' +str(i)+'.jpg',clear_img)
cv2.imwrite(new_wg_dir + '0_blur52_' +str(i)+'.jpg',blur_img5)
cv2.imwrite(new_ng_dir + '1_blur52_' +str(i)+'.jpg',clear_img)
print(i,len(HH))