Python CNN.pre_train Examples

Example #1

File: dbn_kosode.py Project: hamalab-test/t.nishisaki

    makeFolder()

    # 時間計測
    time1 = time.clock()

    print '~~~CNN1~~~'

    cnn1 = CNN(train_set, filter_shape, filter_shift_list[0], input_shape, node_shape[1], cnn_pre_train_lr, cnn_pre_train_epoch, isRGB)

    output_list = cnn1.output()
    cnn_saveColorImage(output_list, node_shape[1], 'cnn1_before_train')
    output_list_norm = local_contrast_normalization(output_list)
    cnn_saveColorImage(output_list_norm, node_shape[1], 'cnn1_before_training_norm')

    cnn1.pre_train()
    # cnn1.setW(cnn1_W)

    output_list = cnn1.output()
    cnn_saveColorImage(output_list, node_shape[1], 'cnn1_after_train')
    output_list_norm = local_contrast_normalization(output_list)
    cnn_saveColorImage(output_list_norm, node_shape[1], 'cnn1_after_train_norm')

    print '~~~CNN2~~~'

    cnn2 = CNN(cnn1.output(), filter_shape, filter_shift_list[1], node_shape[1], node_shape[2], cnn_pre_train_lr, cnn_pre_train_epoch, isRGB)
    output_list = cnn2.output()
    cnn_saveColorImage(output_list, node_shape[2], 'cnn2_before_train')
    output_list_norm = local_contrast_normalization(output_list)
    cnn_saveColorImage(output_list_norm, node_shape[2], 'cnn2_before_train_norm')