-
Notifications
You must be signed in to change notification settings - Fork 0
/
image_outlier.py
51 lines (38 loc) · 1.24 KB
/
image_outlier.py
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
# the purpose is to move files depending on image size
# maybe not the best way but...
import os
from PIL import Image
cwd = os.getcwd()
src = os.path.join(cwd, "images")
dest = os.path.join(cwd, "rejects")
def filter():
# grabs all file names in folder
files = os.listdir(src)
image_area = []
for i in files:
filepath = os.path.join(src, i)
img = Image.open(filepath)
w = img.width
h = img.height
# getting the area of the image
area = w * h
image_area.append([area, i])
image_area.sort()
q1_locator = round((0.25) * (len(image_area)))
Q1 = image_area[q1_locator-1][0]
q3_locator = round((0.75) * (len(image_area)))
Q3 = image_area[q3_locator-1][0]
IQR = Q3 - Q1
uppper_boundary = Q3 + (1.5 * IQR)
lower_boundary = Q1 - (1.5 * IQR)
# grabs unwanted images
filtered = [ i[1] for i in image_area if (lower_boundary <= i[0]) and (i[0] <= uppper_boundary) ]
return filtered
def move():
# returns list of images that
files = filter()
for i in files:
src_path = os.path.join(src, i)
dest_path = os.path.join(dest, i)
os.rename(src_path, dest_path)
move()