p_analyze_b_img.py

# -*- coding: utf-8 -*-
# @Time : 2020/4/10 下午4:02
# @Author : LuoLu
# @FileName: p_analyze_b_img.py
# @Software: PyCharm
# @Github ：https://github.com/luolugithub
# @E-mail ：argluolu@gmail.com
import cv2
import numpy

frame = cv2.imread('/home/luolu/PycharmProjects/ParticleDetection/data/image/thresh_pills_02.png')

if frame is None:
    print('Error loading image')
    exit()

frame = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
ret, frame = cv2.threshold(frame, 127, 255, cv2.THRESH_BINARY)

contours, hierarchy = cv2.findContours(frame, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)

areas = []

for i in range(0, len(contours)):
    areas.append(cv2.contourArea(contours[i]))

mass_centres_x = []
mass_centres_y = []

for i in range(0, len(contours)):
    M = cv2.moments(contours[i], 0)
    mass_centres_x.append(int(M['m10']/M['m00']))
    mass_centres_y.append(int(M['m01']/M['m00']))

print('Num particles: ', len(contours))

for i in range(0, len(contours)):
    print('Area', (i + 1), ':', areas[i])

for i in range(0, len(contours)):
    print('Centre', (i + 1), ':', mass_centres_x[i], mass_centres_y[i])

cv2.imshow("Frame", frame)

cv2.waitKey(0)