mias_preprocess.py 文件源码

def convert_pgm_to_png():
    images = []
    for i, row in normal.iterrows():
        images.append(read_pgm('./mias/pgm/' + row['reference_number'] + '.pgm'))

    j = 0;
    for i, row in normal.iterrows():
        images[j].setflags(write=1)
        if (int(row['reference_number'][-3:]) % 2 == 0):
            images[j][:324, 700:1024] = np.zeros((324, 324))
        else:
            images[j][:324, :324] = np.zeros((324, 324))
            matlabimg.imsave('./mias/png/' + row['reference_number'] + '.png', images[j], vmin=0, vmax=255, cmap='gray')
        j += 1