Initial commit

Tensorflow/tutoriel30/enregistrement.py

@@ -0,0 +1,41 @@
+import cv2
+import numpy as np
+import os
+import config
+
+saut=10
+
+dir=config.dir_neg
+#dir=config.dir_pos
+os.makedirs(dir, exist_ok=True)
+
+id=0
+while os.path.isfile(dir+"image-{:d}.png".format(id)):
+    id+=1
+id*=saut
+
+cap=cv2.VideoCapture(0)
+width=int(cap.get(3))
+enregistre=0
+while True:
+    ret, frame=cap.read()
+
+    if enregistre:
+        if not id%saut:
+            id_=int(id/saut)
+            fichier=dir+"image-{:d}.png".format(id_)
+            print("Création du fichier", fichier)
+            cv2.imwrite(fichier, frame)
+        id+=1
+
+    cv2.rectangle(frame, (0, 0), (width, 30), (100, 100, 100), cv2.FILLED)
+    cv2.putText(frame, "[e] enregistrement  repertoire: {}  [q] quitter".format(dir), (10, 20), cv2.FONT_HERSHEY_PLAIN, 1, (255, 255, 255), 0)
+    if enregistre:
+        cv2.circle(frame, (width-20, 15), 5, (0, 0, 255), 8)
+
+    cv2.imshow('Camera', frame)
+    key=cv2.waitKey(1)&0xFF
+    if key==ord('e'):
+        enregistre=not enregistre
+    if key==ord('q'):
+        quit()