python
1#coding:utf-8 2from psychopy import visual, core, event, gui, data, misc 3import cv2 4import numpy as np 5import os 6import itertools 7import random 8 9myWin=visual.Window(fullscr=True,monitor='SubMonitor',allowGUI=False,screen=1,units='norm',color=(0,0,0)) 10 11#画像ファイルを読み込み 12curD = os.getcwd() 13os.chdir(os.path.join(curD,'stim')) 14 15r =[1,2,3,4,5]*20 16 17Lst = ['a.png','b.png','c.png','d.png','e.png'] 18 19lst1 = ['1_1.png','1_2.png','1_3.png','1_4.png'] 20lst2 = ['2_1.png','2_2.png','2_3.png','2_4.png'] 21lst3 = ['3_1.png','3_2.png','3_3.png','3_4.png'] 22lst4 = ['4_1.png','4_2.png','4_3.png','4_4.png'] 23lst5 = ['5_1.png','5_2.png','5_3.png','5_4.png'] 24 25def harituke(): 26 y1 = yList[0] 27 y2 = y1+10 28 x1 = xList[0] 29 x2 = x1+15 30 src = cv2.imread(fsLst[i][1], -1) # -1を付けることでアルファチャンネル読み込み 31 dst = cv2.imread(fsLst[i][0]) 32 width, height = src.shape[:2] 33 mask = src[:,:,3] # アルファチャンネルだけ抜き出す。 34 mask1 = cv2.cvtColor(mask, cv2.COLOR_GRAY2BGR) # 3色分に増やす。 35 mask2 = mask1 / 255.0 # 0.0-1.0に変更 36 src1 = src[:,:,:3] 37 dst[y1:y2:, x1:x2] =(dst[y1:y2:, x1:x2])* (1 - mask2) 38 dst[y1:y2:, x1:x2] =(dst[y1:y2:, x1:x2])+(src1 * mask2) # 貼り付ける方の画像に透過率をかけて加算。 39 a = cv2.imwrite('f.png', dst) 40 41for i in range(100): 42 43 yList = [100,200] 44 xList = [100,200,300] 45 np.random.shuffle(xList) 46 np.random.shuffle(yList) 47 48 49 if r[i] == 1: 50 fsLst = list(itertools.product(Lst,lst1)) 51 52 elif r[i] == 2: 53 fsLst = list(itertools.product(Lst,lst2)) 54 55 elif r[i] == 3: 56 fsLst = list(itertools.product(Lst,lst3)) 57 58 elif r[i] == 4: 59 fsLst = list(itertools.product(Lst,lst4)) 60 61 elif r[i] == 5: 62 fsLst = list(itertools.product(Lst,lst5)) 63 64 65 harituke() 66 67 cardImagei = visual.ImageStim(win=myWin,image='f.png',units='norm') 68 cardImagei.draw() 69 myWin.flip() 70 core.wait(5) 71 72 myText = visual.ImageStim(win=myWin, image = black.png,pos=(0,0),units='norm') 73 myText.draw() 74 myWin.flip() 75 core.wait(1) 76
上記のプログラムをfor文で100回まわそうとすると,
src = cv2.imread(fsLst[i][1], -1) # -1を付けることでアルファチャンネル読み込み IndexError: list index out of range
というエラーが出ます.
画像の組み合わせがLstとlst1で20通りなので,lst2,lst3,lst4,lst5の5つから全組み合わせが100通りあるはずなので100回まわると考えているのですが,上記エラーの改善ができません.
ご教授願います.
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。
2019/11/25 06:20 編集