From afeed542658925e843ca5dbc32f7e9f146a2a219 Mon Sep 17 00:00:00 2001
From: Fedor Kondratenko <dostoyewski@yandex.ru>
Date: Mon, 25 Apr 2022 15:37:27 +0300
Subject: [PATCH] Initial commit

---
 generate_marker.py   |  11 +++++++++++
 main.py              |  25 +++++++++++++++++++++++++
 markers/marker31.png | Bin 0 -> 1068 bytes
 markers/marker32.png | Bin 0 -> 1116 bytes
 markers/marker33.png | Bin 0 -> 1059 bytes
 5 files changed, 36 insertions(+)
 create mode 100644 generate_marker.py
 create mode 100644 main.py
 create mode 100644 markers/marker31.png
 create mode 100644 markers/marker32.png
 create mode 100644 markers/marker33.png

diff --git a/generate_marker.py b/generate_marker.py
new file mode 100644
index 0000000..4386160
--- /dev/null
+++ b/generate_marker.py
@@ -0,0 +1,11 @@
+import cv2 as cv
+import numpy as np
+
+# Load the predefined dictionary
+dictionary = cv.aruco.Dictionary_get(cv.aruco.DICT_6X6_250)
+
+# Generate the marker
+markerImage = np.zeros((200, 200), dtype=np.uint8)
+markerImage = cv.aruco.drawMarker(dictionary, 33, 200, markerImage, 1)
+
+cv.imwrite("./markers/marker33.png", markerImage)
diff --git a/main.py b/main.py
new file mode 100644
index 0000000..6bfa11e
--- /dev/null
+++ b/main.py
@@ -0,0 +1,25 @@
+import cv2 as cv
+import numpy as np
+import pandas as pd
+
+if __name__ == '__main__':
+    cap = cv.VideoCapture('./test.mp4')
+    while True:
+        success, image = cap.read()
+        if image is None:
+            break
+        key = cv.waitKey(1) & 0xFF
+        if (key == 27) or (key == ord('q')):
+            break
+        if success:
+            dictionary = cv.aruco.Dictionary_get(cv.aruco.DICT_6X6_250)
+            parameters = cv.aruco.DetectorParameters_create()
+            markerCorners, markerIds, rejectedCandidates = cv.aruco.detectMarkers(image,
+                                                                                  dictionary,
+                                                                                  parameters=parameters)
+            if markerIds is not None:
+                coords = list(np.mean(markerCorners[0], axis=1)[0].astype('int'))
+                image = cv.circle(image, coords, 20, (255, 0, 0), -1)
+            cv.imshow("original", image)
+        else:
+            continue
diff --git a/markers/marker31.png b/markers/marker31.png
new file mode 100644
index 0000000000000000000000000000000000000000..75ab98c978d4db9b8ea6bba662c84084ed3cbdf3
GIT binary patch
literal 1068
zcmeAS@N?(olHy`uVBq!ia0vp^CqS43NHBC5e{^DCVE*js;uuoFc=zB&-U9{<hYeo-
zUwoI_B!xp@%9QyNNQLUX3Xbzknpi3?aR@!}RQ8xaGl5eaMOiGIKpTvj=14gn;t*QW
zqp;j<f<Y6Drt*oEf+{@<jzd>TOW}N=H#{W1djh>T7&(t!f#EF<bX}Z^<Ee^?ESj_u
zcyeQhzu(D?h4bf3N`7o-se1faZIM@B@9!s5WTd~J)U-5vH#xW{=Z^Yv8Y%RY{O&2?
p*fgbZev6=r$Dri=X(16FVLvMR>!h)nbT_CN@O1TaS?83{1OQVDu1o*`

literal 0
HcmV?d00001

diff --git a/markers/marker32.png b/markers/marker32.png
new file mode 100644
index 0000000000000000000000000000000000000000..26069643286aeba5f82e49350218a48fce8700a7
GIT binary patch
literal 1116
zcmeAS@N?(olHy`uVBq!ia0vp^CqS43NHBC5e{^DCU{UsTaSW+oynFB<C$pmf^TBt2
znx`pWa5j)hTm3}v6Gvr>jgy>W{{;6F$^^O7ofKRDEDi8btk)C?IuXy+D)NaALT8*5
zT{WKc3>B(IVa`M+JArc^it-%kOD6EQ7zdqTKP0L7g!y5Q$fw2vH?E%!7Rs$OP?olE
zf)dc~p+dcDaq8MI$!VxipkQ5-qPjW2LojU9q)3e=9NLj5bwq+%rmjg5<I+6j6=uZU
xDw3$GJ+l?$uc@H)8gbHVK(rB;ond+Z2c!3sUnezu+qpr-hNr8a%Q~loCIEj-*ku3!

literal 0
HcmV?d00001

diff --git a/markers/marker33.png b/markers/marker33.png
new file mode 100644
index 0000000000000000000000000000000000000000..0a7c1dd448a6a5a854c0b46512a490104d6be34e
GIT binary patch
literal 1059
zcmeAS@N?(olHy`uVBq!ia0vp^CqS43NHBC5e{^DCV1DK4;uuoFc=zBzPG&~|=7aD4
zG*45$;B3&IxcZ6UCyvS%8z(u%{t50UlnHYE6IEOP1V?x({uc=gI$_^BRpb*Lgzj+O
zpX~Im<-Mn3`JwMiCRjhL4mvUSgRSNhzaR5NiqvcTxpoTgQ*Wh#vb2Xib0$VwD5s~0
zojV~pv(>oBjUGa3BAd8c=S&<bRAhvwAXnr`oi$51TGynAg$1=Zg&A>cYdR@v&jcos
zg%hTpIpoz9aKbAsQPou=MRoIrNlGG{CJlr(Phl=gVXnxbLY*X>C6V)QU8~!}li8E3
Q4uNvNr>mdKI;Vst0NxFR`v3p{

literal 0
HcmV?d00001