Add files via upload

Author: fiorezhang

picross_dataset.py

@@ -0,0 +1,15 @@
+#picross dataset
+#copyright fiorezhang@sina.com
+
+import numpy as np
+from picross_generator import generator as gen
+
+def load_data(row, col, visible, num):
+	x = np.zeros((num, row, col), dtype='int16')
+	y = np.zeros((num, 2, row, col), dtype='int16')
+	
+	for i in range(num):
+		x[i], y[i, 0], y[i, 1] = gen(row, col, visible)
+		
+	return x, y
+

picross_generator.py

@@ -0,0 +1,51 @@
+#picross_generator
+#copyright: fiorezhang@sina.com
+
+import numpy as np
+
+#calculate continued '1' for a string
+def calculate(s):
+	g = s.shape[0]
+	p = np.zeros((g), dtype='int16')
+	x, y = 0, 0#x: current position in return string, y: whether in '1' slice
+	for i in range(g):
+		if y == 0:#not in '1' slice
+			if s[i] == 1:
+				y = 1
+				p[x] += 1
+		else:#in '1' slice
+			if s[i] == 1:
+				p[x] +=1
+			else:
+				x += 1
+				y = 0
+	return p
+
+#generate the matrix, and calculated continued '1' number for rows and columns
+def generator(row, col, intense):
+	m = np.random.rand(row, col)
+	m = np.where(m<intense, 1, 0)
+	#print('Expect: ', intense, 'Actual: ', m.sum()/(row*col))
+	#print(m)
+	
+	n_r = np.zeros((row, col), dtype='int16')
+	n_c = np.zeros((row, col), dtype='int16')
+	
+	for i in range(row):
+		n_r[i] = calculate(m[i])
+	
+	for j in range(col):
+		n_c[:,j] = calculate(m[:,j])
+	
+	if (m.sum() != n_r.sum()) or (m.sum() != n_c.sum()):
+		print('ERROR')
+	
+	return m, n_r, n_c
+	
+#test functions
+'''
+x, y, z = generator(10, 10, 0.6)
+print(x)
+print(y)
+print(z)
+'''