ClassifyingEmotionsUsingUnsupervisedLearning/labelImages.py at master · dylangrosz/ClassifyingEmotionsUsingUnsupervisedLearning · GitHub

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
import csv

labels = '/Users/wintonyee/Documents/GitHub/ClassifyingEmotionsUsingUnsupervisedLearning/data/Cohn-Kanade Database FACS codes_updated based on 2002 manual_revised.csv'
rows = 0

def getYearlyCounts():
	f = open(labels)
	reader = csv.reader(f)
	for row in reader:
		label = []
		FACS = row[2].split('+')
		if '12' in FACS and '23' in FACS:
			label.append('Angry')
		if '9' in FACS or '10' in FACS:
			label.append('Disgust')
		if '1' in FACS and '2' in FACS and ('4' in FACS or '5e' in FACS):
			label.append('Fear')
		if '12' in FACS:
			label.append('Happy')
		if ('1' in FACS and '4' in FACS and '15' in FACS) or ('11' in FACS):
			label.append('Sadness')
		if ('1' in FACS and '2' in FACS) or ('5a' in FACS or '5b' in FACS):
			label.append('Surprise')
		if '14' in FACS:
			label.append('Contempt')
		print(FACS, label)


getYearlyCounts()