more book fixes

This commit is contained in:
2022-04-29 12:57:28 +02:00
parent ebafbdf4b2
commit 36e8607ae4
10 changed files with 220 additions and 10 deletions

View File

@@ -97,6 +97,8 @@ def run():
needles.append(cv.imread("equip/chest_2_32.jpg", cv.IMREAD_UNCHANGED))
needles.append(cv.imread("equip/book_1_32.jpg", cv.IMREAD_UNCHANGED))
needles.append(cv.imread("equip/book_2_32.jpg", cv.IMREAD_UNCHANGED))
needles.append(cv.imread("equip/book_3_32.jpg", cv.IMREAD_UNCHANGED))
needles.append(cv.imread("equip/book_4_32.jpg", cv.IMREAD_UNCHANGED))
needles.append(cv.imread("equip/book_5_32.jpg", cv.IMREAD_UNCHANGED))
#needles.append(cv.imread("equip/key_1_32.jpg", cv.IMREAD_UNCHANGED))
@@ -133,17 +135,17 @@ def run():
needles.append(cv.imread("equip/bag_6_32.jpg", cv.IMREAD_UNCHANGED))
# needles.append(cv.imread("equip/bag_7_32.jpg", cv.IMREAD_UNCHANGED))
needles.append(cv.imread("equip/book_3_32.jpg", cv.IMREAD_UNCHANGED))
needles.append(cv.imread("equip/book_4_32.jpg", cv.IMREAD_UNCHANGED))
needles.append(cv.imread("equip/book_6_32.jpg", cv.IMREAD_UNCHANGED))
needles.append(cv.imread("equip/book_7_32.jpg", cv.IMREAD_UNCHANGED))
#needles.append(cv.imread("equip/book_6_32.jpg", cv.IMREAD_UNCHANGED))
#needles.append(cv.imread("equip/book_7_32.jpg", cv.IMREAD_UNCHANGED))
hsv = []
hsv.append(HsvFilter(14, 0, 0, 179, 255, 255, 0, 0, 0, 0))
hsv.append(HsvFilter(0, 128, 0, 179, 255, 255, 0, 0, 0, 0))
hsv.append(HsvFilter(49, 0, 0, 179, 255, 255, 0, 0, 0, 0))
hsv.append(HsvFilter(0, 0, 0, 179, 255, 196, 0, 0, 0, 0))
hsv.append(HsvFilter(0, 156, 0, 179, 255, 255, 0, 0, 0, 0))
hsv.append(HsvFilter(0, 95, 137, 179, 255, 255, 0, 0, 0, 0))
for op in range(1, 50, 1):
hsv.append(HsvFilter(0, 0, 0, 179, 255, 255, 0, 0, 0, 0))
@@ -155,9 +157,9 @@ def run():
processed_needle = vision_stun.apply_hsv_filter(needles[rer], hsv[rer])
rectangles = vision_stun.find(processed_screenshot, processed_needle, 0.8, 5)
# draw the detection results onto the original image
#output_image = vision_stun.draw_rectangles(processed_screenshot, rectangles)
#cv.imshow("output_image", output_image)
#cv.waitKey(150)
output_image = vision_stun.draw_rectangles(processed_screenshot, rectangles)
cv.imshow("output_image", output_image)
cv.waitKey(150)
if len(rectangles) is not 5:
break