"""

“N”  新建文件夾 data/  用來存儲圖像
"S"   開始采集圖像，將采集到的圖像放到 data/ 路徑下
“Q”   退出窗口
"""

import numpy as np  # 數(shù)據(jù)處理的庫 Numpy
import cv2          # 圖像處理的庫 OpenCv
import os           # 讀寫文件
import shutil       # 讀寫文件
from PIL import Image, ImageDraw, ImageFont


# # OpenCv 調(diào)用攝像頭 / Use camera
cap = cv2.VideoCapture(0)
cap.set(cv2.CAP_PROP_FRAME_WIDTH,1920)
cap.set(cv2.CAP_PROP_FRAME_HEIGHT,1080)

"""
#功能函數(shù)，只是用來往圖片中顯示漢字
#示例 img = cv2ImgAddText(cv2.imread("img1.jpg"), "大家好，我是片天邊的云彩", 10, 65, (0, 0, 139), 20)
參數(shù)說明：
img：OpenCV圖片格式的圖片
text：要寫入的漢字
left：字符坐標(biāo)x值
top：字符坐標(biāo)y值
textColor：字體顏色
：textSize：字體大小
"""
def cv2ImgAddText(img, text, left, top, textColor=(0, 255, 0), textSize=20):
    if (isinstance(img, np.ndarray)):  # 判斷是否OpenCV圖片類型
        img = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
    # 創(chuàng)建一個可以在給定圖像上繪圖的對象
    draw = ImageDraw.Draw(img)
    # 字體的格式
    fontStyle = ImageFont.truetype(
        "font/simsun.ttc", textSize, encoding="utf-8")
    # 繪制文本
    draw.text((left, top), text, textColor, font=fontStyle)
    # 轉(zhuǎn)換回OpenCV格式
    return cv2.cvtColor(np.asarray(img), cv2.COLOR_RGB2BGR)

# 存儲圖像的文件夾 
current_dir = ""
# 保存  圖像 的路徑 
path_photos_from_camera = "data/"

press_n_flag = 0
cnt_ss=0



while cap.isOpened():
    flag, img_rd = cap.read()
    #print(img_rd.shape)

    kk = cv2.waitKey(2)
    # 待會要寫的字體 / Font to write
    font = cv2.FONT_ITALIC

    # 4. 按下 "n" 新建存儲人臉的文件夾 / press "n" to create the folders for saving faces
    if kk == ord("N") or kk == ord("n"):
        current_dir = path_photos_from_camera
        #os.makedirs(current_dir)
        if os.path.isdir(current_dir):
            pass
        else:
            os.mkdir(current_dir)
        print("
")
        print("新建的保存圖像的文件夾 / Create folders: ", current_dir)

        press_n_flag = 1        # 已經(jīng)按下 "n" / have pressed "n"


    # 5. 按下 "s" 保存攝像頭中的圖像到本地 / Press "s" to save image into local images
    if kk == ord("S") or kk == ord("s"):
        # 檢查有沒有先按"n"新建文件夾 / check if you have pressed "n"
        if press_n_flag:
            cnt_ss += 1
            cv2.imwrite(current_dir + "/img_" + str(cnt_ss) + ".jpg", img_rd)
            print("寫入本地 / Save into：", str(current_dir) + "/img_face_" + str(cnt_ss) + ".jpg")
        else:
            print("請在按 "S" 之前先按 "N" 來建文件夾 / Please press "N" before "S"")


    # 添加說明 / Add some statements
    #cv2.putText(img_rd, "Face Register", (20, 40), font, 1, (0, 255, 0), 1, cv2.LINE_AA)
    img_rd = cv2ImgAddText(img_rd, "圖片采集系統(tǒng)", 160, 25, (0, 255,0), 30)
    #cv2.putText(img_rd, "N: Create face folder", (20, 350), font, 0.8, (0, 255, 0), 1, cv2.LINE_AA)
    img_rd = cv2ImgAddText(img_rd, "N: 創(chuàng)建保存圖像文件夾", 20, 350, (0, 255, 0), 20)
    #cv2.putText(img_rd, "S: Save current face", (20, 400), font, 0.8, (0, 255, 0), 1, cv2.LINE_AA)
    img_rd = cv2ImgAddText(img_rd, "S: 保存當(dāng)前圖片", 20, 400, (0, 255, 0), 20)
    #cv2.putText(img_rd, "Q: Quit", (20, 450), font, 0.8, (0, 0, 0), 1, cv2.LINE_AA)
    img_rd = cv2ImgAddText(img_rd, "Q: 退出", 20, 450, (0, 255, 0), 20)

    # 6. 按下 "Q" 鍵退出 / Press "q" to exit
    if kk == ord("Q") or kk == ord("q"):
        break
    # 如果需要攝像頭窗口大小可調(diào) / Uncomment this line if you want the camera window is resizeable
    cv2.namedWindow("camera", 0)
    cv2.imshow("camera", img_rd)

# 釋放攝像頭 / Release camera and destroy all windows
cap.release()
cv2.destroyAllWindows()

pip install pillow

sudo apt-get install python3-pillow

/*****************************************************
2021.5.18:按鍵采集圖像
******************************************************/
#include "opencv2/core/core.hpp"    
#include "opencv2/imgproc/imgproc.hpp"    
#include "opencv2/calib3d/calib3d.hpp"    
#include "opencv2/highgui/highgui.hpp"    
#include <iostream>    
#include <fstream>    

using namespace cv;
using namespace std;

#define SRC_WIDTH  1920
#define SRC_HEIGHT 1080

int main()
{
	//測試視頻
	VideoCapture capture;
	capture.open(1);
	//capture.open("v4l2src device=/dev/video4 ! video/x-raw,width=1920,height=1020,framerate=30/1 ! videoconvert ! appsink");
	if (!capture.isOpened())
	{
		printf("文件打開失敗");
	}
	capture.set(CAP_PROP_FRAME_WIDTH, SRC_WIDTH);        //設(shè)置寬度
	capture.set(CAP_PROP_FRAME_HEIGHT, SRC_HEIGHT);  //設(shè)置長度
	Mat frame;
	int n = 0;
	char* cstr = new char[120];
	while (true)
	{
		
		capture >> frame;
		if (frame.data == NULL)
		{
			printf("Image is empty
");
			//writer.write(frame);
			break;
			//continue;
		}
		char kk=waitKey(2);
		if (kk == "S" || kk == "s")
		{

			sprintf(cstr, "%s%d%s", "caliberation/", n++, ".jpg");
			imwrite(cstr, frame);
			printf("保存了圖片
");

		}

		
		namedWindow("111", 0);//參數(shù)為零，則可以自由拖動
		imshow("111", frame);
		waitKey(2);
	}

	return 0;

}