how to work on python open source projects code example
Example: python open source projects
from __future__ import print_function
from __future__ import division
import binascii
import struct
from PIL import Image, ImageColor
import numpy as np
import scipy
import scipy.misc
import scipy.cluster
import os
NUM_CLUSTERS = 5
root = 'A:\\Viktigt\\Mina program\\MostColor.py'
prompt_Root = False
prompt_Stop = True
write_colors_to_file = True
if prompt_Root == True:
root = input("Desired filepath>>> ")
dots3 = ['.png', '.jpg', '.bmp']
dots4 = ['.webp', '.jpeg']
cols = []
num = 0
exist = 0
failed = 0
for root, dirs, files in os.walk(root, topdown=False):
for name in files:
if name[-4:] in dots3 or name[-5:] in dots4:
exist += 1
try:
num += 1
print("\nFile #%s:" % num)
print("File Path: " + root)
print("File Name: " + name)
print("\nReading image . . .")
im = Image.open(root + "\\" + name)
width, height = im.size
print("Image size: " + str(width) + " x " + str(height))
if im.size > (150, 150):
im = im.resize((150, 150))
print("Resized image to 150 x 150.")
ar = np.asarray(im)
shape = ar.shape
try:
ar = ar.reshape(np.product(shape[:2]), shape[2]).astype(float)
except:
pass
print("\nFinding clusters . . .")
try:
codes, dist = scipy.cluster.vq.kmeans(ar, NUM_CLUSTERS)
except:
pass
print("\nCluster centres:\n", codes)
try:
vecs, dist = scipy.cluster.vq.vq(ar, codes)
except:
pass
counts, bins = np.histogram(vecs, len(codes))
index_max = np.argmax(counts)
peak = codes[index_max]
color = binascii.hexlify(bytearray(int(c) for c in peak)).decode('ascii')
if color == "00000000":
try:
peak = codes[index_max - 1]
except:
pass
color = binascii.hexlify(bytearray(int(c) for c in peak)).decode('ascii')
print("\nMost frequent color is %s (#%s)\n\n" % (peak, color))
img = Image.new("RGB", (width, height), ImageColor.getrgb("#" + color))
os.remove(root + "\\" + name)
img.save(root + "\\" + name)
if write_colors_to_file == True:
cols.append("#" + color)
except:
print("Operation failed (%s)" % root + "\\" + name)
failed += 1
if write_colors_to_file == True:
try:
cols = list(dict.fromkeys(cols))
f = open("colors_hex.csv", 'w')
f.truncate(0)
for col in cols:
try:
f.write(col + '\n')
print("\nAppended color to file.")
except:
print("\nCouldn't append color to file")
f.close()
except:
print("\nCouldn't append colors to file.")
if prompt_Stop == True:
input("Completed %s/%s operations>>>" % (str(exist - failed), str(exist)))