/
kohonen_colors.py
50 lines (38 loc) · 1.2 KB
/
kohonen_colors.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
import numpy as np
import pylab
import random
from PIL.ImageQt import ImageQt
from pybrain.structure.modules import KohonenMap
from PyQt4 import QtGui
from scipy import misc
def som2image(som, size):
data = np.zeros( (size,size,3), dtype=np.uint8 )
x = 0
y = 0
for i in som.neurons:
for j in i:
data[x][y] = j
y += 1
if (y >= size):
y = 0
x += 1
return ImageQt(misc.toimage(data))
def aldohonen(size, iterations, training_input, refresh_rate, callback):
som = KohonenMap(3, size)
#just changing initial neurons weight from 0~1 to 0~255
for i in som.neurons:
for j in i:
for k in range(len(j)):
j[k] *= 255
training_size = len(training_input) - 1
for i in range(iterations):
# one forward and one backward (training) pass
som.activate(training_input[random.randint(0,training_size)])
som.backward()
#just draw som in our Qt View
if (i % refresh_rate == 0):
image = som2image(som, size)
callback(image, i)
else:
callback(None, i)
callback(som2image(som, size), iterations - 1)