def test1(): S = loadExamplarPattens() a = np.random.rand(100).reshape(10,10) W = a > 0.5 C = structAwareSynthesize(S, W) return C
P = np.array([1, 0], dtype='int32') W, tileSpectrumMap = createInputSimple(I, V, M, P) print 'create input done.' print 'load examplars ..' examplars_pattern = loadExamplarsPattern() examplars_aabb = loadExamplarsAABB() examplars_tz = loadExamplarsTranslateZSimple() block_extents = loadBlockExtents() examplar_res = loadExamplarResolution() res = I.shape target_pattern = W print 'synthesis ..' C = structAwareSynthesize(examplars_pattern, target_pattern) print 'synthesis done.' # print '(checking)target pattern:' # printPattern(target_pattern) # target_pattern_syn = parsePattern(examplars_pattern, C) # print 'synthesis pattern:' # printPattern(target_pattern_syn) print 'genereate tileId and tileTransform ..' tileId = C[:, :, 0] tileTranslate = np.zeros((res[0], res[1], 3), dtype='float32') for i in range(0, res[0]): for j in range(0, res[1]): tileTranslate[i, j, 0:2] = examplarTranslate(
if __name__ == '__main__': print 'load examplars ..' examplars_pattern = loadExamplarsPattern() examplars_aabb = loadExamplarsAABB() examplars_tz = loadExamplarsTranslateZSimple() block_extents = loadBlockExtents() examplar_res = loadExamplarResolution() print 'load target pattern ..' res = [900, 1500] target_pattern = setWeavePattern(res) print 'synthesis ..' C = structAwareSynthesize(examplars_pattern, target_pattern) print 'synthesis done.' # print '(checking)target pattern:' # printPattern(target_pattern) # target_pattern_syn = parsePattern(examplars_pattern, C) # print 'synthesis pattern:' # printPattern(target_pattern_syn) print 'genereate tileId and tileTransform ..' tileId = C[:,:,0] tileTranslate = np.zeros((res[0], res[1], 3), dtype='float32') for i in range(0, res[0]): for j in range(0, res[1]): tileTranslate[i,j,0:2] = examplarTranslate(examplars_aabb[C[i,j,0]][:,0:2], examplar_res, C[i,j,1:3])
def test2(): S = loadExamplarPattens() W = generteTwill1() #W = np.tile(W, (1, 3)) C = structAwareSynthesize(S, W) return C