ボールトラッキング ball tracking

Question

#質問
私はボールトラッキングをしようとしています。ここのプログラム実行させると私が求めていたものに非常に近いので使いたいのですが、初めの方に書いてあるap = argparse.ArgumentParser()からの６行、pts = deque(maxlen=args["buffer"])の意味がよくわからず、while文の中でもそれが使われているところがあり理解できません。指定した色の範囲以外の色を黒くしてしまい、そこから中心を求め、円を描画しているのだろうと思います。基本的な部分かもしれませんが教えていただきたいです。
あとteratailで質問を書く際にソースコードに行の番号を書きいれることはできますか？(地道に書き入れる以外の方法です。)宜しくお願いします。
#該当のソースコード

from collections import deque
import numpy as np
import argparse

import cv2

# construct the argument parse and parse the arguments
ap = argparse.ArgumentParser()
ap.add_argument("-v", "--video",
    help="path to the (optional) video file")
ap.add_argument("-b", "--buffer", type=int, default=64,
    help="max buffer size")
args = vars(ap.parse_args())

# define the lower and upper boundaries of the "green" 
# ball in the HSV color space, then initialize the
# list of tracked points
greenLower = (29, 86, 6)
greenUpper = (64, 255, 255)
pts = deque(maxlen=args["buffer"])

# if a video path was not supplied, grab the reference
# to the webcam
if not args.get("video", False):
    camera = cv2.VideoCapture(0)
# otherwise, grab a reference to the video file

else:
    camera = cv2.VideoCapture(args["video"])

camera.set(3,320)
camera.set(4,240)
# keep looping
while True:
    # grab the current frame
    (grabbed, frame) = camera.read()

    # if we are viewing a video and we did not grab a frame,
    # then we have reached the end of the video
    if args.get("video") and not grabbed:
        break

    # resize the frame, blur it, and convert it to the HSV
    # color space
    #frame = imutils.resize(frame, width=600)
    # blurred = cv2.GaussianBlur(frame, (11, 11), 0)
    hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)

    # construct a mask for the color "green", then perform
    # a series of dilations and erosions to remove any small
    # blobs left in the mask
    mask = cv2.inRange(hsv, greenLower, greenUpper)
    mask = cv2.erode(mask, None, iterations=2)
    mask = cv2.dilate(mask, None, iterations=2)

    # find contours in the mask and initialize the current
    # (x, y) center of the ball
    cnts = cv2.findContours(mask.copy(), cv2.RETR_EXTERNAL,
                            cv2.CHAIN_APPROX_SIMPLE)[-2]
    center = None

    # only proceed if at least one contour was found
    if len(cnts) > 0:
        # find the largest contour in the mask, then use
        # it to compute the minimum enclosing circle and
        # centroid
        c = max(cnts, key=cv2.contourArea)
        ((x, y), radius) = cv2.minEnclosingCircle(c)
        M = cv2.moments(c)
        center = (int(M["m10"] / M["m00"]), int(M["m01"] / M["m00"]))

        # only proceed if the radius meets a minimum size
        if radius > 10:
            # draw the circle and centroid on the frame,
            # then update the list of tracked points
            cv2.circle(frame, (int(x), int(y)), int(radius),
                       (0, 255, 255), 2)
            cv2.circle(frame, center, 5, (0, 0, 255), -1)

    # update the points queue
    pts.appendleft(center)

    # loop over the set of tracked points
    for i in xrange(1, len(pts)):
        # if either of the tracked points are None, ignore
        # them
        if pts[i - 1] is None or pts[i] is None:
            continue

        # otherwise, compute the thickness of the line and
        # draw the connecting lines
        thickness = int(np.sqrt(args["buffer"] / float(i + 1)) * 2.5)
        cv2.line(frame, pts[i - 1], pts[i], (0, 0, 255), thickness)

    # show the frame to our screen
    cv2.imshow("Frame", frame)
    key = cv2.waitKey(1) & 0xFF

    # if the 'q' key is pressed, stop the loop
    if key == ord("q"):
        break

# cleanup the camera and close any open windows
camera.release()
cv2.destroyAllWindows()

Answer 1

フレームごとに球の検出と重心の計算をしてキューに保存し、描画するプログラムのようですね。

---

キューについて
どこまで理解されているかわからないので、とりあえずここから。
キューはデータ構造の一つで、**古いデータから順に抜き取る(FIFO)**ときに便利です。
このプログラムでは、一定以上古い重心は捨てているようで、そのためにキューが用いられています。
詳しくはググった方が早いです。

argparseについて
コマンドライン引数を処理している部分です。

• buffer： キューの長さ(どれくらい古い重心まで残すか)。デフォルトで64。
• video ： ビデオか／Webカメラか。デフォルトで後者。

ビデオに対する処理が出来るのは便利ですね。
ap.add_argumentなどのメソッドは、上記を設定している部分になります。

pts = deque(maxlen=args["buffer"])
ptsに、コマンドライン引数bufferで指定された長さのキューのインスタンスを代入しています。

処理の流れについて
かなり詳しくコメントが入っているコードなので、そちらをしっかり読むことをおすすめします。
まあ、基本的な流れとしては、以下の感じですかね。
0. 緑色の部分をくりぬいてノイズ除去
0. 輪郭を検出
0. 一番大きな輪郭の重心を計算
0. ある程度輪郭の半径があれば、キューに書き出し
0. キューの中の重心を順に描画

また、以下の部分は、単に見た目をそれっぽくしているだけです。

thickness = int(np.sqrt(args["buffer"] / float(i + 1)) * 2.5)

thicknessに適当に定数値を代入してみると意味がわかるかと思います。

---

コードに番号を付けられるか？
すみません、私は知らないです。
ただ、今回のように、githubへのリンクが張ってあるのであれば、不要な気もします。

Comments

[退会済みユーザー]

ありがとうございます。

Answer 2

argparseはコマンドライン引数を簡単に扱えるモジュールです。
python video_ball_tracking.py -b 10からargs["buffer"]で10という値が取り出せます。

またptsはdequeですが、ざっとソース見る限り、直近"buffer"個分のトラッキング位置を保持し、これらの（履歴）点をつないで軌跡として描画しているようです。

あとteratailで質問を書く際にソースコードに行の番号を書きいれることはできますか？(地道に書き入れる以外の方法です。)宜しくお願いします。

ヘルプを見る限り、できないようです。
ただ、質問者様のように完全なコードを提示していれば、各自がエディタにコードコピーすれば行番号は分かるので行番号を書き入れる必要性はあまりないかと思います。

Comments

[退会済みユーザー]

ありがとうございます。