investigate_emoji.py

import os
import csv
import numpy as np
import cv2
import math

import get_colour

# get list of emoji files, make csv that details dominant colour of each emoji

def get_list_emojis(folder_path):
    list_full = [(folder_path + x) for x in os.listdir(folder_path)]

    return list_full

def get_emoji_colour(list_of_files, csv_path, subsample=1):
    emoji_colour = []
    print("Processing {} emojis...".format(len(list_of_files)))

    with open(csv_path, "w") as outfile:
        outfile.write("path|r|g|b\n")

        for i in range(0, len(list_of_files)):
            print("\rProcessing emoji {} out of {} ({}%)...".format(i+1, len(list_of_files), int(100*i/len(list_of_files))), end="")
            img = cv2.imread(list_of_files[i])
            (r, g, b) = get_colour.get_dominant_colour(img)
            emoji_colour.append([list_of_files[i], (r,g,b)])
            for s in range(0, subsample**2):
                outfile.write("%s|%d|%d|%d\n" % (list_of_files[i], r, g, b))

        print("\n")
        outfile.close()
    
    print("Processed %d emojis and wrote to %s" % (len(list_of_files), csv_path))