mirror of
https://github.com/fishyboteso/fishyboteso.git
synced 2024-08-30 18:32:13 +00:00
66 lines
1.5 KiB
Python
66 lines
1.5 KiB
Python
from systems.fishing_event import *
|
|
|
|
def GetKeypointFromImage(img):
|
|
"""
|
|
convert image int hsv
|
|
creates a mask for brown color
|
|
uses blob detection to find a blob in the mask
|
|
filter the blobs to find the correct one
|
|
|
|
:param img: rgb image
|
|
:return: location of the pixel which is used to detect different fishing states
|
|
"""
|
|
|
|
# Setup SimpleBlobDetector parameters.
|
|
hsvImg = cv2.cvtColor(img, cv2.COLOR_RGB2HSV)
|
|
lower = (99, 254, 100)
|
|
upper = (100, 255, 101)
|
|
mask = cv2.inRange(hsvImg, lower, upper)
|
|
|
|
# Setup SimpleBlobDetector parameters.
|
|
params = cv2.SimpleBlobDetector_Params()
|
|
|
|
# Change thresholds
|
|
params.minThreshold = 10
|
|
params.maxThreshold = 255
|
|
|
|
params.filterByColor = True
|
|
params.blobColor = 255
|
|
|
|
params.filterByCircularity = False
|
|
params.filterByConvexity = False
|
|
params.filterByInertia = False
|
|
|
|
params.filterByArea = True
|
|
params.minArea = 10.0
|
|
|
|
detector = cv2.SimpleBlobDetector_create(params)
|
|
|
|
# Detect blobs.
|
|
keypoints = detector.detect(mask)
|
|
|
|
if len(keypoints) <= 0:
|
|
return None
|
|
|
|
return int(keypoints[0].pt[0]), int(keypoints[0].pt[1])
|
|
|
|
|
|
class PixelLoc:
|
|
"""
|
|
finds the pixel loc and store it
|
|
"""
|
|
|
|
val = None
|
|
|
|
@staticmethod
|
|
def config():
|
|
"""
|
|
Uses the game window to get an image of the game screen
|
|
then uses `GetKeypointFromImage()` to find the ProvisionsChalutier pixel location
|
|
:return: false if pixel loc not found
|
|
"""
|
|
|
|
PixelLoc.val = (0, 0, 1, 1)
|
|
|
|
return True
|