Q&A
「別々に保存」での具体的な操作手順と、どのタイミングでエラーが発生するかを記載ください。
また、上記のエラーの解決方法に加え「～別々に保存したい」と「白で保存したい」は別の質問ということでよろしいでしょうか？
前提・実現したいこと
エラー解決と保存方法の相談
こちらのソースコードを引用させていただきましたhttps://github.com/opencv/opencv/blob/master/samples/python/grabcut.py
OpenCV 4.1.1でマウス操作で前景を抽出しています。
Draw a rectangle around the object using right mouse button”
「マウスの右ボタンで（inputのウィンドウ内で前景を囲む）長方形を書きましょう。」
とのことなので、右クリックして青い枠で前景を囲みます
“Now press the key ‘n’ a few times until no further change”
「’n’を押すと画像が変化します。」
となっているので”n”をキーボードから入力するとoutputウィンドウに背景が黒塗りになった画像が表示されます。
“key ‘0’ – To select areas of sure background”
「0を押すと背景が選択できます。」
“key ‘1’ – To select areas of sure foreground”
「1を押すと前景が選択できます。」
“key ‘2’ – To select areas of probable background”
「2を押すと背景らしいところが選択できます。」
“key ‘3’ – To select areas of probable foreground”
「3を押すと前景らしいところが選択できます。」
と書いているので微調整を行います。
まず、前景と認識されている芝生を背景にしましょう。
input画面を選択して0を押してマウスの左ボタンで芝生の部分を適当に塗る。
元画像と輪郭取得と前景抽出の3の出力はできたのですが,
保存するnコマンド実行時のelif k == ord('n'): # segment the imageworkCount += 1ここでエラーになります
今の状態だと3つの画像が同じ画像として出力されていますなので
元画像と輪郭画像とマスク画像の3つに分けて保存したいです
発生している問題・エラーメッセージ
エラーメッセージ File "test123.py", line 309, in <module> App().run() File "test123.py", line 255, in run workCount += 1 UnboundLocalError: local variable 'workCount' referenced before assignment
該当のソースコード
from __future__ import print_function import numpy as np import cv2 as cv import sys import os #filename = 'image1 (2).jpeg' class App(): BLUE = [255,0,0] # rectangle color RED = [0,0,255] # PR BG GREEN = [0,255,0] # PR FG BLACK = [0,0,0] # sure BG WHITE = [255,255,255] # sure FG DRAW_BG = {'color' : BLACK, 'val' : 0} DRAW_FG = {'color' : WHITE, 'val' : 1} DRAW_PR_FG = {'color' : GREEN, 'val' : 3} DRAW_PR_BG = {'color' : RED, 'val' : 2} # setting up flags rect = (0,0,1,1) drawing = False # flag for drawing curves rectangle = False # flag for drawing rect rect_over = False # flag to check if rect drawn rect_or_mask = 100 # flag for selecting rect or mask mode value = DRAW_FG # drawing initialized to FG thickness = 3 # brush thickness def onmouse(self, event, x, y, flags, param): # Draw Rectangle if event == cv.EVENT_RBUTTONDOWN: self.rectangle = True self.ix, self.iy = x,y elif event == cv.EVENT_MOUSEMOVE: if self.rectangle == True: self.img = self.img2.copy() cv.rectangle(self.img, (self.ix, self.iy), (x, y), self.BLUE, 2) self.rect = (min(self.ix, x), min(self.iy, y), abs(self.ix - x), abs(self.iy - y)) self.rect_or_mask = 0 elif event == cv.EVENT_RBUTTONUP: self.rectangle = False self.rect_over = True cv.rectangle(self.img, (self.ix, self.iy), (x, y), self.BLUE, 2) self.rect = (min(self.ix, x), min(self.iy, y), abs(self.ix - x), abs(self.iy - y)) self.rect_or_mask = 0 print(" Now press the key 'n' a few times until no further change \n") # draw touchup curves if event == cv.EVENT_LBUTTONDOWN: if self.rect_over == False: print("first draw rectangle \n") else: self.drawing = True cv.circle(self.img, (x,y), self.thickness, self.value['color'], -1) cv.circle(self.mask, (x,y), self.thickness, self.value['val'], -1) elif event == cv.EVENT_MOUSEMOVE: if self.drawing == True: cv.circle(self.img, (x, y), self.thickness, self.value['color'], -1) cv.circle(self.mask, (x, y), self.thickness, self.value['val'], -1) elif event == cv.EVENT_LBUTTONUP: if self.drawing == True: self.drawing = False cv.circle(self.img, (x, y), self.thickness, self.value['color'], -1) cv.circle(self.mask, (x, y), self.thickness, self.value['val'], -1) def run(self): # Loading images if len(sys.argv) == 2: filename = sys.argv[1] # for drawing purposes else: print("No input image given, so loading default image, lena.jpg \n") print("Correct Usage: python grabcut.py <filename> \n") filename = 'image1 (2).jpeg' if len(sys.argv) == 2: filename = sys.argv[1] # for drawing purposes workDir = filename[:filename.rfind(".")] if not os.path.exists(workDir): os.mkdir(workDir) workCount = 0 self.img = cv.imread(cv.samples.findFile(filename)) self.img2 = self.img.copy() # a copy of original image self.mask = np.zeros(self.img.shape[:2], dtype = np.uint8) # mask initialized to PR_BG self.output = np.zeros(self.img.shape, np.uint8) # output image to be shown # input and output windows cv.namedWindow('output', cv.WINDOW_NORMAL) cv.namedWindow('input', cv.WINDOW_NORMAL) cv.setMouseCallback('input', self.onmouse) #cv.moveWindow('input', self.img.shape[1]+10,90) print(" Instructions: \n") print(" Draw a rectangle around the object using right mouse button \n") while(1): cv.imshow('output', self.output) cv.imshow('input', self.img) k = cv.waitKey(1) # key bindings if k == 27: # esc to exit break elif k == ord('0'): # BG drawing print(" mark background regions with left mouse button \n") self.value = self.DRAW_BG elif k == ord('1'): # FG drawing print(" mark foreground regions with left mouse button \n") self.value = self.DRAW_FG elif k == ord('2'): # PR_BG drawing self.value = self.DRAW_PR_BG elif k == ord('3'): # PR_FG drawing self.value = self.DRAW_PR_FG elif k == ord('s'): # save image bar = np.zeros((self.img.shape[0], 5, 3), np.uint8) res = np.hstack((self.img2, bar, self.img, bar, self.output)) cv.imwrite('grabcut_output.png', res) print(" Result saved as image \n") elif k == ord('r'): # reset everything print("resetting \n") self.rect = (0,0,1,1) self.drawing = False self.rectangle = False self.rect_or_mask = 100 self.rect_over = False self.value = self.DRAW_FG self.img = self.img2.copy() self.mask = np.zeros(self.img.shape[:2], dtype = np.uint8) # mask initialized to PR_BG self.output = np.zeros(self.img.shape, np.uint8) # output image to be shown elif k == ord('n'): # segment the image workCount += 1 print(""" For finer touchups, mark foreground and background after pressing keys 0-3 and again press 'n' \n""") try: if (self.rect_or_mask == 0): # grabcut with rect bgdmodel = np.zeros((1, 65), np.float64) fgdmodel = np.zeros((1, 65), np.float64) cv.grabCut(self.img2, self.mask, self.rect, bgdmodel, fgdmodel, 1, cv.GC_INIT_WITH_RECT) self.rect_or_mask = 1 elif self.rect_or_mask == 1: # grabcut with mask bgdmodel = np.zeros((1, 65), np.float64) fgdmodel = np.zeros((1, 65), np.float64) cv.grabCut(self.img2, self.mask, self.rect, bgdmodel, fgdmodel, 1, cv.GC_INIT_WITH_MASK) except: import traceback traceback.print_exc() mask2 = np.where((self.mask==1) + (self.mask==3), 255, 0).astype('uint8') self.output = cv.bitwise_and(self.img2, self.img2, mask=mask2) print('Done') bgr = cv2.split(img2) bgra = bgr + [mask2] output_alpha = cv2.merge(bgra) cv2.imwrite(work_dir + "\input" + str(work_count) + ".png", img) cv2.imwrite(work_dir + "\output" + str(work_count) + ".png", output_alpha) cv2.imwrite(work_dir + "\mask" + str(work_count) + ".png", mask2) if __name__ == '__main__': print(__doc__) App().run() cv.destroyAllWindows()
試したこと
補足情報（FW/ツールのバージョンなど）
ここにより詳細な情報を記載してください。
回答ありがとうございます。今の状態だと３つの画像が一緒に出力されています。なので３つに分けて別名で保存という形にしたいと思っています。上記のソースコードのos.mkdir(workDir)があります。まずファイルが作られません。
エラーの解決方法と保存方法の相談です。よろしくお願いします。
質問に記載している操作をしてもエラーは再現しませんでした。
まずはこのプログラムの実行手順をもう少し詳細に記載ください。
たとえばプログラムの実行は「python プログラム.py」と実行するのか
または「python プログラム.py 解析したいファイル名」とコマンドライン引数で指定するのかです。
回答ありがとうございます
python test1.pyで実行 右マウスで前景の輪郭指定して、キーボードｎコマンドで背景に合成します。
キーボードのsコマンドで保存しますという流れでやっています。
回答1件
0
エラーの原因だけ。
プログラム起動時に引数としてファイルパスなりを渡している、すなわちlen(sys.argv) == 2場合はworkCountを初期化していますが、引数を渡していないとworkCountが初期化されないままworkCount += 1に到達し、提示エラーが発生しています。修正方法としては、そもそもworkCountは使われていないので削除してよいかと思います。
投稿2020/02/18 11:37
総合スコア38278
あなたの回答
tips
プレビュー
