Coin-Counter/main.py at main · JustA3DObject/Coin-Counter · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
import cv2
import cvzone
import requests
import numpy as np
import imutils
from cvzone.ColorModule import ColorFinder

# Replace the below URL with your own. Make sure to add "/shot.jpg" at last.
url = "http://192.168.1.5:8080//shot.jpg"

totalMoney = 0

myColorFinder = ColorFinder(False)
hsvVals = {'hmin': 0, 'smin': 47, 'vmin': 49, 'hmax': 32, 'smax': 209, 'vmax': 255}

# def empty(a):
#     pass
# cv2.namedWindow("Settings")
# cv2.resizeWindow("Settings", 640,240)
# cv2.createTrackbar("Threshold1", "Settings", 111, 255, empty)
# cv2.createTrackbar("Threshold2", "Settings", 255, 255, empty)

def preProcessing(img):

    imgPre = cv2.addWeighted(img, 1.2, np.zeros(img.shape, img.dtype), 0, 45)
    imgPre = cv2.GaussianBlur(imgPre, (5,5), 5)
    # threshold1 = cv2.getTrackbarPos("Threshold1", "Settings")
    # threshold2 = cv2.getTrackbarPos("Threshold2", "Settings")
    imgPre = cv2.Canny(imgPre, 60,130)
    kernel = np.ones((4,4), np.uint8)
    imgPre = cv2.dilate(imgPre, kernel, iterations=1)
    imgPre = cv2.morphologyEx(imgPre, cv2.MORPH_CLOSE, kernel)

    return imgPre

# While loop to continuously fetching data from the Url
while True:
    img_resp = requests.get(url)
    img_arr = np.array(bytearray(img_resp.content), dtype=np.uint8)
    img = cv2.imdecode(img_arr, -1)

    img = imutils.resize(img, width=800, height=600)
    imgPre = preProcessing(img)
    imgContours, conFound = cvzone.findContours(img, imgPre, minArea=20)

    totalMoney = 0
    if conFound:
        for contour in conFound:
            peri = cv2.arcLength(contour['cnt'], True)
            approx = cv2.approxPolyDP(contour['cnt'], 0.02*peri, True)

            if len(approx) > 6:
                area = contour['area']

                # print(area)
                x,y,w,h = contour['bbox']
                imgCrop = img[y:y+h, x:x+w]
                # cv2.imshow(str(count), imgCrop)
                imgColor, mask = myColorFinder.update(imgCrop, hsvVals)
                whitePixelCount = cv2.countNonZero(mask)
                # print(whitePixelCount)

                if 8000 < area < 12000 and whitePixelCount > 500:
                    totalMoney += 5
                elif area < 10600:
                    totalMoney += 1
                elif 10600 < area < 12000:
                    totalMoney += 2
                elif area > 12000:
                    totalMoney += 10
            # print(totalMoney)

    # stackedImage = cvzone.stackImages([img, imgPre, imgContours],2,0.5)
    cvzone.putTextRect(imgContours, f'Rs. {totalMoney}', (50,50))
    cv2.imshow("Image", imgContours)
    # cv2.imshow("ImageColor", imgColor)

    # Press Esc key to exit
    if cv2.waitKey(1) == 27:
        break
cv2.destroyAllWindows()