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)
## @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)
# -*- 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)