はじめに

この記事はNoteで連載している記事で扱っているCode部分のみを切り出しています。技術背景等にご興味がある方はNoteの記事の方をご参照ください。

helperの実装

　画像の入出力を多用するのでそれらをhelperとしてまとめて実装しておきます。そうすることでCodeが見やすくなりますし、メンテナンスが楽になります。使用するライブラリーは以下となります。事前にpip等でインストールしておいてください。

glob
OpenCV
numpy

それでは実際の実装を記載します。

指定したFolder内のファイル一覧作成

　globを使用し、指定したFolder内で指定した拡張子を持つファイルの絶対パスリストを作成します。相対バスでも良いですが、後々絶対パスの方が扱いやすいので絶対パスで取得します。

helper.py

def make_file_path_list(path: str, ext: str) -> [str]:
    """ Make file path list from path and ext information
    Parameters
    -----------
    path :str
     project path
    ext :str
     specific the file extention in order to make path list

    Returns
    -----------
    file path list :[str]
     list of file path under the path with the specialied extention
    """
    path_str = path + '*.' + ext
    return glob.glob(path_str)

　次に、得られた絶対パスのリストを元に画像データを読み込みます。PGMファイルであればあまり意味はないですが、一応慣例としてRGBの並びに変更しておきます。OpenCVはDefalt設定でBGRで読み込むので他の処理との整合性を保つためにRGBとしています。

helper.py

def read_img_from_path(filepath: [str], width: int, height: int):
    """ Read image file from path list
    Parameters
    ----------
    filepath :[str]
     File list for image reading
    width :int
     Output image width
    height: int
     Output iamge height

    Returns
    ----------
     Return the read image data as RGB format
    """
    stocker = []
    for path in filepath:
        img = cv2.imread(path)
        img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
        img = cv2.resize(img, dsize=(width, height), interpolation = cv2.INTER_CUBIC)

        stocker.append(img)
    return stocker

　書き出す方もここで実装しておきましょう。引数として書き出す先のパスリストとイメージデーター及び出力Formatを指定しています。特に何も指定しなければJPEGフォーマットで出力されます。

helper.py

def write_img_to_path(outpath: str, filepaths: [str], imgs: [np.ndarray], ext: str='jpg'):
    """ write images to path list
    Parameters
    -----------
    outpath: str
     ouput path
    filepath: [str]
     file path list for file saving
    imgs: [np.ndarray]
     image data
    ext: str
     file format and define extention
    """
    for idx, filepath in enumerate(filepaths):
        # extract file name
        filename = filepath.split('/')[-1]; filename = filename.split('.')[0]

        # make output file path from filename and extention 
        output_file_path = outpath + filename + '.' + ext

        # output images
        cv2.imwrite(output_file_path, imgs[idx])

　後々、Left画像か、Right画像かを判断する必要があるため画像ファイル名からLeftまたはRightを切り出して判断する関数を実装しておきます。Leftの画像であればTrueを返し、RightであればFalseを返します。

helper.py

def loc_detector_from_name(file_name: str) -> bool:
    """ Split location information from file name and return location info
    Parameters
    -----------
    file_name: str
     file name which includes location info such as 'left' or 'right'
    
    Returns
    -----------
    loc: bool
     return location info as boolian, True: 'left', False: 'right'
    """
    loc = file_name.split('_')[-1]; loc = loc.split('.')[0]
    if loc == 'left':
        return True
    
    return False

　後ほど、画像処理で0.0~1.0の範囲に画素値を規格化するので、それ用の関数も定義しておきます。

helper.py

def normarize_img_data(img: np.ndarray, max_value: int = 255):
    """ Normarize image data, find min_value and max_value, and normarize
    img: np.ndarray
     original image data
    max_value: int
     define maximum value after normarization
    """

    # get min, max and range for normarization
    min_value = img.min()
    max_value = img.max()
    value_range = max_value - min_value

    # normalized image
    norm_img = (img - min_value) / value_range * max_value

    # return
    return norm_img.astype(np.uint8)

以上でhalperの準備は終わりとなります。