IndexError:tuple index out of rangeが発生している理由と解決方法が分かりません

Question

実現したいこと

Python、OpenCVなどを用いて、動画をフレーム間差分法で加工した後にその動画を分割してjpgファイルとして保存しています。そして、その分割した画像を一枚一枚自動でトリミングしようとしているのですがその途中の処理でIndexErrorが発生しました。

私自身、初心者なのでほかの方がネットで公開されているソースコードをもとに作成しています。調べても解決方法がわからなかったため今回はこのように質問させていただきました。

該当のソースコード

import cv2
import numpy as np
from PIL import Image
import os
import shutil

def video_2_frames(video_file, image_dir, image_file):
        # Delete the entire directory tree if it exists.
        if os.path.exists(image_dir):
            shutil.rmtree(image_dir)  
    
        # Make the directory if it doesn't exist.
        if not os.path.exists(image_dir):
            os.makedirs(image_dir)
    
        # Video to frames
        global i
        global j
        global count
        i=0
        j=0
        count=0
        cap = cv2.VideoCapture(video_file)
        fgbg = cv2.bgsegm.createBackgroundSubtractorMOG()
        while(cap.isOpened()):
            flag, frame = cap.read()  # Capture frame-by-frame
            if flag == False:  # Is a frame left?
                break
            if i==30:
                fgmask = fgbg.apply(frame)
                cv2.imwrite(image_dir+image_file % str(j), fgmask)  # Save a frame
                print('Save', image_dir+image_file % str(j))
                i = 0
                j += 1
            i+=1
    
        cap.release()  # When everything done, release the capture
        
video_2_frames('./movie.mp4','./image_dir/','img_%s.jpg')
        
def nparray_to_rgb(nparry:np.array):
    return (int(nparry[0]),int(bg_color[1]),int(bg_color[2]))
    
def find_edge(img_path:str):
    img = cv2.imread(img_path,0)
    blur = cv2.blur(img,(5,5))
    print('blur')
    print(blur)
    edges = cv2.Canny(blur,100,200)
    print('edges')
    print(edges)
    return edges
        
def find_target(edges): 
    results = np.where(edges==255)
    print('results')
    print(results)
    top = np.min(results[0])
    bottom = np.max(results[0]) - 1
    left = np.min(results[1])
    right = np.max(results[1]) - 1
    return (left,top,right,bottom)

        
def to_RGB(image:Image):
    if image.mode == 'RGB': return image
    background = Image.new("RGB", image.size, (255, 255, 255))
    print('background')
    print(background)
    background.paste(image, mask=image.split()[3])  # 3 is the alpha channel
    background.format = image.format
    return background
    
def get_crop_img(img_path:str):
    #im=cv2.imread('./hamasaki_trim/img_'+str(count)+'.png',0)
    edges = find_edge(img_path)
    left,top,right,bottom = find_target(edges)
    rgb_img = to_RGB(Image.open(img_path))
    trim_img = rgb_img.crop((left, top, right, bottom))
    cv2.imwrite('./image_dir/img_'+str(count)+'_trim.jpg',trim_img)
    print('crop_img')

for count_img in range(1,j):
    print('crop_'+str(count_img))
    get_crop_img('./image_dir/img_'+str(count_img)+'.jpg')
    count_img+=1
    count+=1

発生している問題・エラーメッセージ

  line 71, in to_RGB
    background.paste(image, mask=image.split()[3])  # 3 is the alpha channel

IndexError: tuple index out of range

このエラーはどのようにすれば解決できるでしょうか？
また、なぜ発生しているのか教えていただけると幸いです。

補足情報（FW/ツールのバージョンなど）

開発環境(IDE):Spyder(Python3.7)

Answer 1

このエラーはどのようにすれば解決できるでしょうか？

image.split()[3]

ということは、画像のチャンネル数は RGBA の4チャンネルであることが前提となっていますが、おそらく image にグレースケール画像が渡されて、1チャンネルしかないので、エラーになっているのではないでしょうか。
憶測なので、image.mode の値を print して RGBA であることを確認してみてください。

このエラーはどのようにすれば解決できるでしょうか？

「RGBA 画像でない場合は処理をしない」ような例外処理を追加する必要があるでしょう。

Comments

[hamasaki]

image.modeの値をprintしたらLと表示されたのでグレイスケールが渡されていたようです。
なので、to_RGB関数の条件式をif image.mode != 'RGBA': return imageに変更したら通りました…!
ありがとうございました。