エラー：UnboundLocalError: local variable 'gray' referenced before assignment を解決したいです。

Question

以下のエラーの改善がわかりません。
UnboundLocalError: local variable 'gray' referenced before assignment

test2というフォルダにチェスボードを撮影した画像を10枚いれています。
最近始めたばかりでわからないことが多いのでご教授いただきたいです。よろしくお願いします。

以下のサイトは見ましたがよくわかりませんでした。
[https://teratail.com/questions/149453
]

import numpy as np
import cv2
import glob

def Calibration(path):
    # termination criteria
    criteria = (cv2.TERM_CRITERIA_EPS + cv2.TERM_CRITERIA_MAX_ITER, 30, 0.001)

    # prepare object points, like (0,0,0), (1,0,0), (2,0,0) ....,(6,5,0)
    objp = np.zeros((6 * 8, 3), np.float32)
    objp[:, :2] = np.mgrid[0:8, 0:6].T.reshape(-1, 2)

    # Arrays to store object points and image points from all the images.
    objpoints = []  # 3d point in real world space
    imgpoints = []  # 2d points in image plane.

    images = sorted(glob.glob(path + '*.jpg'))

    i = 1
    for fname in images:
        img = cv2.imread(fname)
        gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

        # Find the chess board corners
        ret, corners = cv2.findChessboardCorners(gray, (8, 6), flags=cv2.CALIB_CB_FAST_CHECK)
        print(ret)
        # If found, add object points, image points (after refining them)
        if ret == True:
            objpoints.append(objp)

            corners2 = cv2.cornerSubPix(gray, corners, (11, 11), (-1, -1), criteria)
            imgpoints.append(corners2)

            # Draw and display the corners
            img = cv2.drawChessboardCorners(img, (8, 6), corners2, ret)
            height = img.shape[0]
            width = img.shape[1]
            # cv2.imshow('img',img)

            cv2.waitKey(500)
        i = i + 1
    cv2.destroyAllWindows()

    ret, mtx, dist, rvecs, tvecs = cv2.calibrateCamera(objpoints, imgpoints, gray.shape[::-1], None, None)
    i = 0
    for fname in images:
        img = cv2.imread(fname)
        h, w = img.shape[:2]
        newcameramtx, roi = cv2.getOptimalNewCameraMatrix(mtx, dist, (w, h), 1, (w, h))
        # undistort

        dst = cv2.undistort(img, mtx, dist, None, newcameramtx)
        # crop the image
        x, y, w, h = roi
        dst = dst[y:y + h, x:x + w]
        cv2.imwrite('calibration_result/calibresult' + str(i)+ '.jpg', dst)
        i = i + 1
    print("A=\n{0}".format(newcameramtx))
    print("R=\n{0}".format(rvecs[0]))
    print("t=\n{0}".format(tvecs[0]))
    R_t = np.hstack((rvecs, tvecs))
    print("[R|t]={0}".format(R_t))


    np.save('camera_data/camera_matrix', newcameramtx)
    np.save('camera_data/distCoeffs', dist)
    np.save('camera_data/roi',roi)

    return newcameramtx

def video_split(path):
    i = 0
    cap = cv2.VideoCapture(path)
    while (cap.isOpened()):
        flag, frame = cap.read()  # Capture frame-by-frame
        if flag == False:  # Is a frame left?
            break
        if i % 50 ==0:
            cv2.imwrite("calb_test/test2/" + str(i) + ".jpg", frame)  # Save a frame
            print("test" + str(i) + ".jpg")
        i += 1

    cap.release()  # When everything done, release the capture
#
if __name__ == '__main__':
    # video_split('calb_test/test.mov')
    A = Calibration('calb_test/test2/')
    print(A)

Traceback (most recent call last):

File "C:/Users/tampa/PycharmProjects/camera-calib/Camera_Calibration.py", line 134, in <module>
A = Calibration('calb_test/test2/')

File "C:/Users/tampa/PycharmProjects/camera-calib/Camera_Calibration.py", line 91, in Calibration
ret, mtx, dist, rvecs, tvecs = cv2.calibrateCamera(objpoints, imgpoints, gray.shape[::-1], None, None)
UnboundLocalError: local variable 'gray' referenced before assignment

Comments

[hayataka2049]

エラーメッセージ全文を追記してください（tracebackの先頭から）。

[cherei]

Traceback (most recent call last):
File "C:/Users/tampa/PycharmProjects/camera-calib/Camera_Calibration.py", line 134, in <module>
A = Calibration('calb_test/test2/')
File "C:/Users/tampa/PycharmProjects/camera-calib/Camera_Calibration.py", line 91, in Calibration
ret, mtx, dist, rvecs, tvecs = cv2.calibrateCamera(objpoints, imgpoints, gray.shape[::-1], None, None)
UnboundLocalError: local variable 'gray' referenced before assignment

[cherei]

返信遅れてすみません。上記の通りになります。よろしくお願いいたします。

[hayataka2049]

質問文を編集して追記してください(この欄に書かれても見え辛いため)

[hayataka2049]

エラーもコードブロックにてお願いします。

[cherei]

コードブロックというのがわかりません。すみません。これでよろしいですか。

Answer 1

UnboundLocalErrorという例外は、対象のローカル変数に対して代入が行われる前に参照が行われた……ということを意味しています。

    for fname in images:
        img = cv2.imread(fname)
        gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

の部分（22行目）で変数grayに対しては代入が行われているはずですが、forループが一回も実行されなければ質問文の状況になるかと思います。さしあたってはimagesがどうなっているのかを確認すると良いのではないでしょうか。

Comments

[cherei]

if __name__ == '__main__':
A = Calibration('calb_test/test2/')
最後のここの部分でpathを通していると思っているのですが、それは間違いですか

[cherei]

間の部分をコメントアウトして実行してみました。imagesには何も入っていないらしいです。

[hayataka2049]

一般的に「pathを通す」と言われる操作とは無関係です。質問文のコードで得られているimagesは、glob.glob("calb_test/test2/*.jpg")を実行した結果のはずですが、この結果が空のリストになることは十分ありえます（その動作を望まないのであれば修正する必要がありますが、どのように修正したら良いのかは私にはわかりません。情報が限られていて、お答えできません。ヒントになり得ることを書いておくと、一般的なファイル探索は実行時カレントディレクトリ基準の相対パスになるので、これに関連して何かが間違っている可能性はあるでしょう。絶対パスにすればこの部分に関しては上手くいく可能性は蓋然的には高いです）。

[cherei]

参照するファイル名が間違っていました。。。申し訳ありません。
上のエラーは消えましたが、また違ったエラーが出てきたのでまずは自分で考えてみます。
ありがとうございました。

[cherei]

出来ました！！！！！ご丁寧にありがとうございます！！！！