def main(args=None):
args = arg_parser().parse_args(args)
if args.verbosity == 1:
level = logging.getLevelName('INFO')
elif args.verbosity >= 2:
level = logging.getLevelName('DEBUG')
else:
level = logging.getLevelName('WARNING')
logging.basicConfig(
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
level=level)
logger = logging.getLogger(__name__)
try:
_ = nyul.nyul_normalize(args.img_dir, args.mask_dir, args.output_dir,
args.standard_hist)
if args.plot_hist:
with warnings.catch_warnings():
warnings.filterwarnings('ignore', category=FutureWarning)
from intensity_normalization.plot.hist import all_hists
import matplotlib.pyplot as plt
ax = all_hists(args.output_dir, args.mask_dir)
ax.set_title('Nyul and Udupa')
plt.savefig(os.path.join(args.output_dir, 'hist.png'))
return 0
except Exception as e:
logger.exception(e)
return 1
def test_nyul_normalization(self):
from intensity_normalization.normalize import nyul
normalized = nyul.nyul_normalize(self.data_dir, write_to_disk=False)