Esempio n. 1
0
    font_size = 15
    fig.suptitle("NPR-SFS", fontsize=font_size)

    num_cols = len(method_names) + 1
    fig.add_subplot(1, num_cols, 1)
    plt.title("Ground truth", fontsize=font_size)
    plt.imshow(NO_32F)
    plt.axis('off')

    col_id = 2

    for method_name in method_names:
        method_file = methodFile(method_name, data_name)
        N_32F = loadRGBA(method_file)

        fig.add_subplot(1, num_cols, col_id)
        plt.title(method_name, fontsize=font_size)
        plt.imshow(N_32F)
        plt.axis('off')
        col_id += 1

    result_dir = resultDir(batch_name)
    result_file = resultFile(result_dir, data_name)
    plt.savefig(result_file)


if __name__ == '__main__':
    print methodNames()
    batchResults(batch_func, batch_name)
Esempio n. 2
0
## @package npr_sfs.results.ibme
#
#  npr_sfs.results.ibme utility package.
#  @author      tody
#  @date        2015/09/01


from npr_sfs.results.results import batchResults, resultFile, resultDir

from npr_sfs.io_util.image import saveNormal, loadRGBA
from npr_sfs.datasets.loader import loadData
from npr_sfs.cv.image import alpha, luminance
from npr_sfs.methods.ibme import estimateNormal

batch_name = "IBME"


def batch_func(data_name):
    C_8U = loadData(data_name, loader_func=loadRGBA)
    A_8U = alpha(C_8U)
    I_32F = luminance(C_8U)
    N_32F, D_32F = estimateNormal(I_32F)

    result_dir = resultDir(batch_name)
    N_file = resultFile(result_dir, data_name)
    saveNormal(N_file, N_32F, A_8U)


if __name__ == '__main__':
    batchResults(batch_func, batch_name)
Esempio n. 3
0
# -*- coding: utf-8 -*-
## @package npr_sfs.main
#
#  Main functions.
#  @author      tody
#  @date        2015/09/01

from npr_sfs.datasets.loader import dataNames, loadData

import results.ibme
import results.lumo
import results.compare
from npr_sfs.results.results import batchResults

batch_modules = [results.ibme,
                 results.lumo,
                 results.compare]

if __name__ == '__main__':
    for batch_module in batch_modules:
        batchResults(batch_module.batch_func, batch_module.batch_name)


Esempio n. 4
0
# -*- coding: utf-8 -*-
## @package npr_sfs.main
#
#  Main functions.
#  @author      tody
#  @date        2015/09/01

from npr_sfs.datasets.loader import dataNames, loadData

import results.ibme
import results.lumo
import results.compare
from npr_sfs.results.results import batchResults

batch_modules = [results.ibme, results.lumo, results.compare]

if __name__ == '__main__':
    for batch_module in batch_modules:
        batchResults(batch_module.batch_func, batch_module.batch_name)