python opencv 台形補正

Question

前提・実現したいこと

マーカー（円）を検出して台形補正
https://pongsuke.hatenadiary.jp/entry/2017/05/31/160712
のページを参考にして作成していますがエラーで実行ができません。

発生している問題・エラーメッセージ

TypeError
'NoneType' object is not subscriptable

該当のソースコード

    points  = np.array([
        centers [1],
        centers [2],
        centers [3],
        centers [4]])

#!/usr/bin/env python
# -*- coding: utf-8 -*
import cv2
import numpy as np


def main():
    image = cv2.imread("image.JPG")
    #cv2.imshow('original', image)

    height, width ,depth    = image.shape

    # グレースケールに変換
    image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
    # ガウシアンフィルタ
    image = cv2.GaussianBlur(image, (5, 5), 0)
    # 二値変換
    ret, image = cv2.threshold(image,127,255,cv2.THRESH_BINARY)

    # Python: cv2.findCirclesGridDefault(image, patternSize[, centers[, flags]]) → retval, centers
    # CALIB_CB_SYMMETRIC_GRID uses symmetric pattern of circles.
    # CALIB_CB_CLUSTERING uses a special algorithm for grid detection. It is more robust to perspective distortions but much more sensitive to background clutter.
    retval, centers = cv2.findCirclesGrid(image, (2,2), flags=cv2.CALIB_CB_SYMMETRIC_GRID + cv2.CALIB_CB_CLUSTERING)
    if retval:
        print (centers)
        image = cv2.cvtColor(image, cv2.COLOR_GRAY2BGR)
        image = cv2.drawChessboardCorners(image, (2,2), centers, retval)

    # 射影変換のために、並び替えた配列を用意する
    points  = np.array([
        centers [1],
        centers [2],
        centers [3],
        centers [4]])
    print (points)

    # 外接矩形
    bRect = cv2.boundingRect(points)
    print (bRect)
    cv2.rectangle(image, (bRect[0], bRect[1]), (bRect[0]+bRect[2], bRect[1]+bRect[3]), (255,0,0), 2)
    cv2.imshow('drawChessboardCorners', image)
    cv2.imwrite('90.drawChessboardCorners.png', image)

    # 射影変換
    # 右上、左上、左下、右下
    pts1 = np.float32(points)
    pts2 = np.float32([
        [bRect[0]+bRect[2],bRect[1]],
        [bRect[0],bRect[1]],
        [bRect[0],bRect[1]+bRect[3]],
        [bRect[0]+bRect[2],bRect[1]+bRect[3]],
    ])
    print(pts1)
    print(pts2)
    M = cv2.getPerspectiveTransform(pts1,pts2)
    imageOrg = cv2.warpPerspective(imageOrg,M,(width,height))
    cv2.imshow('getPerspectiveTransform', imageOrg)
    cv2.imwrite('90.getPerspectiveTransform.png', imageOrg)

    cv2.waitKey(0)
    cv2.destroyAllWindows()
    return 0

if __name__ == '__main__':
    try:
        main()
    except KeyboardInterrupt:
        pass

試したこと

ネットで調べて修正

Comments

[jbpb0]

pythonのコードの一番最初の行のすぐ上に
```python
だけの行を追加してください
また、pythonのコードの一番最後の行のすぐ下に
```
だけの行を追加してください

または、
https://teratail.storage.googleapis.com/uploads/contributed_images/56957fe805d9d7befa7dba6a98676d2b.gif
を見て、そのようにしてみてください

現状、コードがとても読み辛いです
質問にコードを載せる際に上記をやってくれたら、他人がコードを読みやすくなり、コードの実行による現象確認もやりやすくなるので、回答されやすくなります

[jbpb0]

エラーは、「Traceback」と書いてあるところ以降を省略せずに質問に書いてください
そうしないと、状況がよく分かりません

Answer 1

エラーメッセージが発生している場所の情報が書かれていませんので確実ではありませんが、
cv2でのこういう問題は、image = cv2.imread("image.JPG")で読み込みに失敗している場合が多いです。

import cv2
import numpy as np
import os
print(os.getcwd())
image = cv2.imread("image.JPG")
print(image)

というように、以下の三つの文をして実行してみてください。

import os
print(os.getcwd())
print(image)

print(os.getcwd())で表示されるディレクトリに"image.JPG"が存在しない場合、
print(image)はNoneとなります。

そうなっていた場合は、"image.JPG"をprint(os.getcwd())で表示されるディレクトリにおいてから再度実行してみてください。