From 3f088ffb7d3f744e5b4711b305404e9fbda8eae7 Mon Sep 17 00:00:00 2001 From: Tom Selier Date: Thu, 14 Sep 2023 14:09:13 +0200 Subject: [PATCH] inner corners detected --- src/helpers/template_extraction/script.py | 43 ++++++++++++++++++++++- 1 file changed, 42 insertions(+), 1 deletion(-) diff --git a/src/helpers/template_extraction/script.py b/src/helpers/template_extraction/script.py index 1f79605..fbee1ee 100644 --- a/src/helpers/template_extraction/script.py +++ b/src/helpers/template_extraction/script.py @@ -6,6 +6,17 @@ import os # command: # python template.py input_folder output_folder +# Returns the correct corner based on IDs +def getCorner(id): + if(id == 0): + return 2 + if(id == 1): + return 3 + if(id == 2): + return 0 + if(id == 3): + return 1 + if(len(sys.argv) != 3): print("Wrong amount of arguments") @@ -35,8 +46,38 @@ for file in os.listdir(input_folder): small_img, corners, ids) + for id in ids: + print(id) + print(corners[id[0]]) + print() + + i = 0 + x = np.zeros(4, dtype=np.int32) + y = np.zeros(4, dtype=np.int32) + for corner in corners: + # print(ids[i]) + # print(corner[0]) + # print() + + x[i] = corner[0][getCorner(ids[i])][0] + y[i] = corner[0][getCorner(ids[i])][1] + cv2.circle( + new_image, + (x[i], y[i]), + radius=5, + color=(255, 100, 255), + thickness=-1) + + print(x[i]) + print(y[i]) + print() + + i += 1 + + print("imshape[0]: " + str(small_img.shape[0])) + print("imshape[1]: " + str(small_img.shape[1])) cv2.imshow('window', new_image) - cv2.imwrite(out_filename, new_image) + # cv2.imwrite(out_filename, new_image) cv2.waitKey(0) cv2.destroyAllWindows \ No newline at end of file