python opencv 指定した座標を中心にトリミングができない

Question

実装したいこと

jsonファイルに格納されているopenposeの骨格点の座標から指定した座標の部分を中心にトリミングしたい

現状況

画像から特徴点を算出してその部分のトリミングが行えている。

import cv2
from pathlib import Path
import shutil

# 画像を読み込む。
img = cv2.imread("test16.jpg")


gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

# 2値化する。
ret, binary = cv2.threshold(gray, 0, 255, cv2.THRESH_BINARY + cv2.THRESH_OTSU)

# 膨張処理
kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (5, 5))
binary = cv2.dilate(binary, kernel)

# 輪郭抽出する。
contours, hierarchy = cv2.findContours(
    binary, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE
)

rects = []
for cnt in contours:
    if cv2.contourArea(cnt) < 500:  # ★ ここの無視する面積も要調整
        continue  # 小さい輪郭は無視する。 

    # 回転した外接矩形をとる。
    rect = cv2.minAreaRect(cnt)
    rects.append(rect)


def crop_rect(img, rect):
    center, size, angle = rect
    center = tuple(map(int, center))  # float -> int
    size = tuple(map(int, size))  # float -> int
    h, w = img.shape[:2]  # 画像の高さ、幅

    
    # 切り抜く。
    cropped = cv2.getRectSubPix(img, size, center)

    return cropped


# 保存するディレクトリを作成する。
output_dir = Path("output")
if output_dir.exists():
    shutil.rmtree(output_dir)
output_dir.mkdir()

for i, rect in enumerate(rects):
    # 切り抜く。
    cropped = crop_rect(img, rect)
    # 保存する。
    save_path = output_dir / f"contour{i}.png"
    cv2.imwrite(str(save_path), cropped)

解決したいこと

座標が(x,y)=(2072.9,2159.91)になっておりその座標を中心にトリミングを行いたい。

どうかご教授のほうお願いいたします。

Answer 1

import cv2
from pathlib import Path
import shutil
from matplotlib import pyplot as plt
# 画像を読み込む。
img = cv2.imread("3.png")
newImg=img.copy()
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
plt.imshow(gray,cmap='gray')
# 2値化する。
ret, binary = cv2.threshold(gray, 0, 255, cv2.THRESH_BINARY + cv2.THRESH_OTSU)

# 膨張処理
kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (5, 5))
binary = cv2.dilate(binary, kernel)
plt.imshow(binary,cmap='gray')
# 輪郭抽出する。
contours, hierarchy = cv2.findContours(
    binary, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
# 輪郭を可視化する
cntImg = cv2.drawContours(img.copy(), contours, -1, (0,255,0), 3)
plt.imshow(cntImg)

# make the folder where to save
output_dir = Path("output")
if not output_dir.exists():
       output_dir.mkdir()

index=0
# loop over the coutours
for contour in contours:
    # get the info for bboxes
    x, y, width, height = cv2.boundingRect(contour)
    # extract ROI
    roi = newImg[y:y+height, x:x+width]
    # update the index
    index =index+1
    # visualize the result
    im2=cv2.rectangle(img, (x, y), (x + width, y + height), color=(0, 255, 0), thickness=5)
    # define the path to save
    save_path = output_dir / f"contour{index}.png"
    cv2.imwrite(str(save_path), roi)
    
plt.imshow(im2)

こんにちは、こちらのページなどを参考に作成しました。手元の画像ではうまく切り取った図がoutputという名前のフォルダに保存されています。これをもとにそちらのデータやコードにあわせるとうまく行くと思います。