39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
|
import cv2
|
||
|
|
import numpy as np
|
||
|
|
import os
|
||
|
|
import config
|
||
|
|
|
||
|
|
os.makedirs(config.dir_pos, exist_ok=True)
|
||
|
|
id_pos=0
|
||
|
|
while os.path.isfile(config.dir_pos+"image-{:d}.png".format(id_pos)):
|
||
|
|
id_pos+=1
|
||
|
|
|
||
|
|
os.makedirs(config.dir_neg, exist_ok=True)
|
||
|
|
id_neg=0
|
||
|
|
while os.path.isfile(config.dir_neg+"image-{:d}.png".format(id_neg)):
|
||
|
|
id_neg+=1
|
||
|
|
|
||
|
|
cap=cv2.VideoCapture(0)
|
||
|
|
width=int(cap.get(3))
|
||
|
|
|
||
|
|
while True:
|
||
|
|
ret, frame=cap.read()
|
||
|
|
|
||
|
|
cv2.rectangle(frame, (0, 0), (width, 30), (100, 100, 100), cv2.FILLED)
|
||
|
|
cv2.putText(frame, "[p] photo positive [n] photo negative [q] quitter", (10, 20), cv2.FONT_HERSHEY_PLAIN, 1, (255, 255, 255), 1)
|
||
|
|
|
||
|
|
cv2.imshow('Camera', frame)
|
||
|
|
key=cv2.waitKey(1)&0xFF
|
||
|
|
if key==ord('p'):
|
||
|
|
fichier=config.dir_pos+"image-{:d}.png".format(id_pos)
|
||
|
|
print("Création du fichier", fichier)
|
||
|
|
cv2.imwrite(fichier, frame)
|
||
|
|
id_pos+=1
|
||
|
|
if key==ord('n'):
|
||
|
|
fichier=config.dir_neg+"image-{:d}.png".format(id_neg)
|
||
|
|
print("Création du fichier", fichier)
|
||
|
|
cv2.imwrite(fichier, frame)
|
||
|
|
id_neg+=1
|
||
|
|
if key==ord('q'):
|
||
|
|
quit()
|