前提・実現したいこと
pythonでモザイクアートを生成するプログラムを作っています.
compareHist()でカラーヒストグラムを比較し、結果をListに格納していくところで
エラーが出てしまいました.
発生している問題・エラーメッセージ
Traceback (most recent call last): File "mosaic.py", line 111, in <module> Target.paste(src_hist_dict[result[0][1]][0],(tileRect[1],tileRect[2])) IndexError: list index out of range
該当のソースコード
python
# coding:utf-8 from PIL import Image,ImageFilter import subprocess import os import random import math import time import numpy as np import glob import cv2 #素材画像の入っているフォルダです↓ SourceDir = R"xxxxx" #目標となる画像↓ GoalImage = R"xxxxxx.jpg" #出力の名前 OutputImage="output.jpg" SourceImageSize=(60,40) #ターゲット画像の倍率 TargetZoom=11 # 画像使用制限 used_count = 3 def __create_tile(org_image,height,width): w, h = org_image.size RectList=[] for y in range(0, math.floor(h / height) + 1): for x in range(0, math.floor(w / width) + 1): height2 = y * height width2 = x * width crap_img = org_image.crop((width2, height2, width2 + width, height2 + height)) # crap_img.show() RectList.append((crap_img,width2,height2)) return RectList def __clac_hist(img): hist_list = [] color = ['r','g','b'] images = np.asarray(img) for i in enumerate(color): hist_list.append(cv2.calcHist([images],[i[0]],None,[256],[0,256])) return hist_list if __name__=="__main__": startTime=time.time() # 目標画像を開きリサイズ target=Image.open(GoalImage) target=target.resize(tuple(math.floor(i*TargetZoom) for i in target.size)) # 元画像のタイル化 print("%s:タイルリストを生成開始"%(str(time.time()-startTime))) tiles = __create_tile(target,SourceImageSize[1],SourceImageSize[0]) print("%s:タイルリストを生成完了"%(str(time.time()-startTime))) # 素材画像ファイル名を取得 # 全素材画像をリサイズする file_paths = glob.glob(SourceDir+"\*") src_hist_dict = {} print("%s:全素材画像ヒストグラムを計算開始"%(str(time.time()-startTime))) for file_path in file_paths: file_name = file_path.split("\")[-1] try: src_image = Image.open(file_path).resize(SourceImageSize) except Exception: continue # src_image.show() src_hist_dict[file_name] = [src_image,__clac_hist(src_image),0] print("%s:全素材画像ヒストグラムを計算完了"%(str(time.time()-startTime))) # カラーヒストグラム計算 素材画像 Target=Image.new("RGB",target.size,255) # タイルの数だけループ print("%s:モザイクアート生成開始"%(str(time.time()-startTime))) while(len(tiles) > 0): result = [] print("残りタイル:" + str(len(tiles))) r=random.randrange(len(tiles)) # タイルの中からランダムで1つ選択 tileRect=tiles[r] del tiles[r] # タイルのカラーヒスとグラムを取得する。 tile_hist = __clac_hist(tileRect[0]) # RGBを同じベクトルにする tile_hist = np.array(tile_hist) tile_hist = tile_hist.reshape(tile_hist.shape[0] * tile_hist.shape[1], 1) # 元画像のタイルと素材のヒストグラムを比較 for file_name,src_hist in src_hist_dict.items(): src_hist = np.array(src_hist[1]) src_hist = src_hist.reshape(src_hist.shape[0] * src_hist.shape[1], 1) d = cv2.compareHist(tile_hist, src_hist, cv2.HISTCMP_INTERSECT) result.append([d,file_name]) result.sort(reverse=True) # ソート # ファイルの置き換え Target.paste(src_hist_dict[result[0][1]][0],(tileRect[1],tileRect[2])) # 画像使用カウントアップ src_hist_dict[result[0][1]][2] = src_hist_dict[result[0][1]][2] + 1 if src_hist_dict[result[0][1]][2] == used_count: del src_hist_dict[result[0][1]] print("%s:モザイクアート生成完了"%(str(time.time()-startTime))) Target.save(OutputImage) # v.release()
試したこと
エラーが出る文の前にtileRect.append(2)
としてみましたができませんでした.
補足情報(FW/ツールのバージョンなど)
OSはWindows10です.
Python3.7.3です.
まだ回答がついていません
会員登録して回答してみよう