merge_data.py 文件源码

def generate_context_feature(in_data_dir1, in_data_dir2, out_data_dir, dimension1, dimension2):

    if not os.path.exists(out_data_dir):
        os.makedirs(out_data_dir)

    file_paths, filenames = read_file_list(in_data_dir1)

    context_features = numpy

    i = 0
    for file_path, filename in zip(file_paths, filenames):
        features1, frame_number1 = load_binary_file(file_path, dimension1)
        features2, frame_number2 = load_binary_file(os.path.join(in_data_dir2, filename), dimension2)
        if frame_number1 != frame_number2:
            print(dimension2)
            print(filename)
            print("%s %d != %d" %(filename, frame_number1, frame_number2))
            print(features1.shape, features2.shape)
            os.exit(1)

        context_features = numpy.zeros((frame_number1, dimension1+dimension2))

        context_features[0:frame_number1, 0:dimension1] = features1
        context_features[0:frame_number2, dimension1:dimension1+dimension2] = features2

        print(filename, features1.shape, features2.shape, context_features.shape)

        context_filename = out_data_dir + '/' + filename

        context_features = numpy.asarray(context_features, 'float32')
        fid = open(context_filename, 'wb')
        context_features.tofile(fid)
        fid.close()