deadpyxel/digital-image-processing

View on GitHub
main.py

Summary

Maintainability
A
0 mins
Test Coverage
import argparse
import os

from simple_imaging.image import read_file
from simple_imaging.image import save_file


def main(args):
    image = None
    if args.filepath:
        for filepath in args.filepath:
            image = read_file(filepath)
    if args.output:
        output_dir = args.output
    blurred = image.laplacian_filter(inplace=False)
    result = image.add_image(blurred)

    save_file(f"{output_dir}/atv05/blurry_moon_lapl.pgm", blurred)
    save_file(f"{output_dir}/atv05/blurry_moon_lapl_add.pgm", result)


if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("filepath", nargs="+", help="Input file for reading")
    parser.add_argument(
        "-o",
        "--output",
        help="folder to use for output, defaults to output and will create itif doesn't exists",
        default="output",
    )
    args = parser.parse_args()
    main(args)