파이썬 색 인식 프로젝트 관련 질문입니다.

Question

제가 움직이는(회전하는) 물체를 영상으로 촬영해서(실시간 인식이 아닌 미리 촬영본을 녹화하는 방식도 됩니다!)빨간색 점이 인식될 경우(빨간색 점이 회전에 따라 보였다 안보였다 함)카운트를 하나씩 올려 최종적으로 빨간색 점이 몇번이나 인식됬는지(회전됬는지)결과를 내주는 프로젝트를 하고 있는데 아래와 같이 코드를 작성하여서 웹캠으로 이미지를 촬영한 다음 빨간색 부분만 남기는 부분까진 성공했는데, 영상에서 빨간색이 인식될 경우 어떻게 카운트를 올려야 하는지 모르겠어 이렇게 글을 남깁니다.. 고수님들 도와주세요!!아래는 지금까지 작성된 코드입니다.# 라이브러리 다운로드 # pip install opencv-python # pip install numpy import cv2 import numpy as np import matplotlib.pyplot as plt import extcolors import PIL cap &#61; cv2.VideoCapture(1) # 카메라 모듈 사용. w &#61; round(cap.get(cv2.CAP_PROP_FRAME_WIDTH)) h &#61; round(cap.get(cv2.CAP_PROP_FRAME_HEIGHT)) fps &#61; cap.get(cv2.CAP_PROP_FPS) # 카메라에 따라 값이 정상적, 비정상적 # fourcc 값 받아오기, *는 문자를 풀어쓰는 방식, *'DIVX' &#61;&#61; 'D', 'I', 'V', 'X' fourcc &#61; cv2.VideoWriter_fourcc(*'DIVX') delay &#61; round(1000/fps) out &#61; cv2.VideoWriter('output.avi', fourcc, fps, (w, h)) #colors, pixel_count &#61; extcolors.extract_from_image(cap) while(1): ret, frame &#61; cap.read() # 카메라 모듈 연속프레임 읽기 hsv &#61; cv2.cvtColor(frame, cv2.COLOR_BGR2HSV) # BGR을 HSV로 변환해줌 inversed &#61; ~frame # define range of blue color in HSV lower_blue &#61; np.array([100,100,120]) # 파랑색 범위 upper_blue &#61; np.array([150,255,255]) lower_green &#61; np.array([50, 150, 50]) # 초록색 범위 upper_green &#61; np.array([80, 255, 255]) lower_red &#61; np.array([0, 63, 74]) # 빨강색 범위 upper_red &#61; np.array([20, 255, 255]) # Threshold the HSV image to get only blue colors mask &#61; cv2.inRange(hsv, lower_blue, upper_blue) # 110<->150 Hue(색상) 영역을 지정. mask1 &#61; cv2.inRange(hsv, lower_green, upper_green) # 영역 이하는 모두 날림 검정. 그 이상은 모두 흰색 두개로 Mask를 씌움. mask2 &#61; cv2.inRange(hsv, lower_red, upper_red) # Bitwise-AND mask and original image res &#61; cv2.bitwise_and(frame, frame, mask&#61;mask) # 흰색 영역에 파랑색 마스크를 씌워줌. res1 &#61; cv2.bitwise_and(frame, frame, mask&#61;mask1) # 흰색 영역에 초록색 마스크를 씌워줌. res2 &#61; cv2.bitwise_and(frame, frame, mask&#61;mask2) # 흰색 영역에 빨강색 마스크를 씌워줌. cv2.imshow('frame',frame) # 원본 영상을 보여줌 cv2.imshow('Blue', res) # 마스크 위에 파랑색을 씌운 것을 보여줌. cv2.imshow('Green', res1) # 마스크 위에 초록색을 씌운 것을 보여줌. cv2.imshow('red', res2) # 마스크 위에 빨강색을 씌운 것을 보여줌. preframe &#61; res2 out.write(preframe) k &#61; cv2.waitKey(5) & 0xFF if k &#61;&#61; 27: break cap.release() out.release() cv2.destroyAllWindows()

페코옹 · Answer

빨간점을 트랙킹 해줘야죠빨간점이 발생했으면 다음 프레임과의 이동 위치가 특정 범위 이내다 라면 같은 빨간점으로 인식을 하는거죠.예를 들어빨간점이 보이면 트랙커 생성다음 프레임에서 트랙커와 특정값 이내에 있는 빨간점이 있다면 카운팅 안 하고 트랙커 현재 위치 업데이트다음 프레임에서 트랙커와 특정값 이내에 있는 빨간점이 없다면 트랙커 제거이거를 매 프레임마다 진행하시고 빨간점개수 만큼 n번씩 진행하면 될듯합니다.

파이썬 색 인식 프로젝트 관련 질문입니다.

1개의 답변

#opencv 목록