blob: 88d520276bf1dc9ab6e63dd94f7b31c1be24c33a (
plain)
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
|
import os
import cv2
import FocusStack
"""
Focus stack driver program
This program looks for a series of files of type .jpg, .jpeg, or .png
in a subdirectory "input" and then merges them together using the
FocusStack module. The output is put in the file merged.png
Author: Charles McGuinness (charles@mcguinness.us)
Copyright: Copyright 2015 Charles McGuinness
License: Apache License 2.0
"""
def stackHDRs(image_files):
focusimages = []
for img in image_files:
print "Reading in file {}".format(img)
focusimages.append(cv2.imread("input/{}".format(img)))
merged = FocusStack.focus_stack(focusimages)
cv2.imwrite("merged.png", merged)
if __name__ == "__main__":
image_files = sorted(os.listdir("input"))
for img in image_files:
if img.split(".")[-1].lower() not in ["jpg", "jpeg", "png"]:
image_files.remove(img)
stackHDRs(image_files)
print "That's All Folks!"
|
