前提・実現したいこと
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
1# coding:utf-8 2from PIL import Image,ImageFilter 3import subprocess 4import os 5import random 6import math 7import time 8import numpy as np 9import glob 10import cv2 11 12#素材画像の入っているフォルダです↓ 13SourceDir = R"xxxxx" 14 15#目標となる画像↓ 16GoalImage = R"xxxxxx.jpg" 17 18#出力の名前 19OutputImage="output.jpg" 20 21SourceImageSize=(60,40) 22 23#ターゲット画像の倍率 24TargetZoom=11 25 26# 画像使用制限 27used_count = 3 28 29def __create_tile(org_image,height,width): 30 31 32 w, h = org_image.size 33 RectList=[] 34 for y in range(0, math.floor(h / height) + 1): 35 for x in range(0, math.floor(w / width) + 1): 36 height2 = y * height 37 width2 = x * width 38 crap_img = org_image.crop((width2, height2, width2 + width, height2 + height)) 39 # crap_img.show() 40 RectList.append((crap_img,width2,height2)) 41 return RectList 42 43def __clac_hist(img): 44 45 hist_list = [] 46 color = ['r','g','b'] 47 images = np.asarray(img) 48 49 for i in enumerate(color): 50 hist_list.append(cv2.calcHist([images],[i[0]],None,[256],[0,256])) 51 return hist_list 52 53if __name__=="__main__": 54 55 startTime=time.time() 56 57 # 目標画像を開きリサイズ 58 target=Image.open(GoalImage) 59 target=target.resize(tuple(math.floor(i*TargetZoom) for i in target.size)) 60 61 # 元画像のタイル化 62 print("%s:タイルリストを生成開始"%(str(time.time()-startTime))) 63 tiles = __create_tile(target,SourceImageSize[1],SourceImageSize[0]) 64 print("%s:タイルリストを生成完了"%(str(time.time()-startTime))) 65 66 # 素材画像ファイル名を取得 67 # 全素材画像をリサイズする 68 file_paths = glob.glob(SourceDir+"\*") 69 src_hist_dict = {} 70 print("%s:全素材画像ヒストグラムを計算開始"%(str(time.time()-startTime))) 71 for file_path in file_paths: 72 file_name = file_path.split("\")[-1] 73 74 try: 75 src_image = Image.open(file_path).resize(SourceImageSize) 76 except Exception: 77 continue 78 # src_image.show() 79 src_hist_dict[file_name] = [src_image,__clac_hist(src_image),0] 80 print("%s:全素材画像ヒストグラムを計算完了"%(str(time.time()-startTime))) 81 82 # カラーヒストグラム計算 素材画像 83 84 Target=Image.new("RGB",target.size,255) 85 86 # タイルの数だけループ 87 print("%s:モザイクアート生成開始"%(str(time.time()-startTime))) 88 while(len(tiles) > 0): 89 result = [] 90 91 print("残りタイル:" + str(len(tiles))) 92 r=random.randrange(len(tiles)) # タイルの中からランダムで1つ選択 93 tileRect=tiles[r] 94 del tiles[r] 95 # タイルのカラーヒスとグラムを取得する。 96 tile_hist = __clac_hist(tileRect[0]) 97 # RGBを同じベクトルにする 98 tile_hist = np.array(tile_hist) 99 tile_hist = tile_hist.reshape(tile_hist.shape[0] * tile_hist.shape[1], 1) 100 101 # 元画像のタイルと素材のヒストグラムを比較 102 for file_name,src_hist in src_hist_dict.items(): 103 src_hist = np.array(src_hist[1]) 104 src_hist = src_hist.reshape(src_hist.shape[0] * src_hist.shape[1], 1) 105 106 d = cv2.compareHist(tile_hist, src_hist, cv2.HISTCMP_INTERSECT) 107 result.append([d,file_name]) 108 result.sort(reverse=True) # ソート 109 110 # ファイルの置き換え 111 Target.paste(src_hist_dict[result[0][1]][0],(tileRect[1],tileRect[2])) 112 113 # 画像使用カウントアップ 114 src_hist_dict[result[0][1]][2] = src_hist_dict[result[0][1]][2] + 1 115 116 if src_hist_dict[result[0][1]][2] == used_count: 117 del src_hist_dict[result[0][1]] 118 119 120 print("%s:モザイクアート生成完了"%(str(time.time()-startTime))) 121 Target.save(OutputImage) 122 # v.release() 123
試したこと
エラーが出る文の前にtileRect.append(2)
としてみましたができませんでした.
補足情報(FW/ツールのバージョンなど)
OSはWindows10です.
Python3.7.3です.
回答3件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。