def generate_blur_canny_subfolders(folder, outputFolder):
from src.utils.canny import my_canny
from src.utils.canny import best_edges
from src.utils.io import filename2arr
from src.utils.io import saveimage_pil
import os
for dirName, subdirList, fileList in os.walk(folder):
for subdir in subdirList:
for dirName2, subdirList2, fileList in os.walk(dirName + '/' +subdir):
for filename in fileList:
parts = filename.split('.')
if parts[0] == '':
continue
arr = filename2arr(dirName2 + '/' + filename)
ret = best_edges(arr)
if not os.path.exists(outputFolder + '/' + subdir):
os.makedirs(outputFolder + '/' + subdir)
outfn = outputFolder + '/' + subdir + '/'+ parts[0] + '.jpg'
print "saving canny file " + outfn
saveimage_pil(255 * ret.astype('uint8'), outfn, show=False)
halves = pickle.load(open(fn, "rb"))
halves = non_max_suppression_merge(halves, overlapThresh=0)
if halves is not None:
# 1. detect ROI
bbox = get_huge_bounding_box(halves)
img1 = Image.fromarray(arr)
# fn = result_dir + '/' + testn + '_half_boxes.pkl'
# pickle.dump(halves, open(fn, "wb"))
img1=add_boxes(halves, img1)
img1.show()
img1.save(result_dir +'/' + testn +'_half_boxes.jpg')
# 2. detect edges
ROI = arr[bbox[1]: bbox[3], bbox[0]: bbox[2]]
edges, sigma, cur_ROI= best_edges(ROI, threshold=0.02)
showimage_pil(edges)
# ## 3. using hough_transform to find initial horizontal line
#
# from src.utils.hough import hough_horizontal
# fn = result_dir + '/' + testn + 'canny_hough'
# img2 = Image.fromarray(ROI)
# lines_init, _ = hough_horizontal(edges, fn, hough_line_len=30, line_gap=40, save=True, show=True, raw=img2, xdiff=0, ydiff=0)
#
#
# ## 4. Extend line using smaller sigma
# from src.utils.util import extend_line_canny_points
# from src.utils.canny import my_canny
# fn = fn = result_dir + '/' + testn + 'canny_extend'
