OpenCV питон для подсчета яиц - PullRequest
Новая
       9

OpenCV питон для подсчета яиц

0 голосов
/ 17 сентября 2018

Я сделал подсчет яиц, используя openCV и python, и я получил помощь здесь Обнаружение яиц 

while True:

 (grabbed, frame) = cap.read()  
 hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
 th, bw = cv2.threshold(hsv[:, :, 2], 0, 255, cv2.THRESH_BINARY | cv2.THRESH_OTSU)
 kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (3, 3))
 morph = cv2.morphologyEx(bw, cv2.MORPH_CLOSE, kernel)
 dist = cv2.distanceTransform(morph, cv2.DIST_L2, cv2.DIST_MASK_PRECISE)
 .....

И я получаю координаты и рисую эллипс на яйце. 

x, y, w, h = cv2.boundingRect(contours[i])
_, mx, _, mxloc = cv2.minMaxLoc(dist[y:y+h, x:x+w], peaks8u[y:y+h, x:x+w])
cv2.circle(im, (int(mxloc[0]+x), int(mxloc[1]+y)), int(mx), (255, 0, 0), 2)
cv2.rectangle(im, (x, y), (x+w, y+h), (0, 255, 255), 2)
cv2.drawContours(im, contours, i, (0, 0, 255), 2)

Я создал линию и считаю яйца по одному. 

cv2.line(frame40, (0,coordYEntranceLine), (width,coordYEntranceLine), (255, 0, 0), 2)

def CheckEntranceLineCrossing(coordYContour, coordYEntranceLine):
   absDistance = abs(coordYContour - coordYEntranceLine)

   if ((coordYContour >= coordYEntranceLine) and (absDistance <= 3)):
     return 1
   else:
     return 0

if CheckEntranceLineCrossing(coordYContour, coordYEntranceLine, area):
   eggCount += 1

Проблема начинается здесь. В соответствии с логикой, если яйцо проходит за линию и расстояние <3 считается, но конвейерная лента останавливается, а расстояние <3 снова считает яйцо, и будет восприниматься как слишком много яиц. Я хочу не обнаруживать событие обнаруженное яйцо снова. </p>

Весь код примерно так: 

def CheckEntranceLineCrossing(coordYContour, coordYEntranceLine):
   absDistance = abs(coordYContour - coordYEntranceLine)

   if ((coordYContour >= coordYEntranceLine) and (absDistance <= 3)):
      return 1
   else:
      return 0

def getDistance(coordYEgg1,coordYEgg2):
   dist = abs(coordYEgg1 - coordYEgg2)

   return dist

cap = cv2.VideoCapture('20180910_144521.mp4')

while True:

(grabbed, frame) = cap.read()

hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
th, bw = cv2.threshold(hsv[:, :, 2], 0, 255, cv2.THRESH_BINARY | cv2.THRESH_OTSU)
kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (3, 3))
......

flag = False
egg_list = [[]]
egg_index = 0

for i in range(len(contours)):
    (x, y, w, h) = cv2.boundingRect(contours[i])
    #_, mx, _, mxloc = cv2.minMaxLoc(dist[y:y+h, x:x+w], peaks8u[y:y+h, x:x+w])

    egg_list.append([x, y, flag])

    for i in range(len(egg_list)):
        egg_index = i

        egg_new_X = x
        egg_new_Y = y

        if len(egg_list[egg_index]) >= 1:
            dist = getDistance(egg_new_Y, egg_list[egg_index][1])

            if dist > 50:
                egg_list.append([egg_new_X, egg_new_Y, flag])




    if CheckEntranceLineCrossing(egg_list[i][1], coordYEntranceLine) and not egg_list[i][2]:
        eggCount += 1
        egg_list[i][2] = True

Есть ли способ, который вы можете предложить в этом отношении? У меня есть возможность поместить обнаруженный контур в массив и управлять им или чем-то еще?

1 Ответ

0 голосов
/ 17 сентября 2018

Вы должны отслеживать яйца в каждом кадре.
1) Допустим, у вас в кадре 5 яиц. Сохраните положение и флаг яиц в массиве egg_list. 

flag = False
egg_list = [[]]
for contour in contours:
   ellipse = cv2.fitEllipse(contour)
   (x, y, w, h) = cv2.boundingRect(contour)
   egg_list.append([x, y , flag])

2) Затем найдите все яйца во втором кадре и сравните их с egg_list. если расстояние ниже некоторого предопределенного значения, считайте его одним и тем же яйцом. Еще добавьте новое яйцо в список яиц. 

 for contour in contours:
 ellipse = cv2.fitEllipse(contour)
       (x, y, w, h) = cv2.boundingRect(contour)

       ....       

       for i in egg_list:
           dist = getdist(egg_new,egg_list[i])
           if dist > dist_thresh :
              egg_list.append([egg_new[0],egg_new[1],flag])

3) Когда яйцо пересекает линию, отметьте это яйцо флажком в egg_list и увеличьте счетчик. Затем, когда будет обнаружено, что это же яйцо пересекает линию, вы можете проигнорировать его. 

egg_index = 0
for contour in contours:
   if CheckEntranceLineCrossing(egg_list[egg_index,2], coordYEntranceLine) & ~egg_list[egg_index,2]:
      eggCount += 1
      egg_list[egg_list,2] = true;

Отслеживая все яйца в списке, вы сможете сосчитать только те яйца, которые имеют значение.

...