asdTools-RGB图像转为三通道灰度图

Github - 开源代码及Readme
Blog - 工具介绍

摘要:转换RGB图像为三通道灰度图。

场景

  • 希望对灰度图进行处理,但又希望和RGB的处理保持一致,因此需要三通道的灰度图。

使用示例

from asdTools.Classes.Image.ImageBase import ImageBase

class RGB2Gray3Channel(ImageBase):
    """
    转换RGB图像为三通道灰度图。示例见 Sample\RGB2Gray3Channel
    Convert RGB images into gray images with 3 channels. Example: Sample\RGB2Gray3Channel
    """
    def __init__(self, **kwargs) -> None:
        super().__init__(multipleFiles=True, **kwargs)

    def __call__(self, imgs_dir:str) -> str:
        self.run(imgs_dir)

    def run(self, imgs_dir:str) -> str:
        imgs_path = self.get_paths_from_dir(imgs_dir)
        imgs_path = self.filter_ext(imgs_path, ["png", "jpg", "jpeg"])
        for i, img_path in enumerate(imgs_path):
            img_rgb = self.read_img(img_path)
            img_gray = self.convert_RGB_to_gray(img_rgb, True)
            img_name = self.get_name_of_file(img_path, keepExt=True)
            img_dir = self.get_dir_of_file(img_path, root=imgs_dir)
            output_path = self.save_image(img_gray, output_file=img_name, output_middle_dir=img_dir)
            self.log(f"{i+1}: {img_name} has converted from rgb to gray img with 3 channels, saved in {output_path}.")
            img_rgb.close()
            img_gray.close()
        self.done()

if __name__ == "__main__":
    imgs_dir = r"Sample\RGB2Gray3Channel\before"
    Converter = RGB2Gray3Channel()
    Converter(imgs_dir)

版权声明:
作者:MWHLS
链接:http://panwj.top/4681.html
来源:无镣之涯
文章版权归作者所有,未经允许请勿转载。

THE END
分享
二维码
打赏
< <上一篇
下一篇>>